unicex 0.8.0__tar.gz → 0.10.0__tar.gz

{unicex-0.8.0/unicex.egg-info → unicex-0.10.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unicex
-Version: 0.8.0
+Version: 0.10.0
 Summary: Unified Crypto Exchange API 
 Author-email: LoveBloodAndDiamonds <ayazshakirzyanov27@gmail.com>
 License: BSD 3-Clause License
@@ -43,6 +43,7 @@ Requires-Dist: eth-account>=0.13.7
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: msgpack>=1.1.1
 Requires-Dist: orjson>=3.11.3
+Requires-Dist: protobuf>=6.32.1
 Requires-Dist: websockets>=15.0.1
 Dynamic: license-file
 
@@ -56,11 +57,11 @@ Dynamic: license-file
 |-----------------|--------|------|------------|---------|------------|----------------|--------------|
 | **Binance**     | ✓      | ✓    | ✓          | ✓       | ✓          | ✓              |              |
 | **Bitget**      | ✓      | ✓    | ✓          |         | ✓          |                |              |
-| **Bybit**       | ✓      | ✓    |            |         | ✓          |                |              |
-| **Gateio**      | ✓      | ✓    |            |         | ✓          |                |              |
-| **Hyperliquid** | ✓      | ✓    |            |         | ✓          |                |              |
-| **Mexc**        | ✓      | ✓    |            |         | ✓          |                |              |
-| **Okx**         | ✓      | ✓    |            |         | ✓          |                | ✓            |
+| **Bybit**       | ✓      | ✓    | ✓          |         | ✓          |                |              |
+| **Gateio**      | ✓      | ✓    | ✓          |         | ✓          |                |              |
+| **Hyperliquid** | ✓      | ✓    | ✓          | ✓       | ✓          |                |              |
+| **Mexc**        | ✓      | ✓    | ✓          |         | ✓          |                |              |
+| **Okx**         | ✓      | ✓    | ✓          |         | ✓          |                | ✓            |
 ---
 
 

{unicex-0.8.0 → unicex-0.10.0}/README.md

@@ -8,11 +8,11 @@
 |-----------------|--------|------|------------|---------|------------|----------------|--------------|
 | **Binance**     | ✓      | ✓    | ✓          | ✓       | ✓          | ✓              |              |
 | **Bitget**      | ✓      | ✓    | ✓          |         | ✓          |                |              |
-| **Bybit**       | ✓      | ✓    |            |         | ✓          |                |              |
-| **Gateio**      | ✓      | ✓    |            |         | ✓          |                |              |
-| **Hyperliquid** | ✓      | ✓    |            |         | ✓          |                |              |
-| **Mexc**        | ✓      | ✓    |            |         | ✓          |                |              |
-| **Okx**         | ✓      | ✓    |            |         | ✓          |                | ✓            |
+| **Bybit**       | ✓      | ✓    | ✓          |         | ✓          |                |              |
+| **Gateio**      | ✓      | ✓    | ✓          |         | ✓          |                |              |
+| **Hyperliquid** | ✓      | ✓    | ✓          | ✓       | ✓          |                |              |
+| **Mexc**        | ✓      | ✓    | ✓          |         | ✓          |                |              |
+| **Okx**         | ✓      | ✓    | ✓          |         | ✓          |                | ✓            |
 ---
 
 

{unicex-0.8.0 → unicex-0.10.0}/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.0"
+version = "0.10.0"
 
 description = "Unified Crypto Exchange API "
 readme = "README.md"
@@ -19,6 +19,7 @@ dependencies = [
     "loguru>=0.7.3",
     "msgpack>=1.1.1",
     "orjson>=3.11.3",
+    "protobuf>=6.32.1",
     "websockets>=15.0.1",
 ]
 

{unicex-0.8.0 → unicex-0.10.0}/unicex/_abc/__init__.py

@@ -1,3 +1,5 @@
+"""Пакет с абстракциями и интерфейсами."""
+
 __all__ = [
     "IUniClient",
     "IUniWebsocketManager",

{unicex-0.8.0 → unicex-0.10.0}/unicex/_abc/uni_client.py

@@ -115,7 +115,7 @@ class IUniClient(ABC, Generic[TClient]):
         Возвращает:
             `bool`: True, если апи ключи присутствуют, иначе False.
         """
-        return self._client._api_key is not None and self._client._api_secret is not None
+        return self._client.is_authorized()
 
     async def close_connection(self) -> None:
         """Закрывает сессию клиента."""

unicex-0.10.0/unicex/_base/__init__.py

@@ -0,0 +1,9 @@
+"""Пакет с базовым клиентом для HTTP запросов и базовым вебсокетом."""
+
+__all__ = [
+    "BaseClient",
+    "Websocket",
+]
+
+from .client import BaseClient
+from .websocket import Websocket

{unicex-0.8.0 → unicex-0.10.0}/unicex/_base/websocket.py

@@ -3,7 +3,7 @@ __all__ = ["Websocket"]
 import asyncio
 import time
 from collections.abc import Awaitable, Callable
-from typing import Any
+from typing import Any, Protocol
 
 import orjson
 import websockets
@@ -20,18 +20,30 @@ class Websocket:
     MAX_QUEUE_SIZE: int = 100
     """Максимальная длина очереди."""
 
+    class _DecoderProtocol(Protocol):
+        """Протокол декодирования сообщений."""
+
+        def decode(self, message: Any) -> dict: ...
+
+    class _JsonDecoder:
+        """Протокол декодирования сообщений в формате JSON."""
+
+        def decode(self, message: Any) -> dict:
+            return orjson.loads(message)
+
     def __init__(
         self,
         callback: Callable[[Any], Awaitable[None]],
         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[_DecoderProtocol] = _JsonDecoder,
         **kwargs: Any,  # Не дадим сломаться, если юзер передал ненужные аргументы
     ) -> None:
         """Инициализация вебсокета.
@@ -41,12 +53,13 @@ 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`): Количество рабочих задач для обработки сообщений.
             logger (`LoggerLike | None`): Логгер для записи логов.
+            decoder (`IDecoder | None`): Декодер для обработки входящих сообщений.
         """
         self._callback = callback
         self._url = url
@@ -59,6 +72,7 @@ class Websocket:
         self._last_message_time = time.monotonic()
         self._worker_count = worker_count
         self._logger = logger or _logger
+        self._decoder = decoder()
         self._tasks: list[asyncio.Task] = []
         self._queue = asyncio.Queue()
         self._running = False
@@ -104,24 +118,26 @@ 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()
 
-    async def _handle_message(self, message: str) -> None:
+    async def _handle_message(self, message: str | bytes) -> None:
         """Обрабатывает входящее сообщение вебсокета."""
         try:
             # Обновленяем время последнего сообщения
             self._last_message_time = time.monotonic()
 
             # Ложим сообщение в очередь, предварительно его сериализуя
-            await self._queue.put(orjson.loads(message))
+            decoded_message = self._decoder.decode(message)
+            await self._queue.put(decoded_message)
 
             # Проверяем размер очереди сообщений и выбрасываем ошибку, если он превышает максимальный размер
             self._check_queue_size()
@@ -139,7 +155,7 @@ class Websocket:
         """Проверяет размер очереди и выбрасывает ошибку при переполнении."""
         qsize = self._queue.qsize()
         if qsize >= self.MAX_QUEUE_SIZE:
-            raise QueueOverflowError("Message queue is overflow")
+            raise QueueOverflowError(f"Message queue is overflow: {qsize}")
 
     async def _after_connect(self, conn: ClientConnection) -> None:
         """Вызывается после установки соединения."""
@@ -201,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.0 → unicex-0.10.0}/unicex/binance/client.py

@@ -62,8 +62,8 @@ class Client(BaseClient):
         if not signed:
             return {"params": params, "data": data}, None
 
-        if not self._api_key or not self._api_secret:  # type: ignore[attr-defined]
-            raise NotAuthorized("Api key is required to private endpoints")
+        if not self.is_authorized():
+            raise NotAuthorized("Api key and api secret is required to private endpoints")
 
         # Объединяем все параметры в payload
         payload = {**params, **data}

{unicex-0.8.0 → unicex-0.10.0}/unicex/bitget/client.py

@@ -5,6 +5,7 @@ import time
 from typing import Any, Literal
 
 from unicex._base import BaseClient
+from unicex.exceptions import NotAuthorized
 from unicex.types import RequestMethod
 from unicex.utils import (
     dict_to_query_string,
@@ -20,6 +21,18 @@ class Client(BaseClient):
     _BASE_URL: str = "https://api.bitget.com"
     """Базовый URL для REST API Bitget."""
 
+    def is_authorized(self) -> bool:
+        """Проверяет наличие API‑ключей у клиента.
+
+        Возвращает:
+            `bool`: Признак наличия ключей.
+        """
+        return (
+            self._api_key is not None
+            and self._api_secret is not None
+            and self._api_passphrase is not None
+        )
+
     def _sign_message(
         self,
         method: RequestMethod,
@@ -45,6 +58,9 @@ class Client(BaseClient):
                 - `timestamp (str)`: Временная метка в миллисекундах.
                 - `signature (str)`: Подпись в формате base64.
         """
+        if not self.is_authorized():
+            raise NotAuthorized("Api key and api secret is required to private endpoints")
+
         timestamp = str(int(time.time() * 1000))
 
         path = f"{endpoint}?{dict_to_query_string(params)}" if params else endpoint
@@ -68,16 +84,15 @@ class Client(BaseClient):
             `dict[str, str]`: Словарь заголовков запроса.
         """
         headers = {"Content-Type": "application/json", "Accept": "application/json"}
-        if self._api_key:  # type: ignore[attr-defined]
-            headers.update(
-                {
-                    "ACCESS-KEY": self._api_key,  # type: ignore[attr-defined]
-                    "ACCESS-PASSPHRASE": self._api_passphrase,  # type: ignore[attr-defined]
-                    "ACCESS-TIMESTAMP": timestamp,
-                    "ACCESS-SIGN": signature,
-                    "locale": "en-US",
-                }
-            )
+        headers.update(
+            {
+                "ACCESS-KEY": self._api_key,  # type: ignore[attr-defined]
+                "ACCESS-PASSPHRASE": self._api_passphrase,  # type: ignore[attr-defined]
+                "ACCESS-TIMESTAMP": timestamp,
+                "ACCESS-SIGN": signature,
+                "locale": "en-US",
+            }
+        )
         return headers
 
     def _prepare_request_params(

{unicex-0.8.0 → unicex-0.10.0}/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.0 → unicex-0.10.0}/unicex/bybit/client.py

@@ -41,12 +41,12 @@ class Client(BaseClient):
         Источник: https://github.com/bybit-exchange/api-usage-examples/blob/master/V5_demo/api_demo/Encryption_HMAC.py
         """
         # Проверяем наличие апи ключей для подписи запроса
-        if not self._api_key or not self._api_secret:
-            raise NotAuthorized("API key and secret are required to private endpoints.")
+        if not self.is_authorized():
+            raise NotAuthorized("Api key and api secret is required to private endpoints")
 
         dumped_payload = json.dumps(payload)
-        prepared_query_string = timestamp + self._api_key + self._RECV_WINDOW + dumped_payload
-        return generate_hmac_sha256_signature(self._api_secret, prepared_query_string)
+        prepared_query_string = timestamp + self._api_key + self._RECV_WINDOW + dumped_payload  # type: ignore[attrDefined]
+        return generate_hmac_sha256_signature(self._api_secret, prepared_query_string)  # type: ignore[attrDefined]
 
     async def _make_request(
         self,