unicex 0.14.9__py3-none-any.whl → 0.15.1__py3-none-any.whl

unicex/__init__.py

@@ -88,6 +88,13 @@ __all__ = [
     "KucoinWebsocketManager",
     "KucoinUserWebsocket",
     "KucoinExchangeInfo",
+    # BingX
+    "BingXClient",
+    "BingXUniClient",
+    "BingXUniWebsocketManager",
+    "BingXWebsocketManager",
+    "BingXUserWebsocket",
+    "BingXExchangeInfo",
 ]
 
 # ruff: noqa
@@ -189,6 +196,15 @@ from .kucoin import (
     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:
     """Единожды загружает информацию о тикерах на всех биржах."""
@@ -201,6 +217,7 @@ async def load_exchanges_info() -> None:
         MexcExchangeInfo.load_exchange_info(),
         OkxExchangeInfo.load_exchange_info(),
         KucoinExchangeInfo.load_exchange_info(),
+        BingXExchangeInfo.load_exchange_info(),
     )
 
 
@@ -215,4 +232,5 @@ async def start_exchanges_info(parse_interval_seconds: int = 60 * 60) -> None:
         MexcExchangeInfo.start(parse_interval_seconds),
         OkxExchangeInfo.start(parse_interval_seconds),
         KucoinExchangeInfo.start(parse_interval_seconds),
+        BingXExchangeInfo.start(parse_interval_seconds),
     )

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__(
@@ -114,7 +114,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 +128,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 +136,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 +261,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

@@ -62,18 +62,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]:
         """Преобразует сырой ответ, в котором содержатся данные о котировках тикеров в унифицированный формат.

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)

unicex/bingx/adapter.py

@@ -0,0 +1,284 @@
+__all__ = ["Adapter"]
+
+from typing import Any
+
+from unicex.types import (
+    KlineDict,
+    LiquidationDict,
+    OpenInterestItem,
+    TickerDailyDict,
+    TradeDict,
+)
+from unicex.utils import catch_adapter_errors, decorate_all_methods
+
+
+@decorate_all_methods(catch_adapter_errors)
+class Adapter:
+    """Адаптер для унификации данных с BingX API."""
+
+    @staticmethod
+    def tickers(raw_data: dict, only_usdt: bool) -> list[str]:
+        """Преобразует сырой ответ, в котором содержатся данные о тикерах в список тикеров.
+
+        Параметры:
+            raw_data (dict): Сырой ответ с биржи.
+            only_usdt (bool): Флаг, указывающий, нужно ли включать только тикеры в паре к USDT.
+
+        Возвращает:
+            list[str]: Список тикеров.
+        """
+        data = raw_data["data"]
+        items = data["symbols"] if isinstance(data, dict) and "symbols" in data else data
+        return [
+            item["symbol"] for item in items if item["symbol"].endswith("USDT") or not only_usdt
+        ]
+
+    @staticmethod
+    def ticker_24hr(raw_data: dict) -> TickerDailyDict:
+        """Преобразует сырой ответ, в котором содержатся данные о тикере за последние 24 часа в унифицированный формат.
+
+        Параметры:
+            raw_data (dict): Сырой ответ с биржи.
+
+        Возвращает:
+            TickerDailyDict: Словарь, где ключ - тикер, а значение - статистика за последние 24 часа.
+        """
+        items = raw_data["data"]
+        result = {}
+        for item in items:
+            percent_raw = item["priceChangePercent"]
+            percent_str = str(percent_raw).strip()
+            if percent_str.endswith("%"):
+                percent_str = percent_str[:-1]
+            result[item["symbol"]] = {
+                "p": float(percent_str),
+                "v": float(item["volume"]),
+                "q": float(item["quoteVolume"]),
+            }
+        return result
+
+    @staticmethod
+    def open_interest(raw_data: dict) -> OpenInterestItem:
+        """Преобразует сырой ответ, в котором содержатся данные об открытом интересе, в унифицированный формат.
+
+        Параметры:
+            raw_data (dict): Сырой ответ с биржи.
+
+        Возвращает:
+            OpenInterestItem: Словарь со временем и объемом открытого интереса в монетах.
+        """
+        item = raw_data["data"]
+        return OpenInterestItem(t=int(item["time"]), v=float(item["openInterest"]))
+
+    @staticmethod
+    def funding_rate(raw_data: dict) -> dict[str, float]:
+        """Преобразует сырой ответ, в котором содержатся данные о ставках финансирования, в унифицированный формат.
+
+        Параметры:
+            raw_data (dict): Сырой ответ с биржи.
+
+        Возвращает:
+            dict[str, float]: Словарь, где ключ - тикер, а значение - ставка финансирования.
+        """
+        data = raw_data["data"]
+        items = data if isinstance(data, list) else [data]
+        result: dict[str, float] = {}
+        for item in items:
+            if "lastFundingRate" in item:
+                result[item["symbol"]] = float(item["lastFundingRate"]) * 100
+            elif "fundingRate" in item and item["fundingRate"] != "":
+                result[item["symbol"]] = float(item["fundingRate"]) * 100
+        return result
+
+    @staticmethod
+    def last_price(raw_data: dict) -> dict[str, float]:
+        """Преобразует сырой ответ, в котором содержатся данные о последней цене, в унифицированный формат.
+
+        Параметры:
+            raw_data (dict): Сырой ответ с биржи.
+
+        Возвращает:
+            dict[str, float]: Словарь, где ключ - тикер, а значение - последняя цена.
+        """
+        data = raw_data["data"]
+        if isinstance(data, list):
+            return {
+                item["symbol"]: float(item["lastPrice"] if "lastPrice" in item else item["price"])
+                for item in data
+            }
+        if isinstance(data, dict):
+            return {data["symbol"]: float(data["price"])}
+        return {}
+
+    @staticmethod
+    def klines(raw_data: dict) -> list[KlineDict]:
+        """Преобразует сырой ответ, в котором содержатся данные о свечах, в унифицированный формат.
+
+        Параметры:
+            raw_data (dict): Сырой ответ с биржи.
+
+        Возвращает:
+            list[KlineDict]: Список словарей, где каждый словарь содержит данные о свече.
+        """
+        data = raw_data["data"]
+        items = data["klines"] if isinstance(data, dict) and "klines" in data else data
+        if "symbol" in raw_data:
+            symbol = raw_data["symbol"]
+        elif isinstance(data, dict) and "symbol" in data:
+            symbol = data["symbol"]
+        else:
+            symbol = ""
+
+        if items and isinstance(items[0], dict):
+            return [
+                KlineDict(
+                    s=symbol,
+                    t=int(kline["time"]),
+                    o=float(kline["open"]),
+                    h=float(kline["high"]),
+                    l=float(kline["low"]),
+                    c=float(kline["close"]),
+                    v=float(kline["volume"]),
+                    q=0.0,
+                    T=None,
+                    x=None,
+                )
+                for kline in sorted(
+                    items,
+                    key=lambda x: int(x["time"]),
+                )
+            ]
+
+        return [
+            KlineDict(
+                s=symbol,
+                t=int(kline[0]),
+                o=float(kline[1]),
+                h=float(kline[2]),
+                l=float(kline[3]),
+                c=float(kline[4]),
+                v=float(kline[5]),
+                q=float(kline[7]) if len(kline) > 7 else 0.0,
+                T=int(kline[6]) if len(kline) > 6 else None,
+                x=None,
+            )
+            for kline in sorted(
+                items,
+                key=lambda x: int(x[0]),
+            )
+        ]
+
+    @staticmethod
+    def Klines_message(msg: Any) -> list[KlineDict]:
+        """Преобразует сырое сообщение с вебсокета, в котором содержится информация о
+        свече/свечах в унифицированный вид.
+
+        Параметры:
+            msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[KlineDict]: Список словарей, где каждый словарь содержит данные о свече.
+        """
+        ...
+
+    @staticmethod
+    def futures_klines_message(msg: Any) -> list[KlineDict]:
+        """Преобразует сырое сообщение с вебсокета, в котором содержится информация о
+        свече/свечах в унифицированный вид.
+
+        Параметры:
+            msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[KlineDict]: Список словарей, где каждый словарь содержит данные о свече.
+        """
+        ...
+
+    @staticmethod
+    def aggtrades_message(msg: Any) -> list[TradeDict]:
+        """Преобразует сырое сообщение с вебсокета, в котором содержится информация о
+        аггрегированных сделке/сделках в унифицированный вид.
+
+        Параметры:
+            msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[KlineDict]: Список словарей, где каждый словарь содержит данные о сделке.
+        """
+        ...
+
+    @staticmethod
+    def futures_aggtrades_message(msg: Any) -> list[TradeDict]:
+        """Преобразует сырое сообщение с вебсокета, в котором содержится информация о
+        аггрегированных сделке/сделках в унифицированный вид.
+
+        Параметры:
+            msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[KlineDict]: Список словарей, где каждый словарь содержит данные о сделке.
+        """
+        ...
+
+    @staticmethod
+    def trades_message(msg: Any) -> list[TradeDict]:
+        """Преобразует сырое сообщение с вебсокета, в котором содержится информация о
+        сделке/сделках в унифицированный вид.
+
+        Параметры:
+            msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[KlineDict]: Список словарей, где каждый словарь содержит данные о сделке.
+        """
+        return [
+            TradeDict(
+                t=int(trade["T"]),
+                s=str(trade["s"]),
+                S="SELL" if bool(trade["m"]) else "BUY",
+                p=float(trade["p"]),
+                v=float(trade["q"]),
+            )
+            for trade in sorted(
+                msg["data"],
+                key=lambda x: int(x["T"]),
+            )
+        ]
+
+    @staticmethod
+    def futures_trades_message(msg: Any) -> list[TradeDict]:
+        """Преобразует сырое сообщение с вебсокета, в котором содержится информация о
+        сделке/сделках в унифицированный вид.
+
+        Параметры:
+            msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[KlineDict]: Список словарей, где каждый словарь содержит данные о сделке.
+        """
+        return [
+            TradeDict(
+                t=int(trade["T"]),
+                s=str(trade["s"]),
+                S="SELL" if bool(trade["m"]) else "BUY",
+                p=float(trade["p"]),
+                v=float(trade["q"]),
+            )
+            for trade in sorted(
+                msg["data"],
+                key=lambda x: int(x["T"]),
+            )
+        ]
+
+    @staticmethod
+    def liquidations_message(msg: Any) -> list[LiquidationDict]:
+        """Преобразует сырое сообщение с вебсокета, в котором содержится информация о
+        ликвидации/ликвидациях в унифицированный вид.
+
+        Параметры:
+            msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[LiquidationDict]: Список словарей, где каждый словарь содержит данные о ликвидации.
+        """
+        ...