unicex 0.8.1__tar.gz → 0.10.1__tar.gz

{unicex-0.8.1/unicex.egg-info → unicex-0.10.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unicex
-Version: 0.8.1
+Version: 0.10.1
 Summary: Unified Crypto Exchange API 
 Author-email: LoveBloodAndDiamonds <ayazshakirzyanov27@gmail.com>
 License: BSD 3-Clause License
@@ -58,10 +58,10 @@ Dynamic: license-file
 | **Binance**     | ✓      | ✓    | ✓          | ✓       | ✓          | ✓              |              |
 | **Bitget**      | ✓      | ✓    | ✓          |         | ✓          |                |              |
 | **Bybit**       | ✓      | ✓    | ✓          |         | ✓          |                |              |
-| **Gateio**      | ✓      | ✓    |            |         | ✓          |                |              |
-| **Hyperliquid** | ✓      | ✓    |            |         | ✓          |                |              |
-| **Mexc**        | ✓      | ✓    |            |         | ✓          |                |              |
-| **Okx**         | ✓      | ✓    |            |         | ✓          |                | ✓            |
+| **Gateio**      | ✓      | ✓    | ✓          |         | ✓          |                |              |
+| **Hyperliquid** | ✓      | ✓    | ✓          | ✓       | ✓          |                |              |
+| **Mexc**        | ✓      | ✓    | ✓          |         | ✓          |                |              |
+| **Okx**         | ✓      | ✓    | ✓          |         | ✓          |                | ✓            |
 ---
 
 

{unicex-0.8.1 → unicex-0.10.1}/README.md

@@ -9,10 +9,10 @@
 | **Binance**     | ✓      | ✓    | ✓          | ✓       | ✓          | ✓              |              |
 | **Bitget**      | ✓      | ✓    | ✓          |         | ✓          |                |              |
 | **Bybit**       | ✓      | ✓    | ✓          |         | ✓          |                |              |
-| **Gateio**      | ✓      | ✓    |            |         | ✓          |                |              |
-| **Hyperliquid** | ✓      | ✓    |            |         | ✓          |                |              |
-| **Mexc**        | ✓      | ✓    |            |         | ✓          |                |              |
-| **Okx**         | ✓      | ✓    |            |         | ✓          |                | ✓            |
+| **Gateio**      | ✓      | ✓    | ✓          |         | ✓          |                |              |
+| **Hyperliquid** | ✓      | ✓    | ✓          | ✓       | ✓          |                |              |
+| **Mexc**        | ✓      | ✓    | ✓          |         | ✓          |                |              |
+| **Okx**         | ✓      | ✓    | ✓          |         | ✓          |                | ✓            |
 ---
 
 

{unicex-0.8.1 → unicex-0.10.1}/pyproject.toml

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

{unicex-0.8.1 → unicex-0.10.1}/unicex/_abc/__init__.py

@@ -4,10 +4,8 @@ __all__ = [
     "IUniClient",
     "IUniWebsocketManager",
     "IExchangeInfo",
-    "IDecoder",
 ]
 
-from .decoder import IDecoder
 from .exchange_info import IExchangeInfo
 from .uni_client import IUniClient
 from .uni_websocket_manager import IUniWebsocketManager

{unicex-0.8.1 → unicex-0.10.1}/unicex/_base/websocket.py

@@ -3,7 +3,7 @@ __all__ = ["Websocket"]
 import asyncio
 import time
 from collections.abc import Awaitable, Callable
-from typing import TYPE_CHECKING, Any
+from typing import Any, Protocol
 
 import orjson
 import websockets
@@ -13,9 +13,6 @@ from websockets.asyncio.client import ClientConnection
 from unicex.exceptions import QueueOverflowError
 from unicex.types import LoggerLike
 
-if TYPE_CHECKING:
-    from unicex._abc import IDecoder
-
 
 class Websocket:
     """Базовый класс асинхронного вебсокета."""
@@ -23,10 +20,15 @@ class Websocket:
     MAX_QUEUE_SIZE: int = 100
     """Максимальная длина очереди."""
 
+    class _DecoderProtocol(Protocol):
+        """Протокол декодирования сообщений."""
+
+        def decode(self, message: Any) -> dict: ...
+
     class _JsonDecoder:
-        """Базовый JSON декодер для WebSocket сообщений."""
+        """Протокол декодирования сообщений в формате JSON."""
 
-        def decode(self, message: bytes | str) -> dict:
+        def decode(self, message: Any) -> dict:
             return orjson.loads(message)
 
     def __init__(
@@ -35,13 +37,13 @@ class Websocket:
         url: str,
         subscription_messages: list[dict] | list[str] | None = None,
         ping_interval: int | float = 10,
-        ping_message: str | None = None,
-        pong_message: str | None = None,
+        ping_message: str | Callable | None = None,
+        pong_message: str | Callable | None = None,
         no_message_reconnect_timeout: int | float | None = 60,
         reconnect_timeout: int | float | None = 5,
         worker_count: int = 2,
         logger: LoggerLike | None = None,
-        decoder: type["IDecoder"] = _JsonDecoder,
+        decoder: type[_DecoderProtocol] = _JsonDecoder,
         **kwargs: Any,  # Не дадим сломаться, если юзер передал ненужные аргументы
     ) -> None:
         """Инициализация вебсокета.
@@ -51,8 +53,8 @@ class Websocket:
             url (`str`): URL вебсокета.
             subscription_messages (`list[dict] | list[str] | None`): Сообщения для подписки после подключения.
             ping_interval (`int | float`): Интервал отправки ping, сек.
-            ping_message (`str | None`): Сообщение для ping (если не указано — используется ping‑frame).
-            pong_message (`str | None`): Сообщение для pong (если не указано — используется pong‑frame).
+            ping_message (`str | Callable | None`): Сообщение для ping, или функция генерации ping (если не указано — используется ping‑frame).
+            pong_message (`str | Callable | None`): Сообщение для pong, или функция генерации pong (если не указано — используется pong‑frame).
             no_message_reconnect_timeout (`int | float | None`): Таймаут ожидания без сообщений до рестарта, сек.
             reconnect_timeout (`int | float | None`): Пауза перед переподключением, сек.
             worker_count (`int`): Количество рабочих задач для обработки сообщений.
@@ -116,12 +118,13 @@ class Websocket:
 
                 # Цикл получения сообщений
                 while self._running:
-                    message = await conn.recv(decode=True)
+                    message = await conn.recv()
                     await self._handle_message(message)
 
-            except websockets.exceptions.ConnectionClosed:
-                self._logger.error("Websocket connection was closed unexpectedly")
-                continue
+            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:
                 await asyncio.sleep(self._reconnect_timeout)
                 await self._after_disconnect()
@@ -133,7 +136,8 @@ class Websocket:
             self._last_message_time = time.monotonic()
 
             # Ложим сообщение в очередь, предварительно его сериализуя
-            await self._queue.put(self._decoder.decode(message))
+            decoded_message = self._decoder.decode(message)
+            await self._queue.put(decoded_message)
 
             # Проверяем размер очереди сообщений и выбрасываем ошибку, если он превышает максимальный размер
             self._check_queue_size()
@@ -213,8 +217,12 @@ class Websocket:
         """Периодически отправляет пользовательский ping."""
         while self._running and self._ping_message:
             try:
-                await conn.send(self._ping_message)
-                self._logger.debug(f"Sent ping message: {self._ping_message}")
+                if isinstance(self._ping_message, Callable):
+                    ping_message = self._ping_message()
+                else:
+                    ping_message = self._ping_message
+                await conn.send(ping_message)
+                self._logger.debug(f"Sent ping message: {ping_message}")
             except Exception as e:
                 self._logger.error(f"Error sending ping: {e}")
                 return

{unicex-0.8.1 → unicex-0.10.1}/unicex/bitget/websocket_manager.py

@@ -15,7 +15,7 @@ type CallbackType = Callable[[Any], Awaitable[None]]
 class WebsocketManager:
     """Менеджер асинхронных вебсокетов для Bitget."""
 
-    _BASE_URL: str = "wss://ws.bitget.com/v2/ws/public"
+    _URL: str = "wss://ws.bitget.com/v2/ws/public"
     """Базовый URL для вебсокета."""
 
     def __init__(self, client: Client | None = None, **ws_kwargs: Any) -> None:
@@ -91,7 +91,7 @@ class WebsocketManager:
         )
         return Websocket(
             callback=callback,
-            url=self._BASE_URL,
+            url=self._URL,
             subscription_messages=subsription_messages,
             **self._ws_kwargs,
         )
@@ -124,7 +124,7 @@ class WebsocketManager:
         )
         return Websocket(
             callback=callback,
-            url=self._BASE_URL,
+            url=self._URL,
             subscription_messages=subscription_messages,
             **self._ws_kwargs,
         )
@@ -160,7 +160,7 @@ class WebsocketManager:
         )
         return Websocket(
             callback=callback,
-            url=self._BASE_URL,
+            url=self._URL,
             subscription_messages=subscription_messages,
             **self._ws_kwargs,
         )
@@ -195,7 +195,7 @@ class WebsocketManager:
         )
         return Websocket(
             callback=callback,
-            url=self._BASE_URL,
+            url=self._URL,
             subscription_messages=subscription_messages,
             **self._ws_kwargs,
         )
@@ -226,7 +226,7 @@ class WebsocketManager:
         )
         return Websocket(
             callback=callback,
-            url=self._BASE_URL,
+            url=self._URL,
             subscription_messages=subscription_messages,
             **self._ws_kwargs,
         )

{unicex-0.8.1 → unicex-0.10.1}/unicex/bybit/websocket_manager.py

@@ -15,19 +15,19 @@ type CallbackType = Callable[[Any], Awaitable[None]]
 class WebsocketManager:
     """Менеджер асинхронных вебсокетов для Bybit."""
 
-    _BASE_SPOT_URL: str = "wss://stream.bybit.com/v5/public/spot"
+    _SPOT_URL: str = "wss://stream.bybit.com/v5/public/spot"
     """Базовый URL для вебсокета на спот."""
 
-    _BASE_LINEAR_URL: str = "wss://stream.bybit.com/v5/public/linear"
+    _LINEAR_URL: str = "wss://stream.bybit.com/v5/public/linear"
     """Базовый URL для вебсокета на USDT/USDC перпетуалы и фьючерсы."""
 
-    _BASE_INVERSE_URL: str = "wss://stream.bybit.com/v5/public/inverse"
+    _INVERSE_URL: str = "wss://stream.bybit.com/v5/public/inverse"
     """Базовый URL для вебсокета на инверсные контракты."""
 
-    _BASE_OPTION_URL: str = "wss://stream.bybit.com/v5/public/option"
+    _OPTION_URL: str = "wss://stream.bybit.com/v5/public/option"
     """Базовый URL для вебсокета на опционы."""
 
-    _BASE_PRIVATE_URL: str = "wss://stream.bybit.com/v5/private"
+    _PRIVATE_URL: str = "wss://stream.bybit.com/v5/private"
     """Базовый URL для приватных вебсокетов."""
 
     def __init__(self, client: Client | None = None, **ws_kwargs: Any) -> None:
@@ -72,15 +72,15 @@ class WebsocketManager:
             `str`: URL для вебсокета.
         """
         if category == "spot":
-            return self._BASE_SPOT_URL
+            return self._SPOT_URL
         elif category == "linear":
-            return self._BASE_LINEAR_URL
+            return self._LINEAR_URL
         elif category == "inverse":
-            return self._BASE_INVERSE_URL
+            return self._INVERSE_URL
         elif category == "option":
-            return self._BASE_OPTION_URL
+            return self._OPTION_URL
         elif category == "private":
-            return self._BASE_PRIVATE_URL
+            return self._PRIVATE_URL
         else:
             raise ValueError(f"Unsupported category: {category}")