fix websocket

Author: Nawado-gg

eodhd/websocketclient.py

@@ -27,7 +27,7 @@ def __init__(
             raise ValueError("API key is invalid")
 
         # Validate endpoint
-        if endpoint not in ["us", "us_quote", "forex", "crypto"]:
+        if endpoint not in ["us", "us-quote", "forex", "crypto"]:
             raise ValueError("Endpoint is invalid")
 
         # Validate symbol list

example_websockets.py

@@ -1,30 +1,24 @@
 """Websocket example"""
 
-from time import sleep
+import time
 from eodhd import WebSocketClient
 
-
-def main() -> None:
-    """Main"""
-
-    websocket = WebSocketClient(
-        # Demo API key for testing purposes
+client = WebSocketClient(
         api_key="OeAFFmMliFG5orCUuwAKQ8l4WWFQ67YX",
         endpoint="crypto",
-        symbols=["BTC-USD"]
-        # api_key="OeAFFmMliFG5orCUuwAKQ8l4WWFQ67YX", endpoint="forex", symbols=["EURUSD"]
-        # api_key="OeAFFmMliFG5orCUuwAKQ8l4WWFQ67YX", endpoint="us", symbols=["AAPL"]
-    )
-    websocket.start()
-
-    message_count = 0
-    while True:
-        if websocket:
-            if message_count != websocket.message_count:
-                print(websocket.message)
-                message_count = websocket.message_count
-                sleep(0.25)  # output every 1/4 second, websocket is realtime
-
+        symbols=["BTC-USD"],
+        display_stream=False,
+        display_candle_1m=True,
+        display_candle_5m=False,
+        display_candle_1h=False,
+)
+client.start()
 
-if __name__ == "__main__":
-    main()
+try:
+    while client.running:
+        res = client.get_data()
+        print(res)
+        time.sleep(1)
+except KeyboardInterrupt:
+    print("\nStopping due to user request.")
+    client.stop()