Python websocket server

garciadelcastillo · garciadelcastillo · commit 307272517791 · 2021-06-28T17:21:06.000-04:00
diff --git a/Misc/WebSockets/python-websockets/client.py b/Misc/WebSockets/python-websockets/client.py
@@ -1,20 +1,19 @@
-
+# Importing the relevant libraries
 import websockets
 import asyncio
 
-async def hello():
-    url = "ws://127.0.0.1:7890"
-    async with websockets.connect(url) as ws:
-        await ws.send("Hello Server!")
-
-
+# The main function that will handle connection and communication 
+# with the server
 async def listen():
     url = "ws://127.0.0.1:7890"
-
+    # Connect to the server
     async with websockets.connect(url) as ws:
+        # Send a greeting message
+        await ws.send("Hello Server!")
+        # Stay alive forever, listening to incoming msgs
         while True:
             msg = await ws.recv()
             print(msg)
 
-asyncio.get_event_loop().run_until_complete(hello())
+# Start the connection
 asyncio.get_event_loop().run_until_complete(listen())
diff --git a/Misc/WebSockets/python-websockets/server.py b/Misc/WebSockets/python-websockets/server.py
diff --git a/Misc/WebSockets/python-websockets/server_echo.py b/Misc/WebSockets/python-websockets/server_echo.py
@@ -0,0 +1,22 @@
+# Importing the relevant libraries
+import websockets
+import asyncio
+
+PORT = 7890
+
+print("Server listening on Port " + str(PORT))
+
+async def echo(websocket, path):
+    print("A client just connected")
+    try:
+        async for message in websocket:
+            print("Received message from client: " + message)
+            await websocket.send("Pong: " + message)
+    except websockets.exceptions.ConnectionClosed as e:
+        print("A client just disconnected")
+
+start_server = websockets.serve(echo, "localhost", PORT)
+
+asyncio.get_event_loop().run_until_complete(start_server)
+asyncio.get_event_loop().run_forever()
+
diff --git a/Misc/WebSockets/python-websockets/server_echo_broadcast.py b/Misc/WebSockets/python-websockets/server_echo_broadcast.py
@@ -0,0 +1,35 @@
+# Importing the relevant libraries
+import websockets
+import asyncio
+
+# Server data
+PORT = 7890
+print("Server listening on Port " + str(PORT))
+
+# A set of connected ws clients
+connected = set()
+
+# The main behavior function for this server
+async def echo(websocket, path):
+    print("A client just connected")
+    # Store a copy of the connected client
+    connected.add(websocket)
+    # Handle incoming messages
+    try:
+        async for message in websocket:
+            print("Received message from client: " + message)
+            # Send a response to all connected clients except sender
+            for conn in connected:
+                if conn != websocket:
+                    await conn.send("Someone said: " + message)
+    # Handle disconnecting clients 
+    except websockets.exceptions.ConnectionClosed as e:
+        print("A client just disconnected")
+    finally:
+        connected.remove(websocket)
+
+# Start the server
+start_server = websockets.serve(echo, "localhost", PORT)
+asyncio.get_event_loop().run_until_complete(start_server)
+asyncio.get_event_loop().run_forever()
+
