8 - Websocket Client in Python

Author: codingforentrepreneurs

client.py

@@ -0,0 +1,18 @@
+import asyncio
+import websockets
+import sys
+
+websocket_endpoint = 'ws://localhost:8765' # url / uro
+
+async def connect_to_ws(msg):
+    async with websockets.connect(websocket_endpoint) as ws:
+        await ws.send(msg)
+
+if __name__=="__main__":
+    msg = 'nothing here'
+    try:
+        msg = sys.argv[1]
+    except:
+        pass
+    loop = asyncio.get_event_loop()
+    loop.run_until_complete(connect_to_ws(msg))

index.py

@@ -1,9 +1,18 @@
 import asyncio
 import websockets
+import json
 
-async def webhook_handler(*args, **kwargs):
-    print(args, kwargs)
-    pass
+async def webhook_handler(websocket, path):
+    print(websocket, path)
+    async for message in websocket:
+        data = {}
+        try:
+            data = json.loads(message)
+        except:
+            pass
+        print(data.get('data'))
+        # print(message)
+    # pass
 
 
 server = websockets.serve(webhook_handler, 'localhost', 8765)