Ver código fonte

fixing websocket

ming 3 anos atrás
pai
commit
7218a45655
1 arquivos alterados com 14 adições e 0 exclusões
  1. 14 0
      main.py

+ 14 - 0
main.py

@@ -21,6 +21,8 @@ import re
 import urllib.request
 from fastapi.responses import FileResponse
 from websocket import create_connection
+import asyncio
+import websockets
 from fastapi.middleware.cors import CORSMiddleware
 import dataset
 from datetime import datetime
@@ -492,7 +494,19 @@ def generate_subtitle_image(name_hash,text_content):
 
 
 
+async def hello(uri):
+    async with websockets.connect(uri) as websocket:
+        await websocket.send(100)
+        print(f"(client) send to server: Jimmy")
+        name = await websocket.recv()
+        print(f"(client) recv from server {name}")
+
+#asyncio.get_event_loop().run_until_complete(
+#    hello('ws://www.choozmo.com:8888/progress'))
+
 def anchor_video_v2(name,text_content, image_urls,avatar):
+    asyncio.get_event_loop().run_until_complete(
+    hello('ws://www.choozmo.com:8888/progress'))
     progress = 0
     name_hash = str(time.time()).replace('.','')