saving

test.py

@@ -0,0 +1,63 @@
+import asyncio
+import json
+import websockets
+import time
+
+# Credentials
+API_KEY = "019d2ad3-3755-744b-ace8-ad0f08c958dd"
+API_SECRET = "vXT1UeliaP89z9vcxDtdv47422mftijJkrJYE7CFqvA="
+API_PASSPHRASE = "57e703b801f22333d1a66a48c3a71773d3d3a42825ddcf330c3325856bc99756"
+WS_URL = "wss://ws-subscriptions-clob.polymarket.com/ws/user"
+
+async def heartbeat(websocket):
+    """Sends a heartbeat every 10 seconds to keep the connection alive."""
+    while True:
+        try:
+            await asyncio.sleep(10)
+            await websocket.send(json.dumps({}))
+            # print("Heartbeat sent") 
+        except Exception:
+            break
+
+async def connect_polymarket_user_ws():
+    while True: # Outer loop for reconnection
+        try:
+            async with websockets.connect(WS_URL) as websocket:
+                subscribe_message = {
+                    "type": "user",
+                    "auth": {
+                        "apiKey": API_KEY,
+                        "secret": API_SECRET,
+                        "passphrase": API_PASSPHRASE
+                    },
+                    "markets": [] 
+                }
+
+                await websocket.send(json.dumps(subscribe_message))
+                print(f"[{time.strftime('%H:%M:%S')}] Subscription sent...")
+
+                # Start the heartbeat task in the background
+                heartbeat_task = asyncio.create_task(heartbeat(websocket))
+
+                async for message in websocket:
+                    data = json.loads(message)
+                    
+                    # Log the specific reason if it's an error message
+                    if data.get("type") == "error":
+                        print(f"Server Error: {data.get('message')}")
+                        break
+                    
+                    if data: # Ignore empty heartbeat responses from server
+                        print(f"Update: {data}")
+
+                heartbeat_task.cancel()
+
+        except Exception as e:
+            print(f"Connection lost: {e}. Retrying in 5s...")
+            await asyncio.sleep(5)
+
+if __name__ == "__main__":
+    try:
+        asyncio.run(connect_polymarket_user_ws())
+    except KeyboardInterrupt:
+        print("Stopped by user.")