unicex 0.17.9__tar.gz → 0.17.11__tar.gz

{unicex-0.17.9/unicex.egg-info → unicex-0.17.11}/PKG-INFO

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

{unicex-0.17.9 → unicex-0.17.11}/pyproject.toml

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

{unicex-0.17.9 → unicex-0.17.11}/unicex/_base/websocket.py

@@ -113,7 +113,7 @@ class Websocket:
         self._logger.debug(f"Establishing connection with {self._url}")
         async for conn in websockets.connect(uri=self._url, **self._generate_ws_kwargs()):
             try:
-                self._logger.debug(f"Websocket connection was established to {self._url}")
+                self._logger.info(f"Websocket connection was established to {self._url}")
                 await self._after_connect(conn)
 
                 # Цикл получения сообщений
@@ -151,7 +151,8 @@ class Websocket:
                 # Проверяем размер очереди сообщений и выбрасываем ошибку, если он превышает максимальный размер
                 self._check_queue_size()
         except QueueOverflowError:
-            self._logger.error("Message queue is overflow")
+            cleaned_messages = self._clear_queue()
+            self._logger.error(f"Message queue is overflow, cleaned {cleaned_messages} messages")
         except orjson.JSONDecodeError as e:
             if message in ["ping", "pong"]:
                 self._logger.debug(f"Received ping message: {message}")
@@ -166,6 +167,18 @@ class Websocket:
         if qsize >= self.MAX_QUEUE_SIZE:
             raise QueueOverflowError(f"Message queue is overflow: {qsize}")
 
+    def _clear_queue(self) -> int:
+        """Очищает очередь сообщений."""
+        cleared = 0
+        while True:
+            try:
+                self._queue.get_nowait()
+                self._queue.task_done()
+                cleared += 1
+            except asyncio.QueueEmpty:
+                break
+        return cleared
+
     async def _after_connect(self, conn: ClientConnection) -> None:
         """Вызывается после установки соединения."""
         # Подписываемся на топики
@@ -261,7 +274,9 @@ class Websocket:
 
         while self._running:
             if time.monotonic() - self._last_message_time > self._no_message_reconnect_timeout:
-                self._logger.error("Websocket is not responding, restarting...")
+                self._logger.error(
+                    f"No messages in {self._no_message_reconnect_timeout} seconds, restarting..."
+                )
                 await self.restart()
                 return
             await asyncio.sleep(1)

unicex-0.17.11/unicex/kucoin/websocket_manager.py

@@ -0,0 +1,121 @@
+__all__ = ["WebsocketManager"]
+
+import json
+import time
+from collections.abc import Awaitable, Callable, Sequence
+from typing import Any, Literal
+
+from unicex._base import Websocket
+from unicex.utils import validate_single_symbol_args
+
+from .client import Client
+
+type CallbackType = Callable[[Any], Awaitable[None]]
+
+
+class WebsocketManager:
+    """Менеджер асинхронных вебсокетов для Kucoin."""
+
+    _SPOT_URL: str = "wss://x-push-spot.kucoin.com"
+    """Базовый URL для вебсокета на спот."""
+
+    _FUTURES_URL: str = "wss://x-push-futures.kucoin.com"
+    """Базовый URL для вебсокета на фьючерсы."""
+
+    def __init__(self, client: Client | None = None, **ws_kwargs: Any) -> None:
+        """Инициализирует менеджер вебсокетов для Kucoin.
+
+        Параметры:
+            client (`Client | None`): Клиент для выполнения запросов. Нужен, чтобы открыть приватные вебсокеты.
+            ws_kwargs (`dict[str, Any]`): Дополнительные аргументы, которые прокидываются в `Websocket`.
+        """
+        self.client = client
+        self._ws_kwargs = ws_kwargs
+
+    def _get_url(self, trade_type: Literal["SPOT", "FUTURES"]) -> str:
+        """Возвращает URL для указанного типа рынка."""
+        if trade_type == "SPOT":
+            return self._SPOT_URL
+        if trade_type == "FUTURES":
+            return self._FUTURES_URL
+        raise ValueError(f"Unsupported trade type: {trade_type}")
+
+    def _normalize_depth(self, depth: str | int) -> str:
+        """Нормализует значение глубины стакана."""
+        depth_value = str(depth)
+        if depth_value not in {"1", "5", "50", "increment"}:
+            raise ValueError("depth must be one of: 1, 5, 50, increment")
+        return depth_value
+
+    def _build_subscription_messages(
+        self,
+        trade_type: Literal["SPOT", "FUTURES"],
+        depth: str,
+        symbols: Sequence[str],
+        rpi_filter: Literal[0, 1],
+        request_id: str | None = None,
+    ) -> list[str]:
+        """Формирует сообщения для подписки."""
+        base_id = int(time.time() * 1000)
+        messages: list[str] = []
+        for index, symbol in enumerate(symbols):
+            payload = {
+                "id": request_id or str(base_id + index),
+                "action": "SUBSCRIBE",
+                "channel": "obu",
+                "tradeType": trade_type,
+                "symbol": symbol.upper(),
+                "depth": depth,
+                "rpiFilter": rpi_filter,
+            }
+            messages.append(json.dumps(payload))
+        return messages
+
+    def orderbook(
+        self,
+        callback: CallbackType,
+        trade_type: Literal["SPOT", "FUTURES"],
+        depth: Literal["1", "5", "50", "increment"] | int = "1",
+        symbol: str | None = None,
+        symbols: Sequence[str] | None = None,
+        rpi_filter: Literal[0, 1] = 0,
+        request_id: str | None = None,
+    ) -> Websocket:
+        """Создает вебсокет для получения order book.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            trade_type (`Literal["SPOT", "FUTURES"]`): Тип рынка.
+            depth (`Literal["1", "5", "50", "increment"] | int`): Глубина стакана.
+            symbol (`str | None`): Один символ для подписки.
+            symbols (`Sequence[str] | None`): Список символов для мультиплекс‑подключения.
+            rpi_filter (`Literal[0, 1]`): Фильтр RPI. Доступен только для фьючерсов (depth=5/50).
+            request_id (`str | None`): Опциональный идентификатор запроса.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        validate_single_symbol_args(symbol, symbols)
+
+        depth_value = self._normalize_depth(depth)
+        if rpi_filter not in (0, 1):
+            raise ValueError("rpi_filter must be 0 or 1")
+        if trade_type == "SPOT" and rpi_filter == 1:
+            raise ValueError("rpi_filter=1 is supported only for FUTURES")
+        if rpi_filter == 1 and depth_value not in {"5", "50"}:
+            raise ValueError("rpi_filter=1 supports only depth=5 or depth=50")
+
+        tickers = [symbol] if symbol else symbols
+        subscription_messages = self._build_subscription_messages(
+            trade_type=trade_type,
+            depth=depth_value,
+            symbols=tickers,  # type: ignore[arg-type]
+            rpi_filter=rpi_filter,
+            request_id=request_id,
+        )
+        return Websocket(
+            callback=callback,
+            url=self._get_url(trade_type),
+            subscription_messages=subscription_messages,
+            **self._ws_kwargs,
+        )

{unicex-0.17.9 → unicex-0.17.11/unicex.egg-info}/PKG-INFO

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

unicex-0.17.9/unicex/kucoin/websocket_manager.py

@@ -1,11 +0,0 @@
-__all__ = ["WebsocketManager"]
-
-
-from collections.abc import Awaitable, Callable
-from typing import Any
-
-type CallbackType = Callable[[Any], Awaitable[None]]
-
-
-class WebsocketManager:
-    """Менеджер асинхронных вебсокетов для Kucoin."""