crypto-ws-api 2.0.7__tar.gz → 2.0.9__tar.gz

{crypto-ws-api-2.0.7 → crypto-ws-api-2.0.9}/CHANGELOG.md

@@ -1,3 +1,11 @@
+## 2.0.9 - 2024-04-14
+### Fix
+* Creating asynchronous tasks done right
+
+## 2.0.8 - 2024-03-31
+### Fix
+* tons log records: `websockets socket.send() raised exception.`
+
 ## 2.0.7 - 2024-03-25
 ### Update
 * Refactoring and some minor fixes

{crypto-ws-api-2.0.7 → crypto-ws-api-2.0.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: crypto-ws-api
-Version: 2.0.7
+Version: 2.0.9
 Summary: Crypto WS API connector for ASYNC requests
 Author-email: Jerry Fedorenko <jerry.fedorenko@yahoo.com>
 Requires-Python: >=3.8

{crypto-ws-api-2.0.7 → crypto-ws-api-2.0.9}/crypto_ws_api/__init__.py

@@ -13,7 +13,7 @@ __maintainer__ = "Jerry Fedorenko"
 __contact__ = "https://github.com/DogsTailFarmer"
 __email__ = "jerry.fedorenko@yahoo.com"
 __credits__ = ["https://github.com/DanyaSWorlD"]
-__version__ = "2.0.7"
+__version__ = "2.0.9"
 
 from pathlib import Path
 import shutil
@@ -21,7 +21,7 @@ from platformdirs import user_config_path
 
 
 TIMEOUT = 5  # sec timeout for WSS initialization and get response
-DELAY = 0.1  # sec delay in while loop
+DELAY = 0.1  # sec delay in keepalive loop
 # Maximum str size for unique query ID
 ID_LEN_LIMIT = {
     "binance": 36,

{crypto-ws-api-2.0.7 → crypto-ws-api-2.0.9}/crypto_ws_api/ws_session.py

@@ -66,7 +66,7 @@ class UserWSS:
         "request_limit_reached",
         "in_event",
         "_response_pool",
-        "tasks_list",
+        "tasks",
     )
 
     def __init__(self, method, ws_id, exchange, endpoint, api_key, api_secret, passphrase=None):
@@ -87,10 +87,17 @@ class UserWSS:
         self.order_handling = False
         self.request_limit_reached = False
         self.in_event = asyncio.Event()
-        self.tasks_list = []
+        self.tasks = set()
+
+    def tasks_manage(self, coro, name=None):
+        _t = asyncio.create_task(coro)
+        if name:
+            _t.set_name(name)
+        self.tasks.add(_t)
+        _t.add_done_callback(self.tasks.discard)
 
     async def _ws_listener(self):
-        asyncio.ensure_future(self.ws_login())
+        self.tasks_manage(self.ws_login())
         async for msg in self._ws:
             # logger.info(f"_ws_listener: msg: {self.ws_id}: {msg}")
             if isinstance(msg, str):
@@ -104,7 +111,7 @@ class UserWSS:
                     elif self.exchange in ['okx', 'bitfinex']:
                         self._response_pool[res.get('id') or self.ws_id] = res.get('data') or res
                     self.in_event.set()
-                    await asyncio.sleep(0)
+                await asyncio.sleep(0)
             else:
                 logger.warning(f"UserWSS: {self.ws_id}: {msg}")
                 await self.stop()
@@ -119,8 +126,8 @@ class UserWSS:
                     break
                 else:
                     self.operational_status = False
-                    [task.cancel() for task in self.tasks_list if not task.done()]
-                    self.tasks_list.clear()
+                    [task.cancel() for task in self.tasks if not task.done()]
+                    self.tasks.clear()
                     logger.warning(f"Restart WSS for {self.ws_id}")
                     continue
             except Exception as ex:
@@ -134,16 +141,12 @@ class UserWSS:
         else:
             if self.exchange == 'binance':
                 self._listen_key = res.get('listenKey')
-                _t = asyncio.ensure_future(self.heartbeat())
-                _t.set_name(f"heartbeat-{self.ws_id}")
-                self.tasks_list.append(_t)
+                self.tasks_manage(self.heartbeat(), f"heartbeat-{self.ws_id}")
             else:
                 self._listen_key = f"{int(time.time() * 1000)}{self.ws_id}"
             self.operational_status = True
             self.order_handling = True
-            _t = asyncio.ensure_future(self._keepalive())
-            _t.set_name(f"keepalive-{self.ws_id}")
-            self.tasks_list.append(_t)
+            self.tasks_manage(self._keepalive(), f"keepalive-{self.ws_id}")
             logger.info(f"UserWSS: 'logged in' for {self.ws_id}")
 
     async def request(self, _method=None, _params=None, _api_key=False, _signed=False):
@@ -171,9 +174,10 @@ class UserWSS:
         await self._ws.send(
             json.dumps(self.compose_request(_id, _api_key, method, params, _signed))
         )
+        await asyncio.sleep(0)
         try:
             res = await asyncio.wait_for(self._response_distributor(_id), timeout=TIMEOUT)
-        except asyncio.TimeoutError:
+        except asyncio.exceptions.TimeoutError:
             logger.warning(f"UserWSS: get response timeout error: {self.ws_id}")
             await self.stop()
         except asyncio.CancelledError:
@@ -274,8 +278,8 @@ class UserWSS:
         self.operational_status = None  # Not restart and break all loops
         self.order_handling = False
         self.init = True
-        [task.cancel() for task in self.tasks_list if not task.done()]
-        self.tasks_list.clear()
+        [task.cancel() for task in self.tasks if not task.done()]
+        self.tasks.clear()
         if self._ws and not self._ws.closed:
             await self._ws.close(code=4000)
         gc.collect()
@@ -296,7 +300,7 @@ class UserWSS:
         elif self.exchange == 'bitfinex':
             return await self.bitfinex_error_handle(msg)
 
-    #region BitfinexErrorHandle
+    # region BitfinexErrorHandle
     async def bitfinex_error_handle(self, msg):
         if isinstance(msg, dict):
             return await self._handle_dict_message(msg)
@@ -348,7 +352,7 @@ class UserWSS:
                 msg[2][7]
             ]
         }
-    #endregion
+    # endregion
 
     async def okx_error_handle(self, msg):
         if msg.get('code') == '1':
@@ -390,6 +394,7 @@ class UserWSSession:
         "_api_secret",
         "_passphrase",
         "user_wss",
+        "tasks_wss",
     )
 
     def __init__(self, exchange, endpoint, api_key, api_secret, passphrase=None):
@@ -402,6 +407,7 @@ class UserWSSession:
         self._api_secret = api_secret
         self._passphrase = passphrase
         self.user_wss = {}
+        self.tasks_wss = set()
 
     async def handle_request(
         self,
@@ -428,7 +434,9 @@ class UserWSSession:
         if user_wss.init:
             user_wss.init = False
             user_wss.operational_status = False
-            asyncio.ensure_future(user_wss.start_wss())
+            _t = asyncio.create_task(user_wss.start_wss())
+            self.tasks_wss.add(_t)
+            _t.add_done_callback(self.tasks_wss.discard)
 
         duration = 0
         while not (user_wss.operational_status and user_wss.order_handling):
@@ -450,3 +458,5 @@ class UserWSSession:
         for ws in user_wss_copy.values():
             await ws.stop()
         self.user_wss.clear()
+        [task.cancel() for task in self.tasks_wss if not task.done()]
+        self.tasks_wss.clear()