unicex 0.15.2__py3-none-any.whl → 0.16.0__py3-none-any.whl

unicex/mexc/adapter.py

@@ -1,11 +1,14 @@
 __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
 
@@ -127,6 +130,7 @@ class Adapter:
             result[symbol] = OpenInterestItem(
                 t=item["timestamp"],
                 v=float(item["holdVol"]) * Adapter._get_contract_size(symbol),
+                u="coins",
             )
         return result
 
@@ -230,6 +234,106 @@ class Adapter:
 
         return sorted(klines, key=lambda kline_item: kline_item["t"])
 
+    @staticmethod
+    def klines_message(raw_msg: Any) -> list[KlineDict]:
+        """Преобразует вебсокет-сообщение со свечами в унифицированный формат.
+
+        Параметры:
+            raw_msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[KlineDict]: Список свечей в унифицированном формате.
+        """
+        kline = raw_msg["publicSpotKline"]
+        return [
+            KlineDict(
+                s=raw_msg["symbol"],
+                t=int(kline["windowStart"]) * 1000,
+                o=float(kline["openingPrice"]),
+                h=float(kline["highestPrice"]),
+                l=float(kline["lowestPrice"]),
+                c=float(kline["closingPrice"]),
+                v=float(kline["volume"]),
+                T=int(kline["windowEnd"]) * 1000,
+                x=None,
+                q=float(kline["amount"]),
+            )
+        ]
+
+    @staticmethod
+    def futures_klines_message(raw_msg: Any) -> list[KlineDict]:
+        """Преобразует вебсокет-сообщение со свечами в унифицированный формат.
+
+        Параметры:
+            raw_msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[KlineDict]: Список свечей в унифицированном формате.
+        """
+        data = raw_msg["data"]
+        return [
+            KlineDict(
+                s=data["symbol"],
+                t=data["t"] * 1000,
+                o=data["o"],
+                h=data["h"],
+                l=data["l"],
+                c=data["c"],
+                v=data["q"],  # Контракты
+                q=data["a"],
+                T=None,
+                x=None,
+            )
+        ]
+
+    @staticmethod
+    def trades_message(raw_msg: Any) -> list[TradeDict]:
+        """Преобразует вебсокет-сообщение со сделками в унифицированный формат.
+
+        Параметры:
+            raw_msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[TradeDict]: Список сделок в унифицированном формате.
+        """
+        return [
+            TradeDict(
+                t=trade["time"],
+                s=raw_msg["symbol"],
+                S="BUY" if trade["tradeType"] == 1 else "SELL",
+                p=float(trade["price"]),
+                v=float(trade["quantity"]),
+            )
+            for trade in sorted(
+                raw_msg["publicAggreDeals"]["deals"],
+                key=lambda item: item["time"],
+            )
+        ]
+
+    @staticmethod
+    def futures_trades_message(raw_msg: Any) -> list[TradeDict]:
+        """Преобразует вебсокет-сообщение со сделками в унифицированный формат.
+
+        Параметры:
+            raw_msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[TradeDict]: Список сделок в унифицированном формате.
+        """
+        return [
+            TradeDict(
+                t=item["t"],
+                s=raw_msg["symbol"],
+                S="BUY" if item["T"] == 1 else "SELL",
+                p=item["p"],
+                v=item["v"],
+            )
+            for item in sorted(
+                raw_msg["data"],
+                key=lambda item: item["t"],
+            )
+        ]
+
     @staticmethod
     def _get_contract_size(symbol: str) -> float:
         """Возвращает размер контракта для указанного символа тикера."""

unicex/mexc/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, MarketType, Timeframe
 from unicex.types import LoggerLike
 
 from .adapter import Adapter
@@ -76,7 +76,15 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета для управления соединением.
         """
-        raise NotImplementedError()
+        wrapper = self._make_wrapper(self._adapter.klines_message, callback)
+        return self._websocket_manager.klines(
+            callback=wrapper,
+            symbol=symbol,
+            symbols=symbols,
+            interval=timeframe.to_exchange_format(
+                Exchange.MEXC, MarketType.FUTURES
+            ),  # Тут фьючерсный интервал, потому что для вебсокета MEXC решили что сделают так (идиоты)
+        )
 
     @overload
     def futures_klines(
@@ -118,7 +126,13 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        wrapper = self._make_wrapper(self._adapter.futures_klines_message, callback)
+        return self._websocket_manager.futures_kline(
+            callback=wrapper,
+            symbol=symbol,
+            symbols=symbols,
+            interval=timeframe.to_exchange_format(Exchange.MEXC, MarketType.FUTURES),
+        )
 
     @overload
     def trades(
@@ -156,7 +170,8 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        wrapper = self._make_wrapper(self._adapter.trades_message, callback)
+        return self._websocket_manager.trade(callback=wrapper, symbol=symbol, symbols=symbols)
 
     @overload
     def aggtrades(
@@ -194,7 +209,7 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        return self.trades(callback=callback, symbol=symbol, symbols=symbols)  # type: ignore[reportCallIssue]
 
     @overload
     def futures_trades(
@@ -232,7 +247,10 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        wrapper = self._make_wrapper(self._adapter.futures_trades_message, callback)
+        return self._websocket_manager.futures_trade(
+            callback=wrapper, symbol=symbol, symbols=symbols
+        )
 
     @overload
     def futures_aggtrades(
@@ -270,4 +288,4 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        return self.futures_trades(callback=callback, symbol=symbol, symbols=symbols)  # type: ignore[reportCallIssue]

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
@@ -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
 
@@ -135,6 +138,54 @@ 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"]),
+            )
+            for trade in sorted(raw_msg["data"], key=lambda item: int(item["ts"]))
+        ]

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
@@ -36,6 +36,23 @@ class UniWebsocketManager(IUniWebsocketManager):
         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,
@@ -76,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(
@@ -118,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(
@@ -156,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(
@@ -194,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(
@@ -232,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(
@@ -270,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)

unicex/types.py

@@ -79,10 +79,10 @@ class KlineDict(TypedDict):
     q: float
     """Объем свечи. В долларах."""
 
-    T: int | None
+    T: int | None  # `None` means untrackable
     """Время закрытия. В миллисекундах."""
 
-    x: bool | None
+    x: bool | None  # `None` means untrackable
     """Флаг закрыта ли свеча."""
 
 
@@ -118,7 +118,10 @@ class OpenInterestItem(TypedDict):
     """Время. В миллисекундах."""
 
     v: float
-    """Открытый интерес. В монетах."""
+    """Открытый интерес."""
+
+    u: Literal["coins", "usd"]
+    """Единица измерения открытого интереса."""
 
 
 type OpenInterestDict = dict[str, OpenInterestItem]

unicex/utils.py

@@ -10,6 +10,7 @@ __all__ = [
     "catch_adapter_errors",
     "decorate_all_methods",
     "symbol_to_exchange_format",
+    "validate_single_symbol_args",
 ]
 
 import base64
@@ -17,7 +18,7 @@ import hashlib
 import hmac
 import json
 import time
-from collections.abc import Callable, Iterable
+from collections.abc import Callable, Iterable, Sequence
 from functools import wraps
 from typing import Any, Literal
 from urllib.parse import urlencode
@@ -232,9 +233,29 @@ def symbol_to_exchange_format(
             return symbol.removesuffix("USDT")  # Вот тут мб и не так, там вроде что-то к USDC
     elif exchange == Exchange.KUCOIN:
         if market_type == MarketType.FUTURES:
+            if symbol_upper == "BTCUSDT":
+                return "XBTUSDTM"
             return symbol_upper + "M"
         else:
             return symbol_upper.replace("USDT", "-USDT")
     elif exchange == Exchange.BINGX:
         return symbol_upper.replace("USDT", "-USDT")
     return symbol_upper
+
+
+def validate_single_symbol_args(
+    symbol: str | None = None, symbols: Sequence[str] | None = None
+) -> None:
+    """Проверяет, что передан ровно один из аргументов symbol/symbols.
+
+    Параметры:
+      symbol (`str | None`, опционально): Одиночный торговый символ.
+      symbols (`Sequence[str] | None`, опционально): Список торговых символов.
+
+    Возвращает:
+      `None`: Ничего не возвращает, выбрасывает ValueError при нарушении условий.
+    """
+    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")

{unicex-0.15.2.dist-info → unicex-0.16.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: unicex
-Version: 0.15.2
+Version: 0.16.0
 Summary: Unified Crypto Exchange API 
 Author-email: LoveBloodAndDiamonds <ayazshakirzyanov27@gmail.com>
 License: BSD 3-Clause License
@@ -56,12 +56,12 @@ Dynamic: license-file
 | Exchange        | Client | Auth | WS Manager | User WS | Uni Client | Uni WS Manager | ExchangeInfo |
 |-----------------|--------|------|------------|---------|------------|----------------|--------------|
 | **Binance**     | ✓      | ✓    | ✓          | ✓       | ✓          | ✓              | ✓            |
-| **Bitget**      | ✓      | ✓    | ✓          |         | ✓          |                | ✓            |
+| **Bitget**      | ✓      | ✓    | ✓          |         | ✓          | ✓              | ✓            |
 | **Bybit**       | ✓      | ✓    | ✓          |         | ✓          | ✓              | ✓            |
-| **Gateio**      | ✓      | ✓    | ✓          |         | ✓          |                | ✓            |
+| **Gateio**      | ✓      | ✓    | ✓          |         | ✓          | ✓              | ✓            |
 | **Hyperliquid** | ✓      | ✓    | ✓          | ✓       | ✓          |                |              |
-| **Mexc**        | ✓      | ✓    | ✓          |         | ✓          |                | ✓            |
-| **Okx**         | ✓      | ✓    | ✓          |         | ✓          |                | ✓            |
+| **Mexc**        | ✓      | ✓    | ✓          |         | ✓          | ✓              | ✓            |
+| **Okx**         | ✓      | ✓    | ✓          |         | ✓          | ✓              | ✓            |
 | **Kucoin**      |        |      |            |         | ✓          |                |              |
 | **BingX**       |        |      |            |         | ✓          |                |              |
 ---