unicex 0.13.3__tar.gz → 0.13.5__tar.gz

{unicex-0.13.3/unicex.egg-info → unicex-0.13.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unicex
-Version: 0.13.3
+Version: 0.13.5
 Summary: Unified Crypto Exchange API 
 Author-email: LoveBloodAndDiamonds <ayazshakirzyanov27@gmail.com>
 License: BSD 3-Clause License

{unicex-0.13.3 → unicex-0.13.5}/pyproject.toml

@@ -4,7 +4,7 @@ name = "unicex"
 # •	PATCH (x.y.Z) → увеличивается при багфиксе, который не ломает совместимость.
 # •	MINOR (x.Y.z) → увеличивается при добавлении новой функциональности, но без ломающих изменений (backward-compatible).
 # •	MAJOR (X.y.z) → увеличивается при изменениях, которые ломают обратную совместимость.
-version = "0.13.3"
+version = "0.13.5"
 
 description = "Unified Crypto Exchange API "
 readme = "README.md"

{unicex-0.13.3 → unicex-0.13.5}/unicex/_base/websocket.py

@@ -17,7 +17,7 @@ from unicex.types import LoggerLike
 class Websocket:
     """Базовый класс асинхронного вебсокета."""
 
-    MAX_QUEUE_SIZE: int = 100
+    MAX_QUEUE_SIZE: int = 500
     """Максимальная длина очереди."""
 
     class _DecoderProtocol(Protocol):
@@ -95,12 +95,7 @@ class Websocket:
     async def stop(self) -> None:
         """Останавливает вебсокет и рабочие задачи."""
         self._running = False
-        for task in self._tasks:
-            task.cancel()
-        self._tasks.clear()
-
-        # Очистка очереди
-        self._queue = asyncio.Queue()
+        await self._after_disconnect()
 
     async def restart(self) -> None:
         """Перезапускает вебсокет."""
@@ -111,25 +106,27 @@ class Websocket:
     async def _connect(self) -> None:
         """Подключается к вебсокету и настраивает соединение."""
         self._logger.debug(f"Establishing connection with {self._url}")
-        while self._running:
-            async with websockets.connect(uri=self._url, **self._generate_ws_kwargs()) as conn:
-                try:
-                    self._logger.info(f"Websocket connection was established to {self._url}")
-                    await self._after_connect(conn)
-
-                    # Цикл получения сообщений
-                    while self._running:
-                        message = await conn.recv()
-                        await self._handle_message(message)
-
-                except websockets.exceptions.ConnectionClosed as e:
-                    self._logger.error(f"Websocket connection was closed unexpectedly: {e}")
-                except Exception as e:
-                    self._logger.error(f"Unexpected error in websosocket connection: {e}")
-                finally:
-                    if self._running:
-                        await asyncio.sleep(self._reconnect_timeout)
-                        await self._after_disconnect()
+        async for conn in websockets.connect(uri=self._url, **self._generate_ws_kwargs()):
+            try:
+                self._logger.info(f"Websocket connection was established to {self._url}")
+                await self._after_connect(conn)
+
+                # Цикл получения сообщений
+                while self._running:
+                    message = await conn.recv()
+                    await self._handle_message(message)
+
+            except websockets.exceptions.ConnectionClosed as e:
+                self._logger.error(f"Websocket connection was closed unexpectedly: {e}")
+            except Exception as e:
+                self._logger.error(f"Unexpected error in websosocket connection: {e}")
+            finally:
+                # Делаем реконнект только если вебсокет активен, иначе выходим из итератора
+                if self._running:
+                    await asyncio.sleep(self._reconnect_timeout)
+                    await self._after_disconnect()
+                else:
+                    return  # Выходим из итератора, если вебсокет уже выключен
 
     async def _handle_message(self, message: str | bytes) -> None:
         """Обрабатывает входящее сообщение вебсокета."""
@@ -182,11 +179,22 @@ class Websocket:
 
     async def _after_disconnect(self) -> None:
         """Вызывается после отключения от вебсокета."""
+        # Останавливаем воркеров
         for task in self._tasks:
             task.cancel()
+
+        # Дожидаемся завершения задач (в т.ч. воркеров)
+        for task in self._tasks:
+            try:
+                await task
+            except asyncio.CancelledError:
+                pass
+            except Exception as e:
+                self._logger.warning(f"Worker raised during shutdown: {e}")
+
         self._tasks.clear()
 
-        # Очистить очередь
+        # Очистить очередь уже безопасно, после остановки воркеров
         self._queue = asyncio.Queue()
 
     async def _send_subscribe_messages(self, conn: ClientConnection) -> None:

{unicex-0.13.3 → unicex-0.13.5/unicex.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unicex
-Version: 0.13.3
+Version: 0.13.5
 Summary: Unified Crypto Exchange API 
 Author-email: LoveBloodAndDiamonds <ayazshakirzyanov27@gmail.com>
 License: BSD 3-Clause License