unicex 0.13.17__py3-none-any.whl

unicex/hyperliquid/uni_websocket_manager.py

@@ -0,0 +1,269 @@
+__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):
+    """Реализация менеджера асинхронных унифицированных вебсокетов."""
+
+    def __init__(
+        self, client: Client | UniClient | None = None, logger: LoggerLike | None = None
+    ) -> None:
+        """Инициализирует унифицированный менеджер вебсокетов.
+
+        Параметры:
+            client (`Client | UniClient | None`): Клиент Hyperliquid или унифицированный клиент. Нужен для подключения к приватным топикам.
+            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`: Экземпляр вебсокета.
+        """
+        raise NotImplementedError()
+
+    @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 NotImplementedError()
+
+    @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`: Экземпляр вебсокета.
+        """
+        raise NotImplementedError()
+
+    @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/hyperliquid/user_websocket.py

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

unicex/hyperliquid/websocket_manager.py

@@ -0,0 +1,393 @@
+__all__ = ["WebsocketManager"]
+
+import json
+from collections.abc import Awaitable, Callable
+from typing import Any
+
+from unicex._base import Websocket
+
+from .client import Client
+
+type CallbackType = Callable[[Any], Awaitable[None]]
+
+
+class WebsocketManager:
+    """Менеджер асинхронных вебсокетов для Hyperliquid."""
+
+    _URL: str = "wss://api.hyperliquid.xyz/ws"
+    """Базовый URL для вебсокета."""
+
+    def __init__(self, client: Client | None = None, **ws_kwargs: Any) -> None:
+        """Инициализирует менеджер вебсокетов для Hyperliquid.
+
+        Параметры:
+            client (`Client | None`): Клиент для выполнения запросов. Нужен, чтобы открыть приватные вебсокеты.
+            ws_kwargs (`dict[str, Any]`): Дополнительные аргументы, которые прокидываются в `Websocket`.
+        """
+        self.client = client
+        self._ws_kwargs = {
+            "ping_message": json.dumps({"method": "ping"}),
+            "ping_interval": 30,
+            **ws_kwargs,
+        }
+
+    def _create_subscription_message(self, subscription_type: str, **params: Any) -> str:
+        """Создает сообщение подписки для Hyperliquid.
+
+        Параметры:
+            subscription_type (`str`): Тип подписки.
+            params (`dict[str, Any]`): Параметры подписки.
+
+        Возвращает:
+            `str`: JSON-строка с сообщением подписки.
+        """
+        subscription = {"type": subscription_type, **params}
+        message = {"method": "subscribe", "subscription": subscription}
+        return json.dumps(message)
+
+    def all_mids(self, callback: CallbackType, dex: str | None = None) -> Websocket:
+        """Создает вебсокет для получения средних цен всех активов.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            dex (`str | None`): Perp dex для получения средних цен. Если не указан, используется первый perp dex.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        params = {}
+        if dex is not None:
+            params["dex"] = dex
+
+        subscription_message = self._create_subscription_message("allMids", **params)
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=[subscription_message],
+            **self._ws_kwargs,
+        )
+
+    def notification(self, callback: CallbackType, user: str) -> Websocket:
+        """Создает вебсокет для получения уведомлений пользователя.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            user (`str`): Адрес пользователя.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        subscription_message = self._create_subscription_message("notification", user=user)
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=[subscription_message],
+            **self._ws_kwargs,
+        )
+
+    def web_data2(self, callback: CallbackType, user: str) -> Websocket:
+        """Создает вебсокет для получения веб-данных пользователя.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            user (`str`): Адрес пользователя.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        subscription_message = self._create_subscription_message("webData2", user=user)
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=[subscription_message],
+            **self._ws_kwargs,
+        )
+
+    def candle(
+        self,
+        callback: CallbackType,
+        interval: str,
+        coin: str | None = None,
+        coins: list[str] | None = None,
+    ) -> Websocket:
+        """Создает вебсокет для получения свечей.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            coin (`str`): Символ монеты.
+            coins (`list[str]`): Список символов монет для мультиплекс подключения.
+            interval (`str`): Интервал свечей ("1m", "3m", "5m", "15m", "30m", "1h", "2h", "4h", "8h", "12h", "1d", "3d", "1w", "1M").
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        if coin and coins:
+            raise ValueError("Parameters coin and coins cannot be used together")
+        if not (coin or coins):
+            raise ValueError("Either coin or coins must be provided")
+        if coin:
+            coins = [coin]
+        subscription_messages = [
+            self._create_subscription_message("candle", coin=coin, interval=interval)
+            for coin in coins  # type: ignore
+        ]
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=subscription_messages,
+            **self._ws_kwargs,
+        )
+
+    def l2_book(
+        self,
+        callback: CallbackType,
+        coin: str,
+        n_sig_figs: int | None = None,
+        mantissa: int | None = None,
+    ) -> Websocket:
+        """Создает вебсокет для получения стакана L2.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            coin (`str`): Символ монеты.
+            n_sig_figs (`int | None`): Количество значащих цифр для округления цен.
+            mantissa (`int | None`): Мантисса для округления цен.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        params = {"coin": coin}
+        if n_sig_figs is not None:
+            params["nSigFigs"] = n_sig_figs  # type: ignore
+        if mantissa is not None:
+            params["mantissa"] = mantissa  # type: ignore
+
+        subscription_message = self._create_subscription_message("l2Book", **params)
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=[subscription_message],
+            **self._ws_kwargs,
+        )
+
+    def trades(
+        self, callback: CallbackType, coin: str | None = None, coins: list[str] | None = None
+    ) -> Websocket:
+        """Создает вебсокет для получения сделок.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            coin (`str | None`): Символ монеты.
+            coins (`list[str] | None`): Список символов монет для мультиплекс подключения.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        if coin and coins:
+            raise ValueError("Parameters coin and coins cannot be used together")
+        if not (coin or coins):
+            raise ValueError("Either coin or coins must be provided")
+        if coin:
+            coins = [coin]
+        subscription_messages = [
+            self._create_subscription_message("trades", coin=coin)
+            for coin in coins  # type: ignore
+        ]
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=subscription_messages,
+            **self._ws_kwargs,
+        )
+
+    def order_updates(self, callback: CallbackType, user: str) -> Websocket:
+        """Создает вебсокет для получения обновлений ордеров пользователя.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            user (`str`): Адрес пользователя.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        subscription_message = self._create_subscription_message("orderUpdates", user=user)
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=[subscription_message],
+            **self._ws_kwargs,
+        )
+
+    def user_events(self, callback: CallbackType, user: str) -> Websocket:
+        """Создает вебсокет для получения событий пользователя.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            user (`str`): Адрес пользователя.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        subscription_message = self._create_subscription_message("userEvents", user=user)
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=[subscription_message],
+            **self._ws_kwargs,
+        )
+
+    def user_fills(
+        self, callback: CallbackType, user: str, aggregate_by_time: bool | None = None
+    ) -> Websocket:
+        """Создает вебсокет для получения исполнений пользователя.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            user (`str`): Адрес пользователя.
+            aggregate_by_time (`bool | None`): Агрегировать по времени.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        params = {"user": user}
+        if aggregate_by_time is not None:
+            params["aggregateByTime"] = aggregate_by_time  # type: ignore
+
+        subscription_message = self._create_subscription_message("userFills", **params)
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=[subscription_message],
+            **self._ws_kwargs,
+        )
+
+    def user_fundings(self, callback: CallbackType, user: str) -> Websocket:
+        """Создает вебсокет для получения фандинга пользователя.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            user (`str`): Адрес пользователя.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        subscription_message = self._create_subscription_message("userFundings", user=user)
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=[subscription_message],
+            **self._ws_kwargs,
+        )
+
+    def user_non_funding_ledger_updates(self, callback: CallbackType, user: str) -> Websocket:
+        """Создает вебсокет для получения обновлений леджера пользователя (кроме фандинга).
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            user (`str`): Адрес пользователя.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        subscription_message = self._create_subscription_message(
+            "userNonFundingLedgerUpdates", user=user
+        )
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=[subscription_message],
+            **self._ws_kwargs,
+        )
+
+    def active_asset_ctx(self, callback: CallbackType, coin: str) -> Websocket:
+        """Создает вебсокет для получения контекста активного актива.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            coin (`str`): Символ монеты.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        subscription_message = self._create_subscription_message("activeAssetCtx", coin=coin)
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=[subscription_message],
+            **self._ws_kwargs,
+        )
+
+    def active_asset_data(self, callback: CallbackType, user: str, coin: str) -> Websocket:
+        """Создает вебсокет для получения данных активного актива пользователя (только Perps).
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            user (`str`): Адрес пользователя.
+            coin (`str`): Символ монеты.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        subscription_message = self._create_subscription_message(
+            "activeAssetData", user=user, coin=coin
+        )
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=[subscription_message],
+            **self._ws_kwargs,
+        )
+
+    def user_twap_slice_fills(self, callback: CallbackType, user: str) -> Websocket:
+        """Создает вебсокет для получения TWAP slice fills пользователя.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            user (`str`): Адрес пользователя.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        subscription_message = self._create_subscription_message("userTwapSliceFills", user=user)
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=[subscription_message],
+            **self._ws_kwargs,
+        )
+
+    def user_twap_history(self, callback: CallbackType, user: str) -> Websocket:
+        """Создает вебсокет для получения TWAP истории пользователя.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            user (`str`): Адрес пользователя.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        subscription_message = self._create_subscription_message("userTwapHistory", user=user)
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=[subscription_message],
+            **self._ws_kwargs,
+        )
+
+    def bbo(self, callback: CallbackType, coin: str) -> Websocket:
+        """Создает вебсокет для получения лучшего бида/аска.
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            coin (`str`): Символ монеты.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        subscription_message = self._create_subscription_message("bbo", coin=coin)
+        return Websocket(
+            callback=callback,
+            url=self._URL,
+            subscription_messages=[subscription_message],
+            **self._ws_kwargs,
+        )