Added example and made into proper package

Author: jbonzo

client/__init__.py

@@ -0,0 +1 @@
+from .websocket_client import WebSocketClient, STOCKS_CLUSTER, FOREX_CLUSTER, CRYPTO_CLUSTER

polygon-client.py client/websocket_client.pypolygon-client.py renamed to client/websocket_client.py

@@ -15,15 +15,15 @@ class WebSocketClient(object):
     # TODO: Either an instance of the client couples 1:1 with the cluster or an instance of the Client couples 1:3 with
     #  the 3 possible clusters (I think I like client per, but then a problem is the user can make multiple clients for
     #  the same cluster and that's not desirable behavior,
-    #  somehow keeping track with multiple Client instances will be the issue)
+    #  somehow keeping track with multiple Client instances will be the difficulty)
     def __init__(self, cluster: str, auth_key: str, process_message: Optional[Callable[[str], None]] = None):
         self._host = self.DEFAULT_HOST
         self.url = f"wss://{self._host}/{cluster}"
-        self.auth_key = auth_key
-        self.ws: websocket.WebSocketApp = websocket.WebSocketApp(self.url, on_error=self._default_on_error,
+        self.ws: websocket.WebSocketApp = websocket.WebSocketApp(self.url, on_open=self._default_on_open(),
                                                                  on_close=self._default_on_close,
-                                                                 on_message=self._default_on_message(),
-                                                                 on_open=self._authenticate())
+                                                                 on_error=self._default_on_error,
+                                                                 on_message=self._default_on_message())
+        self.auth_key = auth_key
 
         self.process_message = process_message
 

example.py

@@ -0,0 +1,22 @@
+import time
+
+from client import WebSocketClient, STOCKS_CLUSTER
+
+
+def my_customer_process_message(message):
+    print("this is my custom message processing", message)
+
+
+def main():
+    key = 'your api key'
+    my_client = WebSocketClient(STOCKS_CLUSTER, key, my_customer_process_message)
+    my_client.run_async()
+
+    my_client.subscribe("T.MSFT", "T.AAPL", "T.AMD", "T.NVDA")
+    time.sleep(1)
+
+    my_client.close_connection()
+
+
+if __name__ == "__main__":
+    main()