Initial

Author: vic4key

client.py

@@ -152,14 +152,15 @@ def ws_start(self, use_ssl):
 			on_error=self.ws_on_error,
 		)
 
-		ssl_context = None
+		ssl_opt = None
 		if use_ssl: # websocket secure
 			ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLS_CLIENT)
 			ssl_context.load_verify_locations(self.m_sslfile)
+			ssl_opt = {"context": ssl_context}
 
 		def run(*args):
 			try:
-				ws.run_forever(sslopt={"context": ssl_context})
+				ws.run_forever(sslopt=ssl_opt)
 			except Exception as e:
 				ws.close()
 				self.ws_stop_thread(ws)

test/README.TXT

@@ -0,0 +1,6 @@
+1. Install
+	pip install -r requirements.txt
+
+2. Usage
+	Websocket: python websocket_server.py
+	Websocket Secure: python websocket_server.py secure

test/requirements.txt

@@ -0,0 +1 @@
+websockets

test/websocket_server.py

@@ -0,0 +1,32 @@
+import sys, os
+import pathlib
+import websockets, asyncio, ssl
+
+import logging
+logger = logging.getLogger('websockets')
+logger.setLevel(logging.INFO)
+logger.addHandler(logging.StreamHandler())
+
+async def handler(ws, path):
+    try:
+        while True:
+            data = await ws.recv()
+            print(f"WS Server ({ws.path}) sent '{data}'")
+            await ws.send(data)
+    except websockets.exceptions.ConnectionClosedOK as e:
+        print("Client closed", e)
+    except Exception as e:
+        print("Exception:", e)
+
+if __name__ == "__main__":
+    # create ssl context for websocket secure
+    ssl_context = None
+    if len(sys.argv) == 2 and sys.argv[1].lower() == "secure":
+        normalize_path = lambda path: path.replace("\\", "/").replace("/", os.path.sep)
+        ssl_file_path = normalize_path(os.path.join(pathlib.Path(__file__).resolve().parent.parent, "preferences/local.pem"))
+        ssl_context = ssl.SSLContext(ssl.PROTOCOL_TLS_SERVER)
+        ssl_context.load_cert_chain(ssl_file_path)
+    # start websocket server
+    ws_server = websockets.serve(handler, "127.0.0.1", 1609, ssl=ssl_context)
+    asyncio.get_event_loop().run_until_complete(ws_server)
+    asyncio.get_event_loop().run_forever()