trd-utils 0.0.28__tar.gz → 0.0.30__tar.gz

{trd_utils-0.0.28 → trd_utils-0.0.30}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: trd_utils
-Version: 0.0.28
+Version: 0.0.30
 Summary: Common Basic Utils for Python3. By ALiwoto.
 Keywords: utils,trd_utils,basic-utils,common-utils
 Author: ALiwoto
@@ -19,6 +19,8 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: cryptography (>=41.0.7)
 Requires-Dist: httpx (>=0.21.0)
+Requires-Dist: python-dateutil (>=2.9.0.post0)
+Requires-Dist: websockets (>=15.0.1)
 Project-URL: Homepage, https://github.com/ALiwoto/trd_utils
 Description-Content-Type: text/markdown
 

{trd_utils-0.0.28 → trd_utils-0.0.30}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "trd_utils"
-version = "0.0.28"
+version = "0.0.30"
 description = "Common Basic Utils for Python3. By ALiwoto."
 authors = ["ALiwoto <aminnimaj@gmail.com>"]
 packages = [
@@ -20,6 +20,8 @@ homepage = "https://github.com/ALiwoto/trd_utils"
 python = ">=3.7"
 httpx = ">=0.21.0"
 cryptography = ">=41.0.7"
+websockets = ">=15.0.1"
+python-dateutil = ">=2.9.0.post0"
 
 [tool.poetry.dev-dependencies]
 pytest = ">=7.2"

trd_utils-0.0.30/trd_utils/__init__.py

@@ -0,0 +1,3 @@
+
+__version__ = "0.0.30"
+

trd_utils-0.0.30/trd_utils/common_utils/float_utils.py

@@ -0,0 +1,21 @@
+
+from decimal import Decimal
+
+
+default_quantize = Decimal("1.00")
+
+def dec_to_str(dec_value: Decimal) -> str:
+    return format(dec_value.quantize(default_quantize), "f")
+
+def dec_to_normalize(dec_value: Decimal) -> str:
+    return format(dec_value.normalize(), "f")
+
+def as_decimal(value) -> Decimal:
+    if value is None:
+        return None
+    
+    if isinstance(value, Decimal):
+        # prevent extra allocation
+        return value
+    
+    return Decimal(value)

{trd_utils-0.0.28 → trd_utils-0.0.30}/trd_utils/exchanges/base_types.py

@@ -37,6 +37,14 @@ class UnifiedPositionInfo(BaseModel):
     # The base unit that the open-price is based on (e.g. USD, USDT, USDC)
     open_price_unit: str | None = None
 
+    # The last price of this pair on the target exchange.
+    # not all exchanges support this yet, so use it with caution.
+    last_price: Decimal | None = None
+
+    # The last volume of this pair being traded on the target exchange.
+    # not all exchanges support this yet, so use it with caution.
+    last_volume: Decimal | None = None
+
     def __str__(self):
         parts = []
 

{trd_utils-0.0.28 → trd_utils-0.0.30}/trd_utils/exchanges/bx_ultra/bx_types.py

@@ -3,9 +3,11 @@ from decimal import Decimal
 from datetime import datetime, timedelta
 import pytz
 
+from trd_utils.exchanges.errors import ExchangeError
 from trd_utils.types_helper import BaseModel
 
 from trd_utils.common_utils.float_utils import (
+    as_decimal,
     dec_to_str,
     dec_to_normalize,
 )
@@ -88,6 +90,28 @@ class ExchangeVo(BaseModel):
         return f"{self.exchange_name} ({self.exchange_id}) - {self.account_enum}"
 
 
+class OrderLeverInfo(BaseModel):
+    lever_times: int = None
+    selected: bool = False
+
+    def __str__(self):
+        return f"{self.lever_times}x"
+
+    def __repr__(self):
+        return self.__str__()
+
+
+class MarginDisplayInfo(BaseModel):
+    margin: Decimal = None
+    selected: bool = False
+
+
+class SysForceVoInfo(BaseModel):
+    begin_level: int = None
+    end_level: int = None
+    adjust_margin_rate: Decimal = None
+
+
 # endregion
 
 ###########################################################
@@ -387,12 +411,15 @@ class ZenDeskABStatusResult(BaseModel):
 class ZenDeskABStatusResponse(BxApiResponse):
     data: ZenDeskABStatusResult = None
 
+
 class ZenDeskAuthResult(BaseModel):
     jwt: str = None
 
+
 class ZenDeskAuthResponse(BxApiResponse):
     data: ZenDeskAuthResult = None
 
+
 # endregion
 
 ###########################################################
@@ -550,6 +577,7 @@ class CopyTraderInfo(BaseModel):
     short_uid: int = None
     identity_type: int = None
 
+
 class CopyTraderVo(BaseModel):
     audit_status: int = None
     trader_status: int = None
@@ -565,6 +593,7 @@ class CopyTraderVo(BaseModel):
     rank_account_id: int = None
     last_trader_time: datetime = None
 
+
 class CopyTraderAccountGradeVO(BaseModel):
     uid: int = None
     api_identity: int = None
@@ -572,6 +601,7 @@ class CopyTraderAccountGradeVO(BaseModel):
     label: int = None
     uid_and_api: str = None
 
+
 class CopyTraderSharingAccount(BaseModel):
     category: int = None
     trader: int = None
@@ -588,6 +618,7 @@ class CopyTraderSharingAccount(BaseModel):
     hide_info: int = None
     copy_trade_label_type: Optional[int] = None
 
+
 class CopyTraderResumeResult(BaseModel):
     trader_info: CopyTraderInfo = None
     trader_vo: CopyTraderVo = None
@@ -605,9 +636,11 @@ class CopyTraderResumeResult(BaseModel):
     swap_copy_trade_label_type: int = None
     is_pro: int = None
 
+
 class CopyTraderResumeResponse(BxApiResponse):
     data: CopyTraderResumeResult = None
 
+
 # endregion
 
 ###########################################################
@@ -782,6 +815,13 @@ class SearchCopyTradersResponse(BxApiResponse):
 # region Account Assets types
 
 
+class MinimalAssetInfo(BaseModel):
+    asset_id: int = None
+    asset_amount: Decimal = None
+    asset_name: str = None
+    has_value: bool = None
+
+
 class TotalAssetsInfo(BaseModel):
     amount: Any = None  # unknown
     currency_amount: Decimal = None
@@ -1042,6 +1082,9 @@ class ContractOrdersHistoryResponse(BxApiResponse):
         found_any_for_today: bool = False
         today_earnings = Decimal("0.00")
         today = datetime.now(timezone).date()
+        if not self.data and self.msg:
+            raise ExchangeError(self.msg)
+
         for current_order in self.data.orders:
             # check if the date is for today
             closed_date = (
@@ -1123,6 +1166,107 @@ class ContractOrdersHistoryResponse(BxApiResponse):
         return len(self.data.orders)
 
 
+class ContractConfigData(BaseModel):
+    quotation_coin_id: int = None
+    max_lever: OrderLeverInfo = None
+    min_amount: Decimal = None
+    levers: list[OrderLeverInfo] = None
+    default_stop_loss_rate: Decimal = None
+    default_stop_profit_rate: Decimal = None
+    max_stop_loss_rate: Decimal = None
+    max_stop_profit_rate: Decimal = None
+    fee_rate: Decimal = None
+    interest_rate: Decimal = None
+    lever_fee_rate: Decimal = None
+    sys_force_rate: Decimal = None
+    new_sys_force_vo_list: list[SysForceVoInfo] = None
+    margin_displays: list[MarginDisplayInfo] = None
+    mlr: Decimal = None
+    lsf: Decimal = None
+    lsh: Decimal = None
+    hold_amount: Decimal = None
+    msr: Decimal = None
+    sfa: Decimal = None
+    available_asset: MinimalAssetInfo = None
+    coupon_asset_value: Any = None
+    contract_account_balance: MinimalAssetInfo = None
+    delegate_order_up_threshold_rate: Any = None
+    delegate_order_down_threshold_rate: Any = None
+    profit_loss_extra_vo: Any = None
+    fund_balance: Decimal = None
+    balance: Decimal = None
+    up_amount: Decimal = None
+    down_amount: Decimal = None
+    max_amount: Decimal = None
+    stop_offset_rate: Decimal = None
+
+
+class ContractConfigResponse(BxApiResponse):
+    data: ContractConfigData = None
+
+
+# endregion
+
+###########################################################
+
+# region contract delegation types
+
+
+class CreateOrderDelegationData(BaseModel):
+    order_id: str = None
+    spread_rate: str = None
+
+
+class CreateOrderDelegationResponse(BxApiResponse):
+    data: CreateOrderDelegationData = None
+
+
+# endregion
+
+###########################################################
+
+# region candle types
+
+
+class SingleCandleInfo(BaseModel):
+    # The pair in format of BTC/USDT.
+    pair: str = None
+
+    # This candle's open price.
+    open_price: Decimal = None
+
+    # The close price.
+    close_price: Decimal = None
+
+    # volume in the first pair (e.g. BTC).
+    volume: Decimal = None
+
+    # volume in the second part of the pair (e.g. USDT).
+    quote_volume: Decimal = None
+
+    @staticmethod
+    def deserialize_short(data: dict) -> "SingleCandleInfo":
+        info = SingleCandleInfo()
+        base: str = data.get("n", "")
+        quote: str = data.get("m", "")
+        info.pair = f"{base.upper()}/{quote.upper()}"
+        info.open_price = as_decimal(data.get("o", None))
+        info.close_price = as_decimal(data.get("c", None))
+        info.volume = data.get("v", None)
+        info.quote_volume = data.get("a", None)
+
+        return info
+
+    def __str__(self):
+        return (
+            f"{self.pair}, open: {self.open_price}, "
+            f"close: {self.close_price}, volume: {self.quote_volume}"
+        )
+    
+    def __repr__(self):
+        return super().__str__()
+
+
 # endregion
 
 ###########################################################

{trd_utils-0.0.28 → trd_utils-0.0.30}/trd_utils/exchanges/bx_ultra/bx_ultra_client.py

@@ -3,9 +3,12 @@ BxUltra exchange subclass
 """
 
 import asyncio
+from datetime import datetime, timedelta, timezone
 from decimal import Decimal
 import json
 import logging
+import gzip
+from urllib.parse import urlencode
 import uuid
 
 import httpx
@@ -13,6 +16,11 @@ import httpx
 import time
 from pathlib import Path
 
+import websockets
+import websockets.asyncio
+import websockets.asyncio.client
+import websockets.asyncio.connection
+
 from trd_utils.exchanges.base_types import (
     UnifiedPositionInfo,
     UnifiedTraderInfo,
@@ -21,17 +29,20 @@ from trd_utils.exchanges.base_types import (
 from trd_utils.exchanges.bx_ultra.bx_utils import do_ultra_ss
 from trd_utils.exchanges.bx_ultra.bx_types import (
     AssetsInfoResponse,
+    ContractConfigResponse,
     ContractOrdersHistoryResponse,
     ContractsListResponse,
     CopyTraderFuturesStatsResponse,
     CopyTraderResumeResponse,
     CopyTraderTradePositionsResponse,
+    CreateOrderDelegationResponse,
     HintListResponse,
     HomePageResponse,
     HotSearchResponse,
     QuotationRankResponse,
     SearchCopyTraderCondition,
     SearchCopyTradersResponse,
+    SingleCandleInfo,
     UserFavoriteQuotationResponse,
     ZenDeskABStatusResponse,
     ZenDeskAuthResponse,
@@ -39,7 +50,8 @@ from trd_utils.exchanges.bx_ultra.bx_types import (
 )
 from trd_utils.cipher import AESCipher
 
-from trd_utils.exchanges.exchange_base import ExchangeBase
+from trd_utils.exchanges.errors import ExchangeError
+from trd_utils.exchanges.exchange_base import ExchangeBase, JWTManager
 
 PLATFORM_ID_ANDROID = "10"
 PLATFORM_ID_WEB = "30"
@@ -71,6 +83,7 @@ class BXUltraClient(ExchangeBase):
     # region client parameters
     we_api_base_host: str = "\u0061pi-\u0061pp.w\u0065-\u0061pi.com"
     we_api_base_url: str = "https://\u0061pi-\u0061pp.w\u0065-\u0061pi.com/\u0061pi"
+    ws_we_api_base_url: str = "wss://ws-market-swap.w\u0065-\u0061pi.com/ws"
     original_base_host: str = "https://\u0062ing\u0078.co\u006d"
     qq_os_base_host: str = "https://\u0061pi-\u0061pp.\u0071\u0071-os.com"
     qq_os_base_url: str = "https://\u0061pi-\u0061pp.\u0071\u0071-os.com/\u0061pi"
@@ -85,6 +98,9 @@ class BXUltraClient(ExchangeBase):
     platform_lang: str = "en"
     sys_lang: str = "en"
 
+    # a dict that maps "BTC/USDT" to it single candle info.
+    __last_candle_storage: dict = None
+    __last_candle_lock: asyncio.Lock = None
     # endregion
     ###########################################################
     # region client constructor
@@ -106,8 +122,16 @@ class BXUltraClient(ExchangeBase):
         self.device_brand = device_brand
         self.app_version = app_version
         self._fav_letter = fav_letter
+        self.sessions_dir = sessions_dir
 
-        self.read_from_session_file(f"{sessions_dir}/{self.account_name}.bx")
+        self.read_from_session_file(
+            file_path=f"{self.sessions_dir}/{self.account_name}.bx"
+        )
+        self.__last_candle_storage = {}
+        self.__last_candle_lock = asyncio.Lock()
+        self._internal_lock = asyncio.Lock()
+        self.extra_tasks = []
+        self.ws_connections = []
 
     # endregion
     ###########################################################
@@ -222,6 +246,15 @@ class BXUltraClient(ExchangeBase):
             model_type=ZenDeskAuthResponse,
         )
 
+    async def re_authorize_user(self) -> bool:
+        result = await self.do_zendesk_auth()
+        if not result.data.jwt:
+            return False
+
+        self.authorization_token = result.data.jwt
+        self._save_session_file(file_path=f"{self.sessions_dir}/{self.account_name}.bx")
+        return True
+
     # endregion
     ###########################################################
     # region platform-tool
@@ -244,9 +277,120 @@ class BXUltraClient(ExchangeBase):
             model_type=AssetsInfoResponse,
         )
 
+    # endregion
+    ###########################################################
+    # region ws-subscribes
+    async def do_price_subscribe(self) -> None:
+        """
+        Subscribes to the price changes coming from the exchange.
+        NOTE: This method DOES NOT return. you should do create_task
+            for it.
+        """
+        params = {
+            "platformid": self.platform_id,
+            "app_version": self.app_version,
+            "x-router-tag": self.x_router_tag,
+            "lang": self.platform_lang,
+            "device_id": self.device_id,
+            "channel": self.channel_header,
+            "device_brand": self.device_brand,
+            "traceId": self.trace_id,
+        }
+        url = f"{self.ws_we_api_base_url}?{urlencode(params, doseq=True)}"
+        async with websockets.connect(url, ping_interval=None) as ws:
+            await self._internal_lock.acquire()
+            self.ws_connections.append(ws)
+            self._internal_lock.release()
+
+            await ws.send(json.dumps({
+                "dataType": "swap.market.v2.contracts",
+                "id": uuid.uuid4().hex,
+                "reqType": "sub",
+            }))
+            async for msg in ws:
+                try:
+                    decompressed_message = gzip.decompress(msg)
+                    if decompressed_message.lower() == "ping":
+                        await ws.send("Pong")
+                        continue
+
+                    data: dict = json.loads(decompressed_message, parse_float=Decimal)
+                    if not isinstance(data, dict):
+                        logger.warning(f"invalid data instance: {type(data)}")
+                        continue
+
+                    if data.get("code", 0) == 0 and data.get("data", None) is None:
+                        # it's all fine
+                        continue
+
+                    if data.get("ping", None):
+                        target_id = data["ping"]
+                        target_time = data.get(
+                            "time",
+                            datetime.now(
+                                timezone(timedelta(hours=8))
+                            ).isoformat(timespec="seconds")
+                        )
+                        await ws.send(json.dumps({
+                            "pong": target_id,
+                            "time": target_time,
+                        }))
+                        continue
+
+                    inner_data = data.get("data", None)
+                    if isinstance(inner_data, dict):
+                        if data.get("dataType", None) == "swap.market.v2.contracts":
+                            list_data = inner_data.get("l", None)
+                            await self.__last_candle_lock.acquire()
+                            for current in list_data:
+                                info = SingleCandleInfo.deserialize_short(current)
+                                if info:
+                                    self.__last_candle_storage[info.pair.lower()] = info
+                            self.__last_candle_lock.release()
+                        continue
+
+                    logger.info(f"we got some unknown data: {data}")
+                except Exception as ex:
+                    logger.info(f"failed to handle ws message from exchange: {msg}; {ex}")
+    
+    async def get_last_candle(self, pair: str) -> SingleCandleInfo:
+        """
+        Returns the last candle's info in this exchange.
+        This method is safe to be called ONLY from the exact same thread
+        that the loop is currently operating on.
+        """
+        await self.__last_candle_lock.acquire()
+        info = self.__last_candle_storage.get(pair.lower())
+        self.__last_candle_lock.release()
+        return info
+    
     # endregion
     ###########################################################
     # region contract
+    async def get_contract_config(
+        self,
+        fund_type: int, # e.g. 1
+        coin_name: str, # e.g. "SOL"
+        valuation_name: str, # e.g. "USDT"
+        margin_coin_name: str, # e.g. "USDT"
+    ) -> ContractConfigResponse:
+        params = {
+            "fundType": f"{fund_type}",
+            "coinName": f"{coin_name}",
+            "valuationName": f"{valuation_name}",
+            "marginCoinName": f"{margin_coin_name}",
+        }
+        headers = self.get_headers(
+            payload=params,
+        )
+        return await self.invoke_get(
+            # "https://bingx.com/api/v2/contract/config",
+            f"{self.qq_os_base_url}/v2/contract/config",
+            headers=headers,
+            params=params,
+            model_type=CopyTraderTradePositionsResponse,
+        )
+        
     async def get_contract_list(
         self,
         quotation_coin_id: int = -1,
@@ -403,6 +547,60 @@ class BXUltraClient(ExchangeBase):
 
     # endregion
     ###########################################################
+    # region contract delegation
+    async def create_order_delegation(
+        self,
+        balance_direction: int,  # e.g. 1
+        delegate_price: Decimal,  # e.g. 107414.70
+        fund_type: int,  # # e.g. 1
+        large_spread_rate: int,  # e.g. 0
+        lever_times: int,  # e.g. 5
+        margin: int,  # e.g. 5
+        margin_coin_name: str,  # e.g. "USDT"
+        market_factor: int,  # e.g. 1
+        order_type: int,  # e.g. 0
+        price: Decimal,  # the current price of the market??
+        stop_loss_rate: int,  # e.g. -1
+        stop_profit_rate: int,  # e.g. -1
+        quotation_coin_id: int,  # e.g. 1
+        spread_rate: float,  # something very low. e.g. 0.00003481
+        stop_loss_price: float,  # e.g. -1
+        stop_profit_price: float,  # e.g. -1
+        up_ratio: Decimal,  # e.g. 0.5
+    ) -> CreateOrderDelegationResponse:
+        payload = {
+            "balanceDirection": balance_direction,
+            "delegatePrice": f"{delegate_price}",
+            "fundType": fund_type,
+            "largeSpreadRate": large_spread_rate or 0,
+            "leverTimes": lever_times or 1,
+            "margin": margin,
+            "marginCoinName": margin_coin_name or "USDT",
+            "marketFactor": market_factor or 1,
+            "orderType": f"{order_type or 0}",
+            "price": float(price), # e.g. 107161.27
+            "profitLossRateDto": {
+                "stopProfitRate": stop_profit_rate or -1,
+                "stopLossRate": stop_loss_rate or -1,
+            },
+            "quotationCoinId": quotation_coin_id or 1,
+            "spreadRate": float(spread_rate) or 0.00003481,
+            "stopLossPrice": stop_loss_price or -1,
+            "stopProfitPrice": stop_profit_price or -1,
+            "upRatio": f"{0.5 if up_ratio is None else up_ratio}",
+        }
+        headers = self.get_headers(
+            needs_auth=True,
+            payload=payload,
+        )
+        return await self.invoke_post(
+            f"{self.we_api_base_url}/v2/contract/order/delegation",
+            headers=headers,
+            content=payload,
+            model_type=CreateOrderDelegationResponse,
+        )
+    # endregion
+    ###########################################################
     # region copy-trade-facade
     async def get_copy_trader_positions(
         self,
@@ -505,6 +703,7 @@ class BXUltraClient(ExchangeBase):
         self,
         uid: int | str,
     ) -> int | str:
+        global user_api_identity_cache
         api_identity = user_api_identity_cache.get(uid, None)
         if not api_identity:
             resume = await self.get_copy_trader_resume(
@@ -595,6 +794,8 @@ class BXUltraClient(ExchangeBase):
         self.authorization_token = json_data.get(
             "authorization_token", self.authorization_token
         )
+        if self.authorization_token:
+            self.jwt_manager = JWTManager(self.jwt_manager)
         self.app_id = json_data.get("app_id", self.app_id)
         self.trade_env = json_data.get("trade_env", self.trade_env)
         self.timezone = json_data.get("timezone", self.timezone)
@@ -608,6 +809,9 @@ class BXUltraClient(ExchangeBase):
         """
         Saves current information to the session file.
         """
+        if file_path is None:
+            file_path = f"{self.sessions_dir}/{self.account_name}.bx"
+
         if not self.device_id:
             self.device_id = uuid.uuid4().hex.replace("-", "") + "##"
 
@@ -645,8 +849,6 @@ class BXUltraClient(ExchangeBase):
         self,
         uid: int | str,
     ) -> UnifiedTraderPositions:
-        global user_api_identity_cache
-
         api_identity = await self.get_trader_api_identity(
             uid=uid,
         )
@@ -656,6 +858,13 @@ class BXUltraClient(ExchangeBase):
             api_identity=api_identity,
             page_size=50,  # TODO: make this dynamic I guess...
         )
+        if not result.data:
+            if result.msg:
+                raise ExchangeError(result.msg)
+            raise ExchangeError(
+                f"Unknown error happened while fetching positions of {uid}, "
+                f"code: {result.code}"
+            )
         if result.data.hide == 0 and not result.data.positions:
             # TODO: do proper exceptions here...
             raise ValueError("The trader has made their positions hidden")
@@ -674,6 +883,12 @@ class BXUltraClient(ExchangeBase):
             unified_pos.open_price_unit = (
                 position.valuation_coin_name or position.symbol.split("-")[-1]
             )  # TODO
+
+            last_candle = await self.get_last_candle(unified_pos.position_pair)
+            if last_candle:
+                unified_pos.last_price = last_candle.close_price
+                unified_pos.last_volume = last_candle.quote_volume
+
             unified_result.positions.append(unified_pos)
 
         return unified_result

trd_utils-0.0.30/trd_utils/exchanges/errors.py

@@ -0,0 +1,10 @@
+
+
+
+class ExchangeError(RuntimeError):
+    """
+    Specifies an error coming from the exchange.
+    """
+    def __init__(self, *args):
+        super().__init__(*args)
+

{trd_utils-0.0.28 → trd_utils-0.0.30}/trd_utils/exchanges/exchange_base.py

@@ -1,13 +1,38 @@
+import asyncio
 from decimal import Decimal
 import json
+import logging
 from typing import Type
 from abc import ABC
 
+import base64
+import time
+
 import httpx
+from websockets.asyncio.connection import Connection as WSConnection
 
 from trd_utils.exchanges.base_types import UnifiedTraderInfo, UnifiedTraderPositions
 from trd_utils.types_helper.base_model import BaseModel
 
+logger = logging.getLogger(__name__)
+
+class JWTManager():
+    _jwt_string: str = None
+
+    def __init__(self, jwt_string: str):
+        self._jwt_string = jwt_string
+        try:
+            payload_b64 = self._jwt_string.split('.')[1]
+            payload_bytes = base64.urlsafe_b64decode(payload_b64 + '==')
+            self.payload = json.loads(payload_bytes)
+        except Exception:
+            self.payload = {}
+
+    def is_expired(self):
+        if "exp" not in self.payload:
+            return False
+        
+        return time.time() > self.payload["exp"]
 
 class ExchangeBase(ABC):
     ###########################################################
@@ -22,11 +47,23 @@ class ExchangeBase(ABC):
     device_id: str = None
     trace_id: str = None
     app_version: str = "4.28.3"
+    x_router_tag: str = "gray-develop"
     platform_id: str = "10"
     install_channel: str = "officialAPK"
     channel_header: str = "officialAPK"
 
+    jwt_manager: JWTManager = None
+
     _fav_letter: str = "^"
+
+    # the lock for internal operations.
+    _internal_lock: asyncio.Lock = None
+
+    # extra tasks to be cancelled when the client closes.
+    extra_tasks: list[asyncio.Task] = None
+
+    # the ws connections to be closed when this client is closed.
+    ws_connections: list[WSConnection] = None
     # endregion
     ###########################################################
 
@@ -169,9 +206,32 @@ class ExchangeBase(ABC):
             # Now parse the decompressed content
             return json.loads(content.decode("utf-8"), parse_float=parse_float)
 
+    async def __aenter__(self):
+        return self
+
+    async def __aexit__(
+        self,
+        exc_type=None,
+        exc_value=None,
+        traceback=None,
+    ) -> None:
+        await self.aclose()
+
     async def aclose(self) -> None:
+        await self._internal_lock.acquire()
         await self.httpx_client.aclose()
 
+        if self.ws_connections:
+            for current in self.ws_connections:
+                try:
+                    await current.close()
+                except Exception as ex:
+                    logger.warning(f"failed to close ws connection: {ex}")
+                    continue
+            self.ws_connections = []
+
+        self._internal_lock.release()
+
     # endregion
     ###########################################################
     # region data-files related methods

trd_utils-0.0.28/trd_utils/__init__.py

@@ -1,3 +0,0 @@
-
-__version__ = "0.0.28"
-

trd_utils-0.0.28/trd_utils/common_utils/float_utils.py

@@ -1,11 +0,0 @@
-
-from decimal import Decimal
-
-
-default_quantize = Decimal("1.00")
-
-def dec_to_str(dec_value: Decimal) -> str:
-    return format(dec_value.quantize(default_quantize), "f")
-
-def dec_to_normalize(dec_value: Decimal) -> str:
-    return format(dec_value.normalize(), "f")