update

Author: theshadow76

pocketoptionapi/ws/__pycache__/client.cpython-310.pyc

@@ -59,6 +59,21 @@ def __init__(self, api) -> None:
         self.esperar_segundo_mensaje = False
         self.recibido_updateClosedDeals = False
 
+    async def reconnect(self):
+        regs = self.region.get_regions()
+        for i in regs:
+            print(f"Reconnecting to {i}...")
+            async with websockets.connect(i, extra_headers={"Origin": "https://m.pocketoption.com"}) as ws:
+
+                print("Conectado a: ", i)
+                self.websocket = ws
+                self.url = i
+                on_message_task = asyncio.create_task(self.websocket_listener(ws))
+                sender_task = asyncio.create_task(send_pin(ws))
+                message_task = asyncio.create_task(self.send_message(self.message))
+                await asyncio.gather(on_message_task, sender_task, message_task)
+
+
     async def websocket_listener(self, ws):
         async for message in ws:
             await self.on_message(message)
@@ -139,6 +154,15 @@ async def on_message(self, message):  # pylint: disable=unused-argument
                 global_value.balance = message["balance"]
                 global_value.balance_type = message["isDemo"]
 
+                data = {
+                    "balance_id" :  message["uid"],
+                    "balance" : message["balance"],
+                    "balance_type" : message["isDemo"]
+                }
+
+                with open("balance.json", "w") as f:
+                    json.dump(data, f)
+
             elif "requestId" in message and message["requestId"] == 'buy':
                 global_value.order_data = message
 
@@ -220,11 +244,17 @@ async def on_error(self, error):  # pylint: disable=unused-argument
         """Method to process websocket errors."""
         logger = logging.getLogger(__name__)
         logger.error(error)
-        global_value.websocket_error_reason = str(error)
-        global_value.check_websocket_if_error = True
+        try:
+            self.reconnect()
+        except:
+            global_value.websocket_error_reason = str(error)
+            global_value.check_websocket_if_error = True
 
     async def on_close(self, error):  # pylint: disable=unused-argument
         """Method to process websocket close."""
         logger = logging.getLogger(__name__)
         logger.debug("Websocket connection closed.")
-        global_value.websocket_is_connected = False
+        try:
+            self.reconnect()
+        except:
+            global_value.websocket_is_connected = False

pocketoptionapi/ws/client.py

@@ -8,13 +8,10 @@
 api = PocketOption(ssid)
 print(ssid)
 
-
 def direction():
     # Selecciona aleatoriamente entre 'call' y 'put'
     return random.choice(['call', 'put'])
 
-
 if __name__ == "__main__":
     api.connect()
     time.sleep(2)
-    print(f"Get balance: {api.get_balance()}")