unicex 0.13.17__py3-none-any.whl

unicex/bitget/exchange_info.py

@@ -0,0 +1,48 @@
+__all__ = ["ExchangeInfo"]
+
+import aiohttp
+
+from unicex._abc import IExchangeInfo
+from unicex.types import TickerInfoItem
+
+from .client import Client
+
+
+class ExchangeInfo(IExchangeInfo):
+    """Предзагружает информацию о тикерах для биржи Bitget."""
+
+    exchange_name = "Bitget"
+    """Название биржи, на которой работает класс."""
+
+    @classmethod
+    async def _load_spot_exchange_info(cls, session: aiohttp.ClientSession) -> None:
+        """Загружает информацию о бирже для спотового рынка."""
+        exchange_info = await Client(session).get_symbol_info()
+        tickers_info: dict[str, TickerInfoItem] = {}
+        for symbol_info in exchange_info["data"]:
+            tickers_info[symbol_info["symbol"]] = TickerInfoItem(
+                tick_precision=int(symbol_info["pricePrecision"]),
+                tick_step=None,
+                size_precision=int(symbol_info["quantityPrecision"]),
+                size_step=None,
+                contract_size=1,
+            )
+
+        cls._tickers_info = tickers_info
+
+    @classmethod
+    async def _load_futures_exchange_info(cls, session: aiohttp.ClientSession) -> None:
+        """Загружает информацию о бирже для фьючерсного рынка."""
+        tickers_info: dict[str, TickerInfoItem] = {}
+        exchange_info = await Client(session).futures_get_contracts("USDT-FUTURES")
+        for symbol_info in exchange_info["data"]:
+            symbol = symbol_info["symbol"]
+            tickers_info[symbol] = TickerInfoItem(
+                tick_precision=int(symbol_info["pricePlace"]),
+                tick_step=None,
+                size_precision=int(symbol_info["volumePlace"]),
+                size_step=None,
+                contract_size=float(symbol_info["sizeMultiplier"]),
+            )
+
+        cls._futures_tickers_info = tickers_info

unicex/bitget/uni_client.py

@@ -0,0 +1,198 @@
+__all__ = ["UniClient"]
+
+from typing import overload
+
+from unicex._abc import IUniClient
+from unicex.enums import Exchange, MarketType, Timeframe
+from unicex.types import KlineDict, OpenInterestDict, OpenInterestItem, TickerDailyDict
+
+from .adapter import Adapter
+from .client import Client
+
+
+class UniClient(IUniClient[Client]):
+    """Унифицированный клиент для работы с Bitget API."""
+
+    @property
+    def _client_cls(self) -> type[Client]:
+        """Возвращает класс клиента для конкретной биржи.
+
+        Возвращает:
+            `type[Client]`: Класс клиента.
+        """
+        return Client
+
+    async def tickers(self, only_usdt: bool = True) -> list[str]:
+        """Возвращает список тикеров.
+
+        Параметры:
+            only_usdt (`bool`): Если True, возвращает только тикеры в паре к USDT.
+
+        Возвращает:
+            `list[str]`: Список тикеров.
+        """
+        raw_data = await self._client.get_ticker_information()
+        return Adapter.tickers(raw_data=raw_data, only_usdt=only_usdt)
+
+    async def futures_tickers(self, only_usdt: bool = True) -> list[str]:
+        """Возвращает список тикеров.
+
+        Параметры:
+            only_usdt (`bool`): Если True, возвращает только тикеры в паре к USDT.
+
+        Возвращает:
+            `list[str]`: Список тикеров.
+        """
+        raw_data = await self._client.futures_get_all_tickers("USDT-FUTURES")
+        return Adapter.tickers(raw_data=raw_data, only_usdt=only_usdt)
+
+    async def last_price(self) -> dict[str, float]:
+        """Возвращает последнюю цену для каждого тикера.
+
+        Возвращает:
+            `dict[str, float]`: Словарь с последними ценами для каждого тикера.
+        """
+        raw_data = await self._client.get_ticker_information()
+        return Adapter.last_price(raw_data=raw_data)
+
+    async def futures_last_price(self) -> dict[str, float]:
+        """Возвращает последнюю цену для каждого тикера.
+
+        Возвращает:
+            `dict[str, float]`: Словарь с последними ценами для каждого тикера.
+        """
+        raw_data = await self._client.futures_get_all_tickers("USDT-FUTURES")
+        return Adapter.last_price(raw_data=raw_data)
+
+    async def ticker_24hr(self) -> TickerDailyDict:
+        """Возвращает статистику за последние 24 часа для каждого тикера.
+
+        Возвращает:
+            `TickerDailyDict`: Словарь с статистикой за последние 24 часа для каждого тикера.
+        """
+        raw_data = await self._client.get_ticker_information()
+        return Adapter.ticker_24hr(raw_data=raw_data)
+
+    async def futures_ticker_24hr(self) -> TickerDailyDict:
+        """Возвращает статистику за последние 24 часа для каждого тикера.
+
+        Возвращает:
+            `TickerDailyDict`: Словарь с статистикой за последние 24 часа для каждого тикера.
+        """
+        raw_data = await self._client.futures_get_all_tickers("USDT-FUTURES")
+        return Adapter.ticker_24hr(raw_data=raw_data)
+
+    async def klines(
+        self,
+        symbol: str,
+        interval: Timeframe | str,
+        limit: int | None = None,
+        start_time: int | None = None,
+        end_time: int | None = None,
+    ) -> list[KlineDict]:
+        """Возвращает список свечей.
+
+        Параметры:
+            symbol (`str`): Название тикера.
+            interval (`Timeframe`): Таймфрейм свечей.
+            limit (`int`): Количество свечей.
+            start_time (`int`): Время начала периода в миллисекундах.
+            end_time (`int`): Время окончания периода в миллисекундах.
+
+        Возвращает:
+            `list[KlineDict]`: Список свечей.
+        """
+        interval = (
+            interval.to_exchange_format(Exchange.BITGET, MarketType.SPOT)
+            if isinstance(interval, Timeframe)
+            else interval
+        )
+        raw_data = await self._client.get_candlestick_data(
+            symbol=symbol,
+            granularity=interval,
+            limit=limit,
+            start_time=start_time,
+            end_time=end_time,
+        )
+        return Adapter.klines(raw_data=raw_data, symbol=symbol)
+
+    async def futures_klines(
+        self,
+        symbol: str,
+        interval: Timeframe | str,
+        limit: int | None = None,
+        start_time: int | None = None,
+        end_time: int | None = None,
+    ) -> list[KlineDict]:
+        """Возвращает список свечей.
+
+        Параметры:
+            symbol (`str`): Название тикера.
+            interval (`Timeframe`): Таймфрейм свечей.
+            limit (`int`): Количество свечей.
+            start_time (`int`): Время начала периода в миллисекундах.
+            end_time (`int`): Время окончания периода в миллисекундах.
+
+        Возвращает:
+            `list[KlineDict]`: Список свечей.
+        """
+        interval = (
+            interval.to_exchange_format(Exchange.BITGET, MarketType.FUTURES)
+            if isinstance(interval, Timeframe)
+            else interval
+        )
+        raw_data = await self._client.futures_get_candlestick_data(
+            symbol=symbol,
+            product_type="USDT-FUTURES",
+            granularity=interval,
+            limit=limit,
+            start_time=start_time,
+            end_time=end_time,
+        )
+        return Adapter.klines(raw_data=raw_data, symbol=symbol)
+
+    @overload
+    async def funding_rate(self, symbol: str) -> float: ...
+
+    @overload
+    async def funding_rate(self, symbol: None) -> dict[str, float]: ...
+
+    @overload
+    async def funding_rate(self) -> dict[str, float]: ...
+
+    async def funding_rate(self, symbol: str | None = None) -> dict[str, float] | float:
+        """Возвращает ставку финансирования для тикера или всех тикеров, если тикер не указан.
+
+        - Параметры:
+        symbol (`str | None`): Название тикера (Опционально).
+
+        Возвращает:
+          `dict[str, float] | float`: Ставка финансирования для тикера или словарь со ставками для всех тикеров.
+        """
+        raw_data = await self._client.futures_get_all_tickers("USDT-FUTURES")
+        adapted_data = Adapter.funding_rate(raw_data)
+        return adapted_data[symbol] if symbol else adapted_data
+
+    @overload
+    async def open_interest(self, symbol: str) -> OpenInterestItem: ...
+
+    @overload
+    async def open_interest(self, symbol: None) -> OpenInterestDict: ...
+
+    @overload
+    async def open_interest(self) -> OpenInterestDict: ...
+
+    async def open_interest(self, symbol: str | None = None) -> OpenInterestItem | OpenInterestDict:
+        """Возвращает объем открытого интереса для тикера или всех тикеров, если тикер не указан.
+
+        Параметры:
+            symbol (`str | None`): Название тикера. (Опционально, но обязателен для следующих бирж: BINANCE).
+
+        Возвращает:
+            `OpenInterestItem | OpenInterestDict`: Если тикер передан - словарь со временем и объемом
+                открытого интереса в монетах. Если нет передан - то словарь, в котором ключ - тикер,
+                а значение - словарь с временем и объемом открытого интереса в монетах.
+        """
+        raw_data = await self._client.futures_get_all_tickers("USDT-FUTURES")
+        adapted_data = Adapter.open_interest(raw_data)
+        return adapted_data[symbol] if symbol else adapted_data

unicex/bitget/uni_websocket_manager.py

@@ -0,0 +1,275 @@
+__all__ = ["IUniWebsocketManager"]
+
+from collections.abc import Awaitable, Callable, Sequence
+from typing import Any, overload
+
+from unicex._abc import IUniWebsocketManager
+from unicex._base import Websocket
+from unicex.enums import Timeframe
+from unicex.types import LoggerLike
+
+from .adapter import Adapter
+from .client import Client
+from .uni_client import UniClient
+from .websocket_manager import WebsocketManager
+
+type CallbackType = Callable[[Any], Awaitable[None]]
+
+
+class UniWebsocketManager(IUniWebsocketManager):
+    """Реализация менеджера асинхронных унифицированных вебсокетов для биржи Bitget."""
+
+    def __init__(
+        self, client: Client | UniClient | None = None, logger: LoggerLike | None = None
+    ) -> None:
+        """Инициализирует унифицированный менеджер вебсокетов.
+
+        Параметры:
+            client (`Client | UniClient | None`): Клиент Bitget или унифицированный клиент. Нужен для подключения к приватным топикам.
+            logger (`LoggerLike | None`): Логгер для записи логов.
+        """
+        super().__init__(client=client, logger=logger)
+        self._websocket_manager = WebsocketManager(self._client)  # type: ignore
+        self._adapter = Adapter()
+
+    @overload
+    def klines(
+        self,
+        callback: CallbackType,
+        timeframe: Timeframe,
+        *,
+        symbol: str,
+        symbols: None = None,
+    ) -> Websocket: ...
+
+    @overload
+    def klines(
+        self,
+        callback: CallbackType,
+        timeframe: Timeframe,
+        *,
+        symbol: None = None,
+        symbols: Sequence[str],
+    ) -> Websocket: ...
+
+    def klines(
+        self,
+        callback: CallbackType,
+        timeframe: Timeframe,
+        symbol: str | None = None,
+        symbols: Sequence[str] | None = None,
+    ) -> Websocket:
+        """Открывает стрим свечей (spot) с унификацией сообщений.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            timeframe (`Timeframe`): Временной интервал свечей.
+            symbol (`str | None`): Один символ для подписки.
+            symbols (`Sequence[str] | None`): Список символов для мультиплекс‑подключения.
+
+        Должен быть указан либо `symbol`, либо `symbols`.
+
+        Возвращает:
+            `Websocket`: Экземпляр вебсокета для управления соединением.
+        """
+        raise NotImplementedError()
+
+    @overload
+    def futures_klines(
+        self,
+        callback: CallbackType,
+        timeframe: Timeframe,
+        *,
+        symbol: str,
+        symbols: None = None,
+    ) -> Websocket: ...
+
+    @overload
+    def futures_klines(
+        self,
+        callback: CallbackType,
+        timeframe: Timeframe,
+        *,
+        symbol: None = None,
+        symbols: Sequence[str],
+    ) -> Websocket: ...
+
+    def futures_klines(
+        self,
+        callback: CallbackType,
+        timeframe: Timeframe,
+        symbol: str | None = None,
+        symbols: Sequence[str] | None = None,
+    ) -> Websocket:
+        """Открывает стрим свечей (futures) с унификацией сообщений.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            timeframe (`Timeframe`): Временной интервал свечей.
+            symbol (`str | None`): Один символ для подписки.
+            symbols (`Sequence[str] | None`): Список символов для мультиплекс‑подключения.
+
+        Должен быть указан либо `symbol`, либо `symbols`.
+
+        Возвращает:
+            `Websocket`: Экземпляр вебсокета.
+        """
+        raise NotImplementedError()
+
+    @overload
+    def trades(
+        self,
+        callback: CallbackType,
+        *,
+        symbol: str,
+        symbols: None = None,
+    ) -> Websocket: ...
+
+    @overload
+    def trades(
+        self,
+        callback: CallbackType,
+        *,
+        symbol: None = None,
+        symbols: Sequence[str],
+    ) -> Websocket: ...
+
+    def trades(
+        self,
+        callback: CallbackType,
+        symbol: str | None = None,
+        symbols: Sequence[str] | None = None,
+    ) -> Websocket:
+        """Открывает стрим сделок (spot) с унификацией сообщений.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            symbol (`str | None`): Один символ для подписки.
+            symbols (`Sequence[str] | None`): Список символов для мультиплекс‑подключения.
+
+        Должен быть указан либо `symbol`, либо `symbols`.
+
+        Возвращает:
+            `Websocket`: Экземпляр вебсокета.
+        """
+        wrapper = self._make_wrapper(self._adapter.trades_message, callback)
+        return self._websocket_manager.trade(
+            callback=wrapper, symbol=symbol, symbols=symbols, market_type="SPOT"
+        )
+
+    @overload
+    def aggtrades(
+        self,
+        callback: CallbackType,
+        *,
+        symbol: str,
+        symbols: None = None,
+    ) -> Websocket: ...
+
+    @overload
+    def aggtrades(
+        self,
+        callback: CallbackType,
+        *,
+        symbol: None = None,
+        symbols: Sequence[str],
+    ) -> Websocket: ...
+
+    def aggtrades(
+        self,
+        callback: CallbackType,
+        symbol: str | None = None,
+        symbols: Sequence[str] | None = None,
+    ) -> Websocket:
+        """Открывает стрим агрегированных сделок (spot) с унификацией сообщений.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            symbol (`str | None`): Один символ для подписки.
+            symbols (`Sequence[str] | None`): Список символов для мультиплекс‑подключения.
+
+        Должен быть указан либо `symbol`, либо `symbols`.
+
+        Возвращает:
+            `Websocket`: Экземпляр вебсокета.
+        """
+        return self.trades(callback=callback, symbol=symbol, symbols=symbols)  # type: ignore[reportCallIssue]
+
+    @overload
+    def futures_trades(
+        self,
+        callback: CallbackType,
+        *,
+        symbol: str,
+        symbols: None = None,
+    ) -> Websocket: ...
+
+    @overload
+    def futures_trades(
+        self,
+        callback: CallbackType,
+        *,
+        symbol: None = None,
+        symbols: Sequence[str],
+    ) -> Websocket: ...
+
+    def futures_trades(
+        self,
+        callback: CallbackType,
+        symbol: str | None = None,
+        symbols: Sequence[str] | None = None,
+    ) -> Websocket:
+        """Открывает стрим сделок (futures) с унификацией сообщений.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            symbol (`str | None`): Один символ для подписки.
+            symbols (`Sequence[str] | None`): Список символов для мультиплекс‑подключения.
+
+        Должен быть указан либо `symbol`, либо `symbols`.
+
+        Возвращает:
+            `Websocket`: Экземпляр вебсокета.
+        """
+        wrapper = self._make_wrapper(self._adapter.trades_message, callback)
+        return self._websocket_manager.trade(
+            callback=wrapper, symbol=symbol, symbols=symbols, market_type="USDT-FUTURES"
+        )
+
+    @overload
+    def futures_aggtrades(
+        self,
+        callback: CallbackType,
+        *,
+        symbol: str,
+        symbols: None = None,
+    ) -> Websocket: ...
+
+    @overload
+    def futures_aggtrades(
+        self,
+        callback: CallbackType,
+        *,
+        symbol: None = None,
+        symbols: Sequence[str],
+    ) -> Websocket: ...
+
+    def futures_aggtrades(
+        self,
+        callback: CallbackType,
+        symbol: str | None = None,
+        symbols: Sequence[str] | None = None,
+    ) -> Websocket:
+        """Открывает стрим агрегированных сделок (futures) с унификацией сообщений.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            symbol (`str | None`): Один символ для подписки.
+            symbols (`Sequence[str] | None`): Список символов для мультиплекс‑подключения.
+
+        Должен быть указан либо `symbol`, либо `symbols`.
+
+        Возвращает:
+            `Websocket`: Экземпляр вебсокета.
+        """
+        raise NotImplementedError()

unicex/bitget/user_websocket.py

@@ -0,0 +1,7 @@
+__all__ = ["UserWebsocket"]
+
+
+class UserWebsocket:
+    """Пользовательский вебсокет Bitget."""
+
+    pass