unicex 0.16.7__py3-none-any.whl → 0.17.1__py3-none-any.whl

unicex/aster/exchange_info.py

@@ -0,0 +1,44 @@
+__all__ = ["ExchangeInfo"]
+
+import aiohttp
+
+from unicex._abc import IExchangeInfo
+from unicex.types import TickerInfoItem
+
+from .client import Client
+
+
+class ExchangeInfo(IExchangeInfo):
+    """Предзагружает информацию о тикерах для биржи Aster."""
+
+    exchange_name = "Aster"
+    """Название биржи, на которой работает класс."""
+
+    @classmethod
+    async def _load_spot_exchange_info(cls, session: aiohttp.ClientSession) -> None:
+        """Загружает информацию о бирже для спотового рынка."""
+        # В документации Aster нет спотового API, поэтому не загружаем данные.
+        cls._tickers_info = {}
+
+    @classmethod
+    async def _load_futures_exchange_info(cls, session: aiohttp.ClientSession) -> None:
+        """Загружает информацию о бирже для фьючерсного рынка."""
+        exchange_info = await Client(session).futures_exchange_info()
+        tickers_info: dict[str, TickerInfoItem] = {}
+        for symbol_info in exchange_info.get("symbols", []):
+            filters = {
+                flt["filterType"]: flt
+                for flt in symbol_info.get("filters", [])
+                if "filterType" in flt
+            }
+            price_filter = filters["PRICE_FILTER"]
+            lot_size_filter = filters["LOT_SIZE"]
+            tickers_info[symbol_info["symbol"]] = TickerInfoItem(
+                tick_step=float(price_filter["tickSize"]),
+                tick_precision=None,
+                size_step=float(lot_size_filter["stepSize"]),
+                size_precision=None,
+                contract_size=1,
+            )
+
+        cls._futures_tickers_info = tickers_info

unicex/aster/uni_client.py

@@ -0,0 +1,184 @@
+__all__ = ["UniClient"]
+
+
+from typing import overload
+
+from unicex._abc import IUniClient
+from unicex.enums import Exchange, Timeframe
+from unicex.exceptions import NotSupported
+from unicex.types import KlineDict, OpenInterestDict, OpenInterestItem, TickerDailyDict
+
+from .adapter import Adapter
+from .client import Client
+
+
+class UniClient(IUniClient[Client]):
+    """Унифицированный клиент для работы с Aster API."""
+
+    @property
+    def _client_cls(self) -> type[Client]:
+        """Возвращает класс клиента для Aster.
+
+        Возвращает:
+            type[Client]: Класс клиента для Aster.
+        """
+        return Client
+
+    async def tickers(self, only_usdt: bool = True) -> list[str]:
+        """Возвращает список тикеров.
+
+        Параметры:
+            only_usdt (bool): Если True, возвращает только тикеры в паре к USDT.
+
+        Возвращает:
+            list[str]: Список тикеров.
+        """
+        raise NotSupported("Spot market data is not supported for Aster")
+
+    async def futures_tickers(self, only_usdt: bool = True) -> list[str]:
+        """Возвращает список тикеров.
+
+        Параметры:
+            only_usdt (bool): Если True, возвращает только тикеры в паре к USDT.
+
+        Возвращает:
+            list[str]: Список тикеров.
+        """
+        raw_data = await self._client.futures_ticker_price()
+        return Adapter.tickers(raw_data=raw_data, only_usdt=only_usdt)  # type: ignore[arg-type] | raw_data is list[dict] if symbol param is not ommited
+
+    async def last_price(self) -> dict[str, float]:
+        """Возвращает последнюю цену для каждого тикера.
+
+        Возвращает:
+            dict[str, float]: Словарь с последними ценами для каждого тикера.
+        """
+        raise NotSupported("Spot market data is not supported for Aster")
+
+    async def futures_last_price(self) -> dict[str, float]:
+        """Возвращает последнюю цену для каждого тикера.
+
+        Возвращает:
+            dict[str, float]: Словарь с последними ценами для каждого тикера.
+        """
+        raw_data = await self._client.futures_ticker_price()
+        return Adapter.last_price(raw_data)  # type: ignore[arg-type] | raw_data is list[dict] if symbol param is not ommited
+
+    async def ticker_24hr(self) -> TickerDailyDict:
+        """Возвращает статистику за последние 24 часа для каждого тикера.
+
+        Возвращает:
+            TickerDailyDict: Словарь с статистикой за последние 24 часа для каждого тикера.
+        """
+        raise NotSupported("Spot market data is not supported for Aster")
+
+    async def futures_ticker_24hr(self) -> TickerDailyDict:
+        """Возвращает статистику за последние 24 часа для каждого тикера.
+
+        Возвращает:
+            TickerDailyDict: Словарь с статистикой за последние 24 часа для каждого тикера.
+        """
+        raw_data = await self._client.futures_ticker_24hr()
+        return Adapter.ticker_24hr(raw_data=raw_data)  # type: ignore[arg-type] | raw_data is list[dict] if symbol param is not ommited
+
+    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): Название тикера.
+            limit (int | None): Количество свечей.
+            interval (Timeframe | str): Таймфрейм свечей.
+            start_time (int | None): Время начала периода в миллисекундах.
+            end_time (int | None): Время окончания периода в миллисекундах.
+
+        Возвращает:
+            list[KlineDict]: Список свечей для тикера.
+        """
+        raise NotSupported("Spot market data is not supported for Aster")
+
+    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): Название тикера.
+            limit (int | None): Количество свечей.
+            interval (Timeframe | str): Таймфрейм свечей.
+            start_time (int | None): Время начала периода в миллисекундах.
+            end_time (int | None): Время окончания периода в миллисекундах.
+
+        Возвращает:
+            list[KlineDict]: Список свечей для тикера.
+        """
+        interval = (
+            interval.to_exchange_format(Exchange.ASTER)
+            if isinstance(interval, Timeframe)
+            else interval
+        )
+        raw_data = await self._client.futures_klines(
+            symbol=symbol,
+            interval=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]: Ставка финансирования для каждого тикера.
+        """
+        raw_data = await self._client.futures_mark_price()
+        adapted_data = Adapter.funding_rate(raw_data if isinstance(raw_data, list) else [raw_data])  # type: ignore[arg-type]
+        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`: Если тикер передан - словарь со временем и объемом
+                открытого интереса в USDT. Если нет передан - то словарь, в котором ключ - тикер,
+                а значение - словарь с временем и объемом открытого интереса в USDT.
+        """
+        raw_data = await self._client.open_interest()
+        adapted_data = Adapter.open_interest(raw_data)
+        return adapted_data[symbol] if symbol else adapted_data

unicex/aster/uni_websocket_manager.py

@@ -0,0 +1,286 @@
+__all__ = ["UniWebsocketManager"]
+
+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 Exchange, Timeframe
+from unicex.exceptions import NotSupported
+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):
+    """Реализация менеджера асинхронных унифицированных вебсокетов."""
+
+    def __init__(
+        self,
+        client: Client | UniClient | None = None,
+        logger: LoggerLike | None = None,
+        **ws_kwargs: Any,
+    ) -> None:
+        """Инициализирует унифицированный менеджер вебсокетов.
+
+        Параметры:
+            client (`Client | UniClient | None`): Клиент Aster или унифицированный клиент. Нужен для подключения к приватным топикам.
+            logger (`LoggerLike | None`): Логгер для записи логов.
+            ws_kwargs (`dict[str, Any]`): Дополнительные параметры инициализации, которые будут переданы WebsocketManager/Websocket.
+        """
+        super().__init__(client=client, logger=logger)
+        self._websocket_manager = WebsocketManager(self._client, **ws_kwargs)  # 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 NotSupported("Spot market data is not supported for Aster")
+
+    @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`: Экземпляр вебсокета.
+        """
+        wrapper = self._make_wrapper(self._adapter.Klines_message, callback)
+        return self._websocket_manager.futures_klines(
+            callback=wrapper,
+            symbol=symbol,
+            symbols=symbols,
+            interval=timeframe.to_exchange_format(Exchange.ASTER),
+        )
+
+    @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`: Экземпляр вебсокета.
+        """
+        raise NotSupported("Spot market data is not supported for Aster")
+
+    @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`: Экземпляр вебсокета.
+        """
+        raise NotSupported("Spot market data is not supported for Aster")
+
+    @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`.
+
+        В Aster доступен только поток агрегированных сделок, поэтому `futures_aggtrades`
+        будет использован автоматически.
+
+        Возвращает:
+            `Websocket`: Экземпляр вебсокета.
+        """
+        return self.futures_aggtrades(callback, symbol, symbols)  # type: ignore
+
+    @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`: Экземпляр вебсокета.
+        """
+        wrapper = self._make_wrapper(self._adapter.trades_message, callback)
+        return self._websocket_manager.futures_agg_trade(
+            callback=wrapper, symbol=symbol, symbols=symbols
+        )