unicex 0.13.17__py3-none-any.whl → 0.16.5__py3-none-any.whl

unicex/mexc/websocket_manager.py

@@ -8,6 +8,7 @@ import orjson
 from google.protobuf.json_format import MessageToDict
 
 from unicex._base import Websocket
+from unicex.utils import validate_single_symbol_args
 
 from ._spot_ws_proto import PushDataV3ApiWrapper
 from .client import Client
@@ -27,7 +28,7 @@ class WebsocketManager:
     class _MexcProtobufDecoder:
         """Класс для декодирования сообщений в формате Protobuf со спотового рынка Mexc."""
 
-        def decode(self, message: Any) -> dict:
+        def decode(self, message: Any) -> dict | Literal["ping"]:
             if isinstance(message, bytes):
                 wrapper = PushDataV3ApiWrapper()  # noqa
                 wrapper.ParseFromString(message)
@@ -55,10 +56,7 @@ class WebsocketManager:
         **template_kwargs: Any,
     ) -> list[str]:
         """Сформировать сообщение для подписки на вебсокет."""
-        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")
+        validate_single_symbol_args(symbol, symbols)
 
         if symbol:
             params = [channel_template.format(symbol=symbol, **template_kwargs)]
@@ -382,7 +380,30 @@ class WebsocketManager:
             `Websocket`: Объект для управления вебсокет соединением.
         """
         subscription_messages = self._generate_futures_subscription_message(
-            topic="sub.deal", symbol=symbol, symbols=symbols, interval=interval
+            topic="sub.kline", symbol=symbol, symbols=symbols, interval=interval
+        )
+        return self._create_futures_websocket(callback, subscription_messages)
+
+    def futures_trade(
+        self,
+        callback: CallbackType,
+        symbol: str | None = None,
+        symbols: Sequence[str] | None = None,
+    ) -> Websocket:
+        """Создает вебсокет для получения сделок по фьючерсным контрактам.
+
+        https://mexcdevelop.github.io/apidocs/contract_v1_en/#public-channels
+
+        Параметры:
+            callback (`CallbackType`): Асинхронная функция обратного вызова для обработки сообщений.
+            symbol (`str | None`): Символ фьючерсного контракта.
+            symbols (`Sequence[str] | None`): Последовательность символов фьючерсных контрактов.
+
+        Возвращает:
+            `Websocket`: Объект для управления вебсокет соединением.
+        """
+        subscription_messages = self._generate_futures_subscription_message(
+            topic="sub.deal", symbol=symbol, symbols=symbols
         )
         return self._create_futures_websocket(callback, subscription_messages)
 

unicex/okx/adapter.py

@@ -1,12 +1,15 @@
 __all__ = ["Adapter"]
 
 
+from typing import Any
+
 from unicex.types import (
     KlineDict,
     OpenInterestDict,
     OpenInterestItem,
     TickerDailyDict,
     TickerDailyItem,
+    TradeDict,
 )
 from unicex.utils import catch_adapter_errors, decorate_all_methods
 
@@ -137,10 +140,58 @@ class Adapter:
             item["instId"]: OpenInterestItem(
                 t=int(item["ts"]),
                 v=float(item["oiCcy"]),
+                u="coins",
             )
             for item in raw_data["data"]
         }
 
+    @staticmethod
+    def klines_message(raw_msg: Any) -> list[KlineDict]:
+        """Преобразует вебсокет-сообщение со свечами в унифицированный формат.
+
+        Параметры:
+            raw_msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[KlineDict]: Список свечей в унифицированном формате.
+        """
+        return [
+            KlineDict(
+                s=raw_msg["arg"]["instId"],
+                t=int(kline[0]),
+                o=float(kline[1]),
+                h=float(kline[2]),
+                l=float(kline[3]),
+                c=float(kline[4]),
+                v=float(kline[6]),
+                q=float(kline[7]),
+                T=None,
+                x=bool(int(kline[8])),
+            )
+            for kline in sorted(raw_msg["data"], key=lambda item: int(item[0]))
+        ]
+
+    @staticmethod
+    def trades_message(raw_msg: Any) -> list[TradeDict]:
+        """Преобразует вебсокет-сообщение со сделками в унифицированный формат.
+
+        Параметры:
+            raw_msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[TradeDict]: Список сделок в унифицированном формате.
+        """
+        return [
+            TradeDict(
+                t=int(trade["ts"]),
+                s=trade["instId"],
+                S=trade["side"].upper(),
+                p=float(trade["px"]),
+                v=float(trade["sz"]) * Adapter._get_contract_size(trade["instId"]),
+            )
+            for trade in sorted(raw_msg["data"], key=lambda item: int(item["ts"]))
+        ]
+
     @staticmethod
     def _get_contract_size(symbol: str) -> float:
         """Возвращает размер контракта для указанного символа тикера."""

unicex/okx/client.py

@@ -6,7 +6,7 @@ from typing import Any, Literal
 
 from unicex._base import BaseClient
 from unicex.exceptions import NotAuthorized
-from unicex.types import RequestMethod
+from unicex.types import NumberLike, RequestMethod
 from unicex.utils import filter_params, generate_hmac_sha256_signature
 
 
@@ -510,7 +510,7 @@ class Client(BaseClient):
         inst_id: str,
         td_mode: Literal["cross", "isolated", "cash", "spot_isolated"],
         ccy: str | None = None,
-        px: str | None = None,
+        px: NumberLike | None = None,
         leverage: str | None = None,
         trade_quote_ccy: str | None = None,
     ) -> dict:
@@ -540,7 +540,7 @@ class Client(BaseClient):
         td_mode: Literal["cross", "isolated", "cash", "spot_isolated"],
         ccy: str | None = None,
         reduce_only: bool | None = None,
-        px: str | None = None,
+        px: NumberLike | None = None,
         trade_quote_ccy: str | None = None,
     ) -> dict:
         """Получение максимального доступного баланса/эквити.
@@ -1313,14 +1313,14 @@ class Client(BaseClient):
             "mmp_and_post_only",
             "op_fok",
         ],
-        sz: str,
+        sz: NumberLike,
         ccy: str | None = None,
         cl_ord_id: str | None = None,
         tag: str | None = None,
         pos_side: Literal["net", "long", "short"] | None = None,
-        px: str | None = None,
-        px_usd: str | None = None,
-        px_vol: str | None = None,
+        px: NumberLike | None = None,
+        px_usd: NumberLike | None = None,
+        px_vol: NumberLike | None = None,
         reduce_only: bool | None = None,
         tgt_ccy: Literal["base_ccy", "quote_ccy"] | None = None,
         ban_amend: bool | None = None,
@@ -1430,10 +1430,10 @@ class Client(BaseClient):
         ord_id: str | None = None,
         cl_ord_id: str | None = None,
         *,
-        new_sz: str | None = None,
-        new_px: str | None = None,
-        new_px_usd: str | None = None,
-        new_px_vol: str | None = None,
+        new_sz: NumberLike | None = None,
+        new_px: NumberLike | None = None,
+        new_px_usd: NumberLike | None = None,
+        new_px_vol: NumberLike | None = None,
         cxl_on_fail: bool | None = None,
         req_id: str | None = None,
         px_amend_type: Literal["0", "1"] | None = None,
@@ -1987,9 +1987,9 @@ class Client(BaseClient):
             "ioc",
             "optimal_limit_ioc",
         ],
-        sz: str,
+        sz: NumberLike,
         pos_side: Literal["net", "long", "short"] | None = None,
-        px: str | None = None,
+        px: NumberLike | None = None,
         reduce_only: bool | None = None,
         tgt_ccy: Literal["base_ccy", "quote_ccy"] | None = None,
         attach_algo_orders: list[dict[str, Any]] | None = None,
@@ -2592,9 +2592,9 @@ class Client(BaseClient):
     async def convert_contract_coin(
         self,
         inst_id: str,
-        sz: str,
+        sz: NumberLike,
         type_: Literal["1", "2"] | None = None,
-        px: str | None = None,
+        px: NumberLike | None = None,
         unit: Literal["coin", "usds"] | None = None,
         op_type: Literal["open", "close"] | None = None,
     ) -> dict:

unicex/okx/exchange_info.py

@@ -22,9 +22,9 @@ class ExchangeInfo(IExchangeInfo):
         for el in exchange_info["data"]:
             tickers_info[el["instId"]] = TickerInfoItem(
                 tick_precision=None,
-                tick_step=float(el["tickSz"]),
+                tick_step=float(el["tickSz"] or "0"),
                 size_precision=None,
-                size_step=float(el["lotSz"]),
+                size_step=float(el["lotSz"] or "0"),
                 contract_size=1,
             )
 

unicex/okx/uni_websocket_manager.py

@@ -5,7 +5,7 @@ from typing import Any, overload
 
 from unicex._abc import IUniWebsocketManager
 from unicex._base import Websocket
-from unicex.enums import Timeframe
+from unicex.enums import Exchange, Timeframe
 from unicex.types import LoggerLike
 
 from .adapter import Adapter
@@ -20,18 +20,39 @@ class UniWebsocketManager(IUniWebsocketManager):
     """Реализация менеджера асинхронных унифицированных вебсокетов."""
 
     def __init__(
-        self, client: Client | UniClient | None = None, logger: LoggerLike | None = None
+        self,
+        client: Client | UniClient | None = None,
+        logger: LoggerLike | None = None,
+        **ws_kwargs: Any,
     ) -> None:
         """Инициализирует унифицированный менеджер вебсокетов.
 
         Параметры:
             client (`Client | UniClient | None`): Клиент Okx или унифицированный клиент. Нужен для подключения к приватным топикам.
             logger (`LoggerLike | None`): Логгер для записи логов.
+            ws_kwargs (`dict[str, Any]`): Дополнительные параметры инициализации, которые будут переданы WebsocketManager/Websocket.
         """
         super().__init__(client=client, logger=logger)
-        self._websocket_manager = WebsocketManager(self._client)  # type: ignore
+        self._websocket_manager = WebsocketManager(self._client, **ws_kwargs)  # type: ignore
         self._adapter = Adapter()
 
+    def _normalize_symbol(
+        self,
+        symbol: str | None,
+        symbols: Sequence[str] | None,
+    ) -> str:
+        """Преобразует параметры symbol/symbols в один тикер."""
+        if symbol and symbols:
+            raise ValueError("Parameters symbol and symbols cannot be used together")
+        if symbol:
+            return symbol
+        if symbols:
+            normalized = list(symbols)
+            if len(normalized) != 1:
+                raise ValueError("OKX websocket поддерживает только один тикер на соединение")
+            return normalized[0]
+        raise ValueError("Either symbol or symbols must be provided")
+
     @overload
     def klines(
         self,
@@ -72,7 +93,13 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета для управления соединением.
         """
-        raise NotImplementedError()
+        inst_id = self._normalize_symbol(symbol, symbols)
+        wrapper = self._make_wrapper(self._adapter.klines_message, callback)
+        return self._websocket_manager.candlesticks(
+            callback=wrapper,
+            interval=timeframe.to_exchange_format(Exchange.OKX),  # type: ignore
+            inst_id=inst_id,
+        )
 
     @overload
     def futures_klines(
@@ -114,7 +141,13 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        inst_id = self._normalize_symbol(symbol, symbols)
+        wrapper = self._make_wrapper(self._adapter.klines_message, callback)
+        return self._websocket_manager.candlesticks(
+            callback=wrapper,
+            interval=timeframe.to_exchange_format(Exchange.OKX),  # type: ignore
+            inst_id=inst_id,
+        )
 
     @overload
     def trades(
@@ -152,7 +185,9 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        inst_id = self._normalize_symbol(symbol, symbols)
+        wrapper = self._make_wrapper(self._adapter.trades_message, callback)
+        return self._websocket_manager.all_trades(callback=wrapper, inst_id=inst_id)
 
     @overload
     def aggtrades(
@@ -190,7 +225,9 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        inst_id = self._normalize_symbol(symbol, symbols)
+        wrapper = self._make_wrapper(self._adapter.trades_message, callback)
+        return self._websocket_manager.trades(callback=wrapper, inst_id=inst_id)
 
     @overload
     def futures_trades(
@@ -228,7 +265,9 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        inst_id = self._normalize_symbol(symbol, symbols)
+        wrapper = self._make_wrapper(self._adapter.trades_message, callback)
+        return self._websocket_manager.all_trades(callback=wrapper, inst_id=inst_id)
 
     @overload
     def futures_aggtrades(
@@ -266,4 +305,6 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        inst_id = self._normalize_symbol(symbol, symbols)
+        wrapper = self._make_wrapper(self._adapter.trades_message, callback)
+        return self._websocket_manager.trades(callback=wrapper, inst_id=inst_id)