unicex 0.13.17__py3-none-any.whl

unicex/bitget/websocket_manager.py

@@ -0,0 +1,232 @@
+__all__ = ["WebsocketManager"]
+
+
+import json
+from collections.abc import Awaitable, Callable, Sequence
+from typing import Any, Literal
+
+from unicex._base import Websocket
+
+from .client import Client
+
+type CallbackType = Callable[[Any], Awaitable[None]]
+
+
+class WebsocketManager:
+    """Менеджер асинхронных вебсокетов для Bitget."""
+
+    _URL: str = "wss://ws.bitget.com/v2/ws/public"
+    """Базовый URL для вебсокета."""
+
+    def __init__(self, client: Client | None = None, **ws_kwargs: Any) -> None:
+        """Инициализирует менеджер вебсокетов для Bitget.
+
+        Параметры:
+            client (`Client | None`): Клиент для выполнения запросов. Нужен, чтобы открыть приватные вебсокеты.
+            ws_kwargs (`dict[str, Any]`): Дополнительные аргументы, котоыре прокидываются в `Websocket`.
+        """
+        self.client = client
+        self._ws_kwargs = {"ping_message": "ping", **ws_kwargs}
+
+    def _generate_subscription_message(
+        self,
+        topic: str,
+        market_type: Literal["SPOT", "USDT-FUTURES"],
+        symbol: str | None = None,
+        symbols: Sequence[str] | None = None,
+    ) -> list[str]:
+        """Сформировать сообщение для подписки на вебсокет.
+
+        Параметры:
+            topic (`str`): Канал подписки (например: "ticker", "candle1m", "depth").
+            market_type (`"SPOT" | "USDT-FUTURES"`): Тип рынка для подписки.
+            symbol (`str | None`): Торговая пара. Нельзя использовать одновременно с `symbols`.
+            symbols (`Sequence[str] | None`): Список торговых пар. Нельзя использовать одновременно с `symbol`.
+
+        Возвращает:
+            `str`: JSON-строка с сообщением для подписки на вебсокет.
+        """
+        if symbol and symbols:
+            raise ValueError("Parameters symbol and symbols cannot be used together")
+        if not (symbol or symbols):
+            raise ValueError("Either symbol or symbols must be provided")
+
+        tickers = [symbol] if symbol else symbols
+        streams: list[dict] = [
+            {
+                "instType": market_type,
+                "channel": topic,
+                "instId": ticker.upper(),
+            }
+            for ticker in tickers  # type: ignore
+        ]
+
+        return [json.dumps({"op": "subscribe", "args": streams})]
+
+    def trade(
+        self,
+        callback: CallbackType,
+        market_type: Literal["SPOT", "USDT-FUTURES"],
+        symbol: str | None = None,
+        symbols: Sequence[str] | None = None,
+    ) -> Websocket:
+        """Создает вебсокет для получения сделок.
+
+        https://www.bitget.com/api-doc/spot/websocket/public/Trades-Channel
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            market_type (`Literal["SPOT", "USDT-FUTURES"]`): Тип рынка.
+            symbol (`str | None`): Один символ для подписки.
+            symbols (`Sequence[str] | None`): Список символов для мультиплекс‑подключения.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        subsription_messages = self._generate_subscription_message(
+            topic="trade",
+            market_type=market_type,
+            symbol=symbol,
+            symbols=symbols,
+        )
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=subsription_messages,
+            **self._ws_kwargs,
+        )
+
+    def ticker(
+        self,
+        callback: CallbackType,
+        market_type: Literal["SPOT", "USDT-FUTURES"],
+        symbol: str | None = None,
+        symbols: Sequence[str] | None = None,
+    ) -> Websocket:
+        """Создает вебсокет для получения тикеров.
+
+        https://www.bitget.com/api-doc/spot/websocket/public/Tickers-Channel
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            market_type (`Literal["SPOT", "USDT-FUTURES"]`): Тип рынка.
+            symbol (`str | None`): Один символ для подписки.
+            symbols (`Sequence[str] | None`): Список символов для мультиплекс‑подключения.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        subscription_messages = self._generate_subscription_message(
+            topic="ticker",
+            market_type=market_type,
+            symbol=symbol,
+            symbols=symbols,
+        )
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=subscription_messages,
+            **self._ws_kwargs,
+        )
+
+    def candlestick(
+        self,
+        callback: CallbackType,
+        market_type: Literal["SPOT", "USDT-FUTURES"],
+        interval: str,
+        symbol: str | None = None,
+        symbols: Sequence[str] | None = None,
+    ) -> Websocket:
+        """Создает вебсокет для получения данных свечей (candlestick).
+
+        https://www.bitget.com/api-doc/spot/websocket/public/Candlesticks-Channel
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            market_type (`Literal["SPOT", "USDT-FUTURES"]`): Тип рынка.
+            interval (`str`): Интервал свечи, например "1m", "1H", "1Dutc".
+            symbol (`str | None`): Один символ для подписки.
+            symbols (`Sequence[str] | None`): Список символов для мультиплекс‑подключения.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        topic = f"candle{interval}"
+        subscription_messages = self._generate_subscription_message(
+            topic=topic,
+            market_type=market_type,
+            symbol=symbol,
+            symbols=symbols,
+        )
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=subscription_messages,
+            **self._ws_kwargs,
+        )
+
+    def depth(
+        self,
+        callback: CallbackType,
+        market_type: Literal["SPOT", "USDT-FUTURES"],
+        depth_type: str = "books",
+        symbol: str | None = None,
+        symbols: Sequence[str] | None = None,
+    ) -> Websocket:
+        """Создает вебсокет для получения данных глубины рынка (order book).
+
+        https://www.bitget.com/api-doc/spot/websocket/public/Depth-Channel
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            market_type (`Literal["SPOT", "USDT-FUTURES"]`): Тип рынка.
+            depth_type (`str`): Тип глубины: "books", "books1", "books5", "books15".
+            symbol (`str | None`): Один символ для подписки.
+            symbols (`Sequence[str] | None`): Список символов для мультиплекс‑подключения.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        subscription_messages = self._generate_subscription_message(
+            topic=depth_type,
+            market_type=market_type,
+            symbol=symbol,
+            symbols=symbols,
+        )
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=subscription_messages,
+            **self._ws_kwargs,
+        )
+
+    def auction(
+        self,
+        callback: CallbackType,
+        symbol: str | None = None,
+        symbols: Sequence[str] | None = None,
+    ) -> Websocket:
+        """Создает вебсокет для получения информации о Call Auction.
+
+        https://www.bitget.com/api-doc/spot/websocket/public/Auction-Channel
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            symbol (`str | None`): Один символ для подписки.
+            symbols (`Sequence[str] | None`): Список символов для мультиплекс‑подключения.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        subscription_messages = self._generate_subscription_message(
+            topic="auction",
+            market_type="SPOT",
+            symbol=symbol,
+            symbols=symbols,
+        )
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=subscription_messages,
+            **self._ws_kwargs,
+        )

unicex/bybit/__init__.py

@@ -0,0 +1,27 @@
+"""Пакет, содержащий реализации клиентов и менеджеров для работы с биржей Bybit."""
+
+__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:
+    """Загружает информацию о бирже Bybit."""
+    await ExchangeInfo.load_exchange_info()
+
+
+async def start_exchange_info(parse_interval_seconds: int = 60 * 60) -> None:
+    """Запускает процесс обновления информации о бирже Bybit."""
+    await ExchangeInfo.start(parse_interval_seconds)

unicex/bybit/adapter.py

@@ -0,0 +1,208 @@
+__all__ = ["Adapter"]
+
+from typing import Any
+
+from unicex.types import (
+    KlineDict,
+    LiquidationDict,
+    OpenInterestDict,
+    OpenInterestItem,
+    TickerDailyDict,
+    TickerDailyItem,
+    TradeDict,
+)
+from unicex.utils import catch_adapter_errors, decorate_all_methods
+
+
+@decorate_all_methods(catch_adapter_errors)
+class Adapter:
+    """Адаптер для унификации данных с Bybit API."""
+
+    @staticmethod
+    def tickers(raw_data: dict, only_usdt: bool) -> list[str]:
+        """Преобразует сырой ответ, в котором содержатся данные о тикерах в список тикеров.
+
+        Параметры:
+            raw_data (dict): Сырой ответ с биржи.
+            only_usdt (bool): Флаг, указывающий, нужно ли включать только тикеры в паре к USDT.
+
+        Возвращает:
+            list[str]: Список тикеров.
+        """
+        return [
+            item["symbol"]
+            for item in raw_data["result"]["list"]
+            if item["symbol"].endswith("USDT") or not only_usdt
+        ]
+
+    @staticmethod
+    def ticker_24hr(raw_data: dict) -> TickerDailyDict:
+        """Преобразует сырой ответ, в котором содержатся данные о тикере за последние 24 часа в унифицированный формат.
+
+        Параметры:
+            raw_data (dict): Сырой ответ с биржи.
+
+        Возвращает:
+            TickerDailyDict: Словарь, где ключ - тикер, а значение - статистика за последние 24 часа.
+        """
+        return {
+            item["symbol"]: TickerDailyItem(
+                p=round(float(item["price24hPcnt"]) * 100, 2),
+                v=float(item["volume24h"]),
+                q=float(item["turnover24h"]),
+            )
+            for item in raw_data["result"]["list"]
+        }
+
+    @staticmethod
+    def open_interest(raw_data: dict) -> OpenInterestDict:
+        """Преобразует сырой ответ, в котором содержатся данные об открытом интересе, в унифицированный формат.
+
+        Параметры:
+            raw_data (dict): Сырой ответ с биржи.
+
+        Возвращает:
+            OpenInterestDict: Словарь, где ключ - тикер, а значение - агрегированные данные открытого интереса.
+        """
+        return {
+            item["symbol"]: OpenInterestItem(
+                t=raw_data["time"],
+                v=float(item["openInterest"]),
+            )
+            for item in raw_data["result"]["list"]
+        }
+
+    @staticmethod
+    def funding_rate(raw_data: dict) -> dict[str, float]:
+        """Преобразует сырой ответ, в котором содержатся данные о ставках финансирования, в унифицированный формат.
+
+        Параметры:
+            raw_data (dict): Сырой ответ с биржи.
+
+        Возвращает:
+            dict[str, float]: Словарь, где ключ - тикер, а значение - ставка финансирования.
+        """
+        return {
+            item["symbol"]: float(item["fundingRate"]) * 100
+            for item in raw_data["result"]["list"]
+            if item["fundingRate"]
+        }
+
+    @staticmethod
+    def last_price(raw_data: dict) -> dict[str, float]:
+        """Преобразует сырой ответ, в котором содержатся данные о последней цене, в унифицированный формат.
+
+        Параметры:
+            raw_data (dict): Сырой ответ с биржи.
+
+        Возвращает:
+            dict[str, float]: Словарь, где ключ - тикер, а значение - последняя цена.
+        """
+        return {item["symbol"]: float(item["lastPrice"]) for item in raw_data["result"]["list"]}
+
+    @staticmethod
+    def klines(raw_data: dict) -> list[KlineDict]:
+        """Преобразует сырой ответ, в котором содержатся данные о свечах, в унифицированный формат.
+
+        Параметры:
+            raw_data (dict): Сырой ответ с биржи.
+
+        Возвращает:
+            list[KlineDict]: Список словарей, где каждый словарь содержит данные о свече.
+        """
+        return [
+            KlineDict(
+                s=raw_data["result"]["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[6]),
+                T=None,
+                x=None,
+            )
+            for kline in sorted(
+                raw_data["result"]["list"],
+                key=lambda x: int(x[0]),
+            )
+        ]
+
+    @staticmethod
+    def Klines_message(raw_msg: Any) -> list[KlineDict]:
+        """Преобразует вебсокет-сообщение с данными о свечах в унифицированный формат.
+
+        Параметры:
+        raw_msg (`Any`): Сырое сообщение из вебсокета Bybit.
+
+        Возвращает:
+          `list[KlineDict]`: Список свечей в унифицированном формате.
+        """
+        symbol = raw_msg["topic"].split(".")[-1]
+        return [
+            KlineDict(
+                s=symbol,
+                t=kline["start"],
+                o=float(kline["open"]),
+                h=float(kline["high"]),
+                l=float(kline["low"]),
+                c=float(kline["close"]),
+                v=float(kline["volume"]),
+                q=float(kline["turnover"]),
+                T=kline["end"],
+                x=kline["confirm"],
+            )
+            for kline in sorted(
+                raw_msg["data"],
+                key=lambda x: int(x["start"]),
+            )
+        ]
+
+    @staticmethod
+    def trades_message(raw_msg: Any) -> list[TradeDict]:
+        """Преобразует вебсокет-сообщение с данными о сделках в унифицированный формат.
+
+        Параметры:
+        raw_msg (`Any`): Сырое сообщение из вебсокета Bybit.
+
+        Возвращает:
+          `list[TradeDict]`: Список сделок в унифицированном формате.
+        """
+        return [
+            TradeDict(
+                t=trade["T"],
+                s=trade["s"],
+                S=trade["S"].upper(),
+                p=float(trade["p"]),
+                v=float(trade["v"]),
+            )
+            for trade in sorted(
+                raw_msg["data"],
+                key=lambda x: int(x["T"]),
+            )
+        ]
+
+    @staticmethod
+    def liquidations_message(raw_msg: Any) -> list[LiquidationDict]:
+        """Преобразует вебсокет-сообщение с данными о ликвидациях в унифицированный формат.
+
+        Параметры:
+        raw_msg (`Any`): Сырое сообщение из вебсокета Bybit.
+
+        Возвращает:
+          `list[LiquidationDict]`: Список ликвидаций в унифицированном формате.
+        """
+        return [
+            LiquidationDict(
+                t=liquidation["T"],
+                s=liquidation["s"],
+                S=liquidation["S"].upper(),
+                v=float(liquidation["v"]),
+                p=float(liquidation["p"]),
+            )
+            for liquidation in sorted(
+                raw_msg["data"],
+                key=lambda x: int(x["T"]),
+            )
+        ]