Browse Source

merged duplicated Server implementation

Rodja Trappe 2 years ago
parent
commit
e106db1264
1 changed files with 3 additions and 9 deletions
  1. 3 9
      nicegui/run.py

+ 3 - 9
nicegui/run.py

@@ -18,15 +18,6 @@ from .language import Language
 from .storage import RequestTrackingMiddleware
 
 
-class Server(uvicorn.Server):
-
-    def run(self, sockets: List[Any] = None) -> None:
-        if self.config.storage_secret is not None:
-            globals.app.add_middleware(RequestTrackingMiddleware)
-            globals.app.add_middleware(SessionMiddleware, secret_key=self.config.storage_secret)
-        super().run(sockets=sockets)
-
-
 class Server(uvicorn.Server):
 
     def run(self, sockets: Optional[List[socket.socket]] = None) -> None:
@@ -35,6 +26,9 @@ class Server(uvicorn.Server):
         native_module.response_queue = self.config.response_queue
         if native_module.method_queue is not None:
             globals.app.native.main_window = native_module.WindowProxy()
+        if self.config.storage_secret is not None:
+            globals.app.add_middleware(RequestTrackingMiddleware)
+            globals.app.add_middleware(SessionMiddleware, secret_key=self.config.storage_secret)
         super().run(sockets=sockets)