unicex 0.15.4__py3-none-any.whl → 0.16.2__py3-none-any.whl

unicex/__init__.py

@@ -18,7 +18,6 @@ __all__ = [
     "TickerDailyItem",
     "KlineDict",
     "TradeDict",
-    "AggTradeDict",
     "RequestMethod",
     "LoggerLike",
     "OpenInterestDict",
@@ -112,7 +111,6 @@ from .types import (
     TickerDailyItem,
     KlineDict,
     TradeDict,
-    AggTradeDict,
     RequestMethod,
     LoggerLike,
     OpenInterestDict,

unicex/binance/adapter.py

@@ -2,7 +2,6 @@ __all__ = ["Adapter"]
 
 
 from unicex.types import (
-    AggTradeDict,
     KlineDict,
     OpenInterestItem,
     TickerDailyDict,
@@ -150,7 +149,7 @@ class Adapter:
         ]
 
     @staticmethod
-    def aggtrades_message(raw_msg: dict) -> list[AggTradeDict]:
+    def aggtrades_message(raw_msg: dict) -> list[TradeDict]:
         """Преобразует сырое сообщение с вебсокета, в котором содержится информация о
         аггрегированных сделке/сделках в унифицированный вид.
 
@@ -162,7 +161,7 @@ class Adapter:
         """
         msg = raw_msg.get("data", raw_msg)
         return [
-            AggTradeDict(
+            TradeDict(
                 t=int(msg["T"]),
                 s=str(msg["s"]),
                 S="SELL" if bool(msg["m"]) else "BUY",

unicex/bingx/uni_websocket_manager.py

@@ -199,7 +199,7 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        return self.trades(callback, symbol=symbol, symbols=symbols)  # type: ignore
 
     @overload
     def futures_trades(
@@ -280,4 +280,4 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        return self.futures_trades(callback, symbol=symbol, symbols=symbols)  # type: ignore

unicex/bitget/adapter.py

@@ -1,7 +1,6 @@
 from typing import Any
 
 from unicex.types import (
-    AggTradeDict,
     KlineDict,
     OpenInterestDict,
     OpenInterestItem,
@@ -155,7 +154,7 @@ class Adapter:
         symbol = raw_msg["arg"]["instId"]
 
         return [
-            AggTradeDict(
+            TradeDict(
                 t=int(trade["ts"]),
                 s=symbol,
                 S=trade["side"].upper(),

unicex/bitget/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,17 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета для управления соединением.
         """
-        raise NotImplementedError()
+        wrapper = self._make_wrapper(self._adapter.klines_message, callback)
+        return self._websocket_manager.candlestick(
+            callback=wrapper,
+            market_type="SPOT",
+            symbol=symbol,
+            symbols=symbols,
+            interval=timeframe.to_exchange_format(
+                Exchange.BITGET,
+                MarketType.FUTURES,  # Тут пришлось поставить Futures, потому что:
+            ),  # кто бы мог подумать, что у Bitget на споте для вебсокетов и HTTP запросов совершенно разные перечисления. Тупые ублюдки.
+        )
 
     @overload
     def futures_klines(
@@ -118,7 +128,14 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        wrapper = self._make_wrapper(self._adapter.klines_message, callback)
+        return self._websocket_manager.candlestick(
+            callback=wrapper,
+            market_type="USDT-FUTURES",
+            symbol=symbol,
+            symbols=symbols,
+            interval=timeframe.to_exchange_format(Exchange.BITGET, MarketType.FUTURES),
+        )
 
     @overload
     def trades(
@@ -276,4 +293,4 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        return self.futures_trades(callback=callback, symbol=symbol, symbols=symbols)  # type: ignore[reportCallIssue]

unicex/gate/adapter.py

@@ -1,6 +1,7 @@
 from __future__ import annotations
 
 import time
+from typing import Any
 
 __all__ = ["Adapter"]
 
@@ -10,9 +11,12 @@ from unicex.types import (
     OpenInterestItem,
     TickerDailyDict,
     TickerDailyItem,
+    TradeDict,
 )
 from unicex.utils import catch_adapter_errors, decorate_all_methods
 
+from .exchange_info import ExchangeInfo
+
 
 @decorate_all_methods(catch_adapter_errors)
 class Adapter:
@@ -177,3 +181,111 @@ class Adapter:
             )
             for item in raw_data
         }
+
+    @staticmethod
+    def klines_message(raw_msg: Any) -> list[KlineDict]:
+        """Преобразует вебсокет-сообщение со свечами в унифицированный формат.
+
+        Параметры:
+            raw_msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[KlineDict]: Список свечей в унифицированном формате.
+        """
+        data = raw_msg["result"]
+        return [
+            KlineDict(
+                s=data["n"].split("_", 1)[1],  # XRP_USDT
+                t=int(data["t"]) * 1000,
+                o=float(data["o"]),
+                h=float(data["h"]),
+                l=float(data["l"]),
+                c=float(data["c"]),
+                v=float(data["a"]),
+                q=float(data["v"]),
+                T=None,
+                x=not data["w"],  # w=False → свеча закрыта
+            )
+        ]
+
+    @staticmethod
+    def futures_klines_message(raw_msg: Any) -> list[KlineDict]:
+        """Преобразует вебсокет-сообщение со свечами в унифицированный формат.
+
+        Параметры:
+            raw_msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[KlineDict]: Список свечей в унифицированном формате.
+        """
+        return [
+            KlineDict(
+                s=item["n"].split("_", 1)[1],  # XRP_USDT
+                t=int(item["t"]) * 1000,
+                o=float(item["o"]),
+                h=float(item["h"]),
+                l=float(item["l"]),
+                c=float(item["c"]),
+                v=float(item["a"]),
+                q=float(item["v"]),
+                T=None,
+                x=not item["w"],  # w=False → свеча закрыта
+            )
+            for item in sorted(
+                raw_msg["result"],
+                key=lambda x: int(x["t"]),
+            )
+        ]
+
+    @staticmethod
+    def trades_message(raw_msg: Any) -> list[TradeDict]:
+        """Преобразует вебсокет-сообщение со сделками в унифицированный формат.
+
+        Параметры:
+            raw_msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[TradeDict]: Список сделок в унифицированном формате.
+        """
+        trade = raw_msg["result"]
+        return [
+            TradeDict(
+                t=trade["create_time_ms"],
+                s=trade["currency_pair"],
+                S=trade["side"].upper(),
+                p=float(trade["price"]),
+                v=float(trade["amount"]),
+            )
+        ]
+
+    @staticmethod
+    def futures_trades_message(raw_msg: Any) -> list[TradeDict]:
+        """Преобразует вебсокет-сообщение со сделками в унифицированный формат.
+
+        Параметры:
+            raw_msg (Any): Сырое сообщение с вебсокета.
+
+        Возвращает:
+            list[TradeDict]: Список сделок в унифицированном формате.
+        """
+        return [
+            TradeDict(
+                t=item["create_time_ms"],
+                s=item["contract"],
+                S="BUY" if float(item["size"]) > 0 else "SELL",
+                p=float(item["price"]),
+                v=abs(float(item["size"])) * Adapter._get_contract_size(item["contract"]),
+            )
+            for item in sorted(
+                raw_msg["result"],
+                key=lambda x: x["create_time_ms"],
+            )
+        ]
+
+    @staticmethod
+    def _get_contract_size(symbol: str) -> float:
+        """Возвращает размер контракта для указанного символа тикера."""
+        try:
+            return ExchangeInfo.get_futures_ticker_info(symbol)["contract_size"] or 1
+        except:  # noqa
+            return 1

unicex/gate/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
@@ -36,6 +36,20 @@ class UniWebsocketManager(IUniWebsocketManager):
         self._websocket_manager = WebsocketManager(self._client, **ws_kwargs)  # type: ignore
         self._adapter = Adapter()
 
+    def _normalize_symbols(
+        self,
+        symbol: str | None,
+        symbols: Sequence[str] | None,
+    ) -> list[str]:
+        """Преобразует параметры symbol/symbols в список тикеров."""
+        if symbol and symbols:
+            raise ValueError("Parameters symbol and symbols cannot be used together")
+        if symbol:
+            return [symbol]
+        if symbols:
+            return list(symbols)
+        raise ValueError("Either symbol or symbols must be provided")
+
     @overload
     def klines(
         self,
@@ -76,7 +90,14 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета для управления соединением.
         """
-        raise NotImplementedError()
+        tickers = self._normalize_symbols(symbol, symbols)
+
+        wrapper = self._make_wrapper(self._adapter.klines_message, callback)
+        return self._websocket_manager.candlesticks(
+            callback=wrapper,
+            interval=timeframe.to_exchange_format(Exchange.GATE, MarketType.SPOT),
+            symbols=tickers,
+        )
 
     @overload
     def futures_klines(
@@ -118,7 +139,14 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        tickers = self._normalize_symbols(symbol, symbols)
+
+        wrapper = self._make_wrapper(self._adapter.futures_klines_message, callback)
+        return self._websocket_manager.futures_candlesticks(
+            callback=wrapper,
+            interval=timeframe.to_exchange_format(Exchange.GATE, MarketType.FUTURES),
+            symbols=tickers,
+        )
 
     @overload
     def trades(
@@ -156,7 +184,10 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        tickers = self._normalize_symbols(symbol, symbols)
+
+        wrapper = self._make_wrapper(self._adapter.trades_message, callback)
+        return self._websocket_manager.trades(callback=wrapper, symbols=tickers)
 
     @overload
     def aggtrades(
@@ -194,7 +225,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 +263,10 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        tickers = self._normalize_symbols(symbol, symbols)
+
+        wrapper = self._make_wrapper(self._adapter.futures_trades_message, callback)
+        return self._websocket_manager.futures_trades(callback=wrapper, symbols=tickers)
 
     @overload
     def futures_aggtrades(
@@ -270,4 +304,4 @@ class UniWebsocketManager(IUniWebsocketManager):
         Возвращает:
             `Websocket`: Экземпляр вебсокета.
         """
-        raise NotImplementedError()
+        return self.futures_trades(callback=callback, symbol=symbol, symbols=symbols)  # type: ignore[reportCallIssue]

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
 
@@ -231,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"] * Adapter._get_contract_size(raw_msg["symbol"]),
+            )
+            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

@@ -380,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,15 +1,20 @@
 __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
 
+from .exchange_info import ExchangeInfo
+
 
 @decorate_all_methods(catch_adapter_errors)
 class Adapter:
@@ -139,3 +144,58 @@ class Adapter:
             )
             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:
+        """Возвращает размер контракта для указанного символа тикера."""
+        try:
+            return ExchangeInfo.get_futures_ticker_info(symbol)["contract_size"] or 1
+        except:  # noqa
+            return 1

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,
             )