Upload app.py

app.py

@@ -128,9 +128,24 @@ class CombinedHandler(http.server.SimpleHTTPRequestHandler):
             websocket_server = serve(proxy_handler, "localhost", ws_port)
             loop.run_until_complete(websocket_server)
 
-        # 创建到WebSocket服务器的连接
-        ws_client = websockets.connect(f"ws://localhost:{ws_port}")
-        asyncio.get_event_loop().run_until_complete(ws_client)
+        # 创建新的事件循环
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+
+        try:
+            # 创建到WebSocket服务器的连接
+            ws_client = loop.run_until_complete(websockets.connect(f"ws://localhost:{ws_port}"))
+            
+            # 在这里处理WebSocket连接
+            # 例如：loop.run_until_complete(self.handle_websocket_communication(ws_client))
+
+        finally:
+            loop.close()
+
+    # 添加这个新方法来处理WebSocket通信
+    async def handle_websocket_communication(self, ws_client):
+        # 在这里实现WebSocket通信逻辑
+        pass
 
     @staticmethod
     def find_free_port():
@@ -146,7 +161,8 @@ def run_server(port=PORT):
             print(f"HTTP 服务器运行在 http://{DOMAIN}:{port}")
             
             # 启动 WebSocket 服务器
-            loop = asyncio.get_event_loop()
+            loop = asyncio.new_event_loop()
+            asyncio.set_event_loop(loop)
             ws_port = CombinedHandler.find_free_port()
             websocket_server = serve(proxy_handler, "localhost", ws_port)
             loop.run_until_complete(websocket_server)
@@ -159,6 +175,8 @@ def run_server(port=PORT):
                 print("服务器正在关闭...")
                 httpd.shutdown()
                 loop.run_until_complete(websocket_server.close())
+            finally:
+                loop.close()
 
 if __name__ == "__main__":
     refresh_tokens = get_refresh_tokens()