unicex 0.13.17__py3-none-any.whl → 0.16.5__py3-none-any.whl

unicex/__init__.py

@@ -18,7 +18,6 @@ __all__ = [
     "TickerDailyItem",
     "KlineDict",
     "TradeDict",
-    "AggTradeDict",
     "RequestMethod",
     "LoggerLike",
     "OpenInterestDict",
@@ -81,6 +80,20 @@ __all__ = [
     "GateioWebsocketManager",
     "GateioUserWebsocket",
     "GateioExchangeInfo",
+    # Kucoin
+    "KucoinClient",
+    "KucoinUniClient",
+    "KucoinUniWebsocketManager",
+    "KucoinWebsocketManager",
+    "KucoinUserWebsocket",
+    "KucoinExchangeInfo",
+    # BingX
+    "BingXClient",
+    "BingXUniClient",
+    "BingXUniWebsocketManager",
+    "BingXWebsocketManager",
+    "BingXUserWebsocket",
+    "BingXExchangeInfo",
 ]
 
 # ruff: noqa
@@ -98,7 +111,6 @@ from .types import (
     TickerDailyItem,
     KlineDict,
     TradeDict,
-    AggTradeDict,
     RequestMethod,
     LoggerLike,
     OpenInterestDict,
@@ -173,6 +185,24 @@ from .okx import (
     ExchangeInfo as OkxExchangeInfo,
 )
 
+from .kucoin import (
+    Client as KucoinClient,
+    UniClient as KucoinUniClient,
+    UniWebsocketManager as KucoinUniWebsocketManager,
+    UserWebsocket as KucoinUserWebsocket,
+    WebsocketManager as KucoinWebsocketManager,
+    ExchangeInfo as KucoinExchangeInfo,
+)
+
+from .bingx import (
+    Client as BingXClient,
+    UniClient as BingXUniClient,
+    UniWebsocketManager as BingXUniWebsocketManager,
+    UserWebsocket as BingXUserWebsocket,
+    WebsocketManager as BingXWebsocketManager,
+    ExchangeInfo as BingXExchangeInfo,
+)
+
 
 async def load_exchanges_info() -> None:
     """Единожды загружает информацию о тикерах на всех биржах."""
@@ -184,6 +214,8 @@ async def load_exchanges_info() -> None:
         HyperliquidExchangeInfo.load_exchange_info(),
         MexcExchangeInfo.load_exchange_info(),
         OkxExchangeInfo.load_exchange_info(),
+        KucoinExchangeInfo.load_exchange_info(),
+        BingXExchangeInfo.load_exchange_info(),
     )
 
 
@@ -197,4 +229,6 @@ async def start_exchanges_info(parse_interval_seconds: int = 60 * 60) -> None:
         HyperliquidExchangeInfo.start(parse_interval_seconds),
         MexcExchangeInfo.start(parse_interval_seconds),
         OkxExchangeInfo.start(parse_interval_seconds),
+        KucoinExchangeInfo.start(parse_interval_seconds),
+        BingXExchangeInfo.start(parse_interval_seconds),
     )

unicex/_abc/exchange_info.py

@@ -17,7 +17,7 @@ if TYPE_CHECKING:
 
 
 class IExchangeInfo(ABC):
-    """Интерфейс для наследников, которые предзагружают и обновляют информацию о бирже."""
+    """Интерфейс класса, который обновляет информацию о правилах торговли на бирже."""
 
     _loaded: bool
     """Флаг, указывающий, была ли информация о бирже загружена."""

unicex/_abc/uni_client.py

@@ -1,5 +1,6 @@
 __all__ = ["IUniClient"]
 
+import time
 from abc import ABC, abstractmethod
 from typing import Generic, Self, TypeVar, overload
 
@@ -14,7 +15,7 @@ TClient = TypeVar("TClient", bound="BaseClient")
 
 
 class IUniClient(ABC, Generic[TClient]):
-    """Интерфейс для реализации асинхронного унифицированного клиента."""
+    """Интерфейс для унифицированного клиента."""
 
     def __init__(
         self,
@@ -175,6 +176,35 @@ class IUniClient(ABC, Generic[TClient]):
         tickers = await self.tickers(only_usdt)
         return batched_list(tickers, batch_size)
 
+    @staticmethod
+    def limit_to_start_and_end_time(
+        interval: Timeframe, limit: int, use_milliseconds: bool = True
+    ) -> tuple[int, int]:
+        """Преобразует `limit` в `start_time` и `end_time`.
+
+        Параметры:
+            interval (`Timeframe`): Интервал времени.
+            limit (`int`): Количество элементов.
+            use_milliseconds (`bool`): Использовать миллисекунды.
+
+        Нужен, потому что на некоторых биржах параметр `limit` не принимается напрямую.
+        """
+        end_time = int(time.time())
+        start_time = end_time - (limit * interval.to_seconds)  # type: ignore[reportOptionalOperand]
+        if use_milliseconds:
+            start_time *= 1000
+            end_time *= 1000
+        return start_time, end_time
+
+    @staticmethod
+    def to_seconds(value: int | None) -> int | None:
+        """Преобразует значение из миллисекунд в секунды для передачи в API."""
+        if value is None:
+            return None
+        if value >= 1_000_000_000_000:
+            return value // 1000
+        return value
+
     @abstractmethod
     async def futures_tickers(self, only_usdt: bool = True) -> list[str]:
         """Возвращает список тикеров.

unicex/_abc/uni_websocket_manager.py

@@ -16,7 +16,7 @@ type CallbackType = Callable[[Any], Awaitable[None]]
 
 
 class IUniWebsocketManager(ABC):
-    """Интерфейс менеджера асинхронных унифицированных вебсокетов."""
+    """Интерфейс менеджера унифицированных вебсокетов."""
 
     def __init__(
         self, client: BaseClient | IUniClient | None = None, logger: LoggerLike | None = None

unicex/_base/client.py

@@ -203,6 +203,9 @@ class BaseClient:
                 response_json=response_json,
             ) from None
 
+        # Валидирование ответа в конерктной реализации клиента
+        self._validate_response(response_json)
+
         # Логирование ответа
         try:
             self._logger.debug(
@@ -212,3 +215,7 @@ class BaseClient:
             self._logger.error(f"Error while logging response: {e}")
 
         return response_json
+
+    def _validate_response(self, response_json: dict[str, Any]) -> None:
+        """Проверка ответа API на ошибки биржи. Переопределяется в клиентах конкретных бирж."""
+        return None

unicex/_base/websocket.py

@@ -3,7 +3,7 @@ __all__ = ["Websocket"]
 import asyncio
 import time
 from collections.abc import Awaitable, Callable
-from typing import Any, Protocol
+from typing import Any, Literal, Protocol
 
 import orjson
 import websockets
@@ -23,12 +23,12 @@ class Websocket:
     class _DecoderProtocol(Protocol):
         """Протокол декодирования сообщений."""
 
-        def decode(self, message: Any) -> dict: ...
+        def decode(self, message: Any) -> dict | Literal["ping"]: ...
 
     class _JsonDecoder:
         """Протокол декодирования сообщений в формате JSON."""
 
-        def decode(self, message: Any) -> dict:
+        def decode(self, message: Any) -> dict | Literal["ping"]:
             return orjson.loads(message)
 
     def __init__(
@@ -103,6 +103,11 @@ class Websocket:
         await asyncio.sleep(self._reconnect_timeout)
         await self.start()
 
+    @property
+    def running(self) -> bool:
+        """Возвращает статус вебсокета."""
+        return self._running
+
     async def _connect(self) -> None:
         """Подключается к вебсокету и настраивает соединение."""
         self._logger.debug(f"Establishing connection with {self._url}")
@@ -114,7 +119,7 @@ class Websocket:
                 # Цикл получения сообщений
                 while self._running:
                     message = await conn.recv()
-                    await self._handle_message(message)
+                    await self._handle_message(message, conn)
 
             except websockets.exceptions.ConnectionClosed as e:
                 self._logger.error(f"Websocket connection was closed unexpectedly: {e}")
@@ -128,7 +133,7 @@ class Websocket:
                 else:
                     return  # Выходим из итератора, если вебсокет уже выключен
 
-    async def _handle_message(self, message: str | bytes) -> None:
+    async def _handle_message(self, message: str | bytes, conn: ClientConnection) -> None:
         """Обрабатывает входящее сообщение вебсокета."""
         try:
             # Обновленяем время последнего сообщения
@@ -136,17 +141,22 @@ class Websocket:
 
             # Ложим сообщение в очередь, предварительно его сериализуя
             decoded_message = self._decoder.decode(message)
-            await self._queue.put(decoded_message)
 
-            # Проверяем размер очереди сообщений и выбрасываем ошибку, если он превышает максимальный размер
-            self._check_queue_size()
+            # Проверяем - вдруг декордер вернул "ping"
+            if decoded_message == "ping":
+                await self._send_pong(conn)
+            else:
+                await self._queue.put(decoded_message)
+
+                # Проверяем размер очереди сообщений и выбрасываем ошибку, если он превышает максимальный размер
+                self._check_queue_size()
         except QueueOverflowError:
             self._logger.error("Message queue is overflow")
         except orjson.JSONDecodeError as e:
             if message in ["ping", "pong"]:
                 self._logger.debug(f"Received ping message: {message}")
             else:
-                self._logger.error(f"Failed to decode JSON message: {message}, error: {e}")
+                self._logger.error(f"Failed to decode message: {message}, error: {e}")
         except Exception as e:
             self._logger.error(f"Unexpected error: {e}")
 
@@ -256,6 +266,18 @@ class Websocket:
                 return
             await asyncio.sleep(1)
 
+    async def _send_pong(self, conn: ClientConnection) -> None:
+        """Отправляет pong сообщение."""
+        if self._pong_message:
+            if isinstance(self._pong_message, Callable):
+                pong_message = self._pong_message()
+            else:
+                pong_message = self._pong_message
+            await conn.send(pong_message)
+        else:
+            await conn.pong()
+        self._logger.debug("Sent pong message")
+
     def __repr__(self) -> str:
         """Репрезентация вебсокета."""
         return f"<Websocket(url={self._url[:15]}...)>"

unicex/binance/adapter.py

@@ -2,7 +2,6 @@ __all__ = ["Adapter"]
 
 
 from unicex.types import (
-    AggTradeDict,
     KlineDict,
     OpenInterestItem,
     TickerDailyDict,
@@ -62,18 +61,6 @@ class Adapter:
         """
         return {item["symbol"]: float(item["price"]) for item in raw_data}
 
-    @staticmethod
-    def futures_last_price(raw_data: list[dict]) -> dict[str, float]:
-        """Преобразует сырой ответ, в котором содержатся данные о тикере за последние 24 часа в унифицированный формат.
-
-        Параметры:
-            raw_data (list[dict]): Сырой ответ с биржи.
-
-        Возвращает:
-            dict[str, float]: Словарь, где ключ - тикер, а значение - последняя цена.
-        """
-        return Adapter.last_price(raw_data)
-
     @staticmethod
     def klines(raw_data: list[list], symbol: str) -> list[KlineDict]:
         """Преобразует сырой ответ, в котором содержатся данные о котировках тикеров в унифицированный формат.
@@ -127,7 +114,11 @@ class Adapter:
         Возвращает:
             OpenInterestItem: Словарь со временем и объемом открытого интереса в монетах.
         """
-        return OpenInterestItem(t=raw_data["time"], v=float(raw_data["openInterest"]))
+        return OpenInterestItem(
+            t=raw_data["time"],
+            v=float(raw_data["openInterest"]),
+            u="coins",
+        )
 
     @staticmethod
     def klines_message(raw_msg: dict) -> list[KlineDict]:
@@ -158,7 +149,7 @@ class Adapter:
         ]
 
     @staticmethod
-    def aggtrades_message(raw_msg: dict) -> list[AggTradeDict]:
+    def aggtrades_message(raw_msg: dict) -> list[TradeDict]:
         """Преобразует сырое сообщение с вебсокета, в котором содержится информация о
         аггрегированных сделке/сделках в унифицированный вид.
 
@@ -170,7 +161,7 @@ class Adapter:
         """
         msg = raw_msg.get("data", raw_msg)
         return [
-            AggTradeDict(
+            TradeDict(
                 t=int(msg["T"]),
                 s=str(msg["s"]),
                 S="SELL" if bool(msg["m"]) else "BUY",

unicex/binance/client.py

@@ -7,7 +7,7 @@ from typing import Any
 
 from unicex._base import BaseClient
 from unicex.exceptions import NotAuthorized
-from unicex.types import RequestMethod
+from unicex.types import NumberLike, RequestMethod
 from unicex.utils import dict_to_query_string, filter_params, generate_hmac_sha256_signature
 
 
@@ -367,13 +367,13 @@ class Client(BaseClient):
         symbol: str,
         side: str,
         type: str,
-        quantity: float | None = None,
-        quote_order_qty: float | None = None,
-        price: float | None = None,
-        stop_price: float | None = None,
+        quantity: NumberLike | None = None,
+        quote_order_qty: NumberLike | None = None,
+        price: NumberLike | None = None,
+        stop_price: NumberLike | None = None,
         time_in_force: str | None = None,
         new_client_order_id: str | None = None,
-        iceberg_qty: float | None = None,
+        iceberg_qty: NumberLike | None = None,
         new_order_resp_type: str | None = None,
         self_trade_prevention_mode: str | None = None,
     ) -> dict:
@@ -405,13 +405,13 @@ class Client(BaseClient):
         symbol: str,
         side: str,
         type: str,
-        quantity: float | None = None,
-        quote_order_qty: float | None = None,
-        price: float | None = None,
-        stop_price: float | None = None,
+        quantity: NumberLike | None = None,
+        quote_order_qty: NumberLike | None = None,
+        price: NumberLike | None = None,
+        stop_price: NumberLike | None = None,
         time_in_force: str | None = None,
         new_client_order_id: str | None = None,
-        iceberg_qty: float | None = None,
+        iceberg_qty: NumberLike | None = None,
         new_order_resp_type: str | None = None,
         self_trade_prevention_mode: str | None = None,
     ) -> dict:
@@ -482,26 +482,26 @@ class Client(BaseClient):
         self,
         symbol: str,
         side: str,
-        quantity: str,
+        quantity: NumberLike,
         list_client_order_id: str | None = None,
         # ABOVE ORDER
         above_type: str = "TAKE_PROFIT_LIMIT",
         above_client_order_id: str | None = None,
-        above_price: str | None = None,
-        above_stop_price: str | None = None,
+        above_price: NumberLike | None = None,
+        above_stop_price: NumberLike | None = None,
         above_trailing_delta: int | None = None,
         above_time_in_force: str | None = None,
-        above_iceberg_qty: str | None = None,
+        above_iceberg_qty: NumberLike | None = None,
         above_strategy_id: int | None = None,
         above_strategy_type: int | None = None,
         # BELOW ORDER
         below_type: str = "STOP_LOSS_LIMIT",
         below_client_order_id: str | None = None,
-        below_price: str | None = None,
-        below_stop_price: str | None = None,
+        below_price: NumberLike | None = None,
+        below_stop_price: NumberLike | None = None,
         below_trailing_delta: int | None = None,
         below_time_in_force: str | None = None,
-        below_iceberg_qty: str | None = None,
+        below_iceberg_qty: NumberLike | None = None,
         below_strategy_id: int | None = None,
         below_strategy_type: int | None = None,
         # EXTRA
@@ -1126,13 +1126,13 @@ class Client(BaseClient):
         symbol: str,
         side: str,
         type: str,
-        quantity: float | None = None,
+        quantity: NumberLike | None = None,
         reduce_only: bool | None = None,
-        price: float | None = None,
+        price: NumberLike | None = None,
         new_client_order_id: str | None = None,
-        stop_price: float | None = None,
+        stop_price: NumberLike | None = None,
         close_position: bool | None = None,
-        activation_price: float | None = None,
+        activation_price: NumberLike | None = None,
         callback_rate: float | None = None,
         time_in_force: str | None = None,
         working_type: str | None = None,
@@ -1176,8 +1176,8 @@ class Client(BaseClient):
         orig_client_order_id: str | None = None,
         symbol: str | None = None,
         side: str | None = None,
-        quantity: float | None = None,
-        price: float | None = None,
+        quantity: NumberLike | None = None,
+        price: NumberLike | None = None,
         price_match: str | None = None,
     ) -> dict:
         """Изменение ордера на фьючерсах.
@@ -1356,7 +1356,7 @@ class Client(BaseClient):
         self,
         symbol: str,
         position_side: str,
-        amount: float,
+        amount: NumberLike,
         type: int,
     ) -> dict:
         """Изменение изолированной маржи позиции.

unicex/binance/uni_websocket_manager.py

@@ -20,16 +20,20 @@ class UniWebsocketManager(IUniWebsocketManager):
     """Реализация менеджера асинхронных унифицированных вебсокетов для биржи Binance."""
 
     def __init__(
-        self, client: Client | UniClient | None = None, logger: LoggerLike | None = None
+        self,
+        client: Client | UniClient | None = None,
+        logger: LoggerLike | None = None,
+        **ws_kwargs: Any,
     ) -> None:
         """Инициализирует унифицированный менеджер вебсокетов.
 
         Параметры:
             client (`Client | UniClient | None`): Клиент Binance или унифицированный клиент. Нужен для подключения к приватным топикам.
             logger (`LoggerLike | None`): Логгер для записи логов.
+            ws_kwargs (`dict[str, Any]`): Дополнительные параметры инициализации, которые будут переданы WebsocketManager/Websocket.
         """
         super().__init__(client=client, logger=logger)
-        self._websocket_manager = WebsocketManager(self._client)  # type: ignore
+        self._websocket_manager = WebsocketManager(self._client, **ws_kwargs)  # type: ignore
         self._adapter = Adapter()
 
     def klines(

unicex/bingx/__init__.py

@@ -0,0 +1,27 @@
+"""Пакет, содержащий реализации клиентов и менеджеров для работы с биржей BingX."""
+
+__all__ = [
+    "Client",
+    "UniClient",
+    "UserWebsocket",
+    "WebsocketManager",
+    "UniWebsocketManager",
+    "ExchangeInfo",
+]
+
+from .client import Client
+from .exchange_info import ExchangeInfo
+from .uni_client import UniClient
+from .uni_websocket_manager import UniWebsocketManager
+from .user_websocket import UserWebsocket
+from .websocket_manager import WebsocketManager
+
+
+async def load_exchange_info() -> None:
+    """Загружает информацию о бирже BingX."""
+    await ExchangeInfo.load_exchange_info()
+
+
+async def start_exchange_info(parse_interval_seconds: int = 60 * 60) -> None:
+    """Запускает процесс обновления информации о бирже BingX."""
+    await ExchangeInfo.start(parse_interval_seconds)