trd-utils 0.0.4__tar.gz → 0.0.6__tar.gz

{trd_utils-0.0.4 → trd_utils-0.0.6}/LICENSE

@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2024 ALi.w
+Copyright (c) 2024-2025 ALi.w
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal

{trd_utils-0.0.4 → trd_utils-0.0.6}/PKG-INFO

@@ -1,8 +1,7 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.3
 Name: trd_utils
-Version: 0.0.4
+Version: 0.0.6
 Summary: Common Basic Utils for Python3. By ALiwoto.
-Home-page: https://github.com/ALiwoto/trd_utils
 Keywords: utils,trd_utils,basic-utils,common-utils
 Author: ALiwoto
 Author-email: aminnimaj@gmail.com
@@ -20,9 +19,20 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
 Requires-Dist: cryptography (>=41.0.7)
 Requires-Dist: httpx (>=0.21.0)
+Project-URL: Homepage, https://github.com/ALiwoto/trd_utils
 Description-Content-Type: text/markdown
 
 # Trd Utils
 
 Basic common utils for Python.
 
+## How to run tests
+
+Use this command first:
+
+```bash
+pip install -e .
+```
+
+Then run the tests in vscode.
+

trd_utils-0.0.6/README.md

@@ -0,0 +1,13 @@
+# Trd Utils
+
+Basic common utils for Python.
+
+## How to run tests
+
+Use this command first:
+
+```bash
+pip install -e .
+```
+
+Then run the tests in vscode.

{trd_utils-0.0.4 → trd_utils-0.0.6}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "trd_utils"
-version = "0.0.4"
+version = "0.0.6"
 description = "Common Basic Utils for Python3. By ALiwoto."
 authors = ["ALiwoto <aminnimaj@gmail.com>"]
 packages = [

trd_utils-0.0.6/trd_utils/__init__.py

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

trd_utils-0.0.6/trd_utils/common_utils/float_utils.py

@@ -0,0 +1,11 @@
+
+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")

trd_utils-0.0.6/trd_utils/exchanges/__init__.py

@@ -0,0 +1,11 @@
+
+from .exchange_base import ExchangeBase
+from .blofin import BlofinClient
+from .bx_ultra import BXUltraClient
+
+
+__all__ = [
+    ExchangeBase,
+    BXUltraClient,
+    BlofinClient,
+]

trd_utils-0.0.6/trd_utils/exchanges/blofin/__init__.py

@@ -0,0 +1,6 @@
+
+from .blofin_client import BlofinClient
+
+__all__ = [
+    BlofinClient,
+]

trd_utils-0.0.6/trd_utils/exchanges/blofin/blofin_client.py

@@ -0,0 +1,238 @@
+from decimal import Decimal
+import json
+import logging
+from typing import Type
+import httpx
+
+import time
+from pathlib import Path
+
+from trd_utils.exchanges.blofin.blofin_types import (
+    BlofinApiResponse,
+    CmsColorResponse,
+    CopyTraderInfoResponse,
+    CopyTraderOrderHistoryResponse,
+    CopyTraderOrderListResponse,
+    ShareConfigResponse,
+)
+from trd_utils.cipher import AESCipher
+from trd_utils.exchanges.exchange_base import ExchangeBase
+
+logger = logging.getLogger(__name__)
+
+
+class BlofinClient(ExchangeBase):
+    ###########################################################
+    # region client parameters
+    blofin_api_base_host: str = "https://\u0062lofin.co\u006d"
+    blofin_api_base_url: str = "https://\u0062lofin.co\u006d/uapi/v1"
+    origin_header: str = "https://\u0062lofin.co\u006d"
+
+    timezone: str = "Etc/UTC"
+
+    # endregion
+    ###########################################################
+    # region client constructor
+    def __init__(
+        self,
+        account_name: str = "default",
+        http_verify: bool = True,
+        fav_letter: str = "^",
+        read_session_file: bool = True,
+        sessions_dir: str = "sessions",
+    ):
+        self.httpx_client = httpx.AsyncClient(
+            verify=http_verify,
+            http2=True,
+            http1=False,
+        )
+        self.account_name = account_name
+        self._fav_letter = fav_letter
+        self.sessions_dir = sessions_dir
+
+        if read_session_file:
+            self.read_from_session_file(f"{sessions_dir}/{self.account_name}.bf")
+
+    # endregion
+    ###########################################################
+    # region v1/cms/
+    async def get_share_config(self) -> ShareConfigResponse:
+        headers = self.get_headers()
+        return await self.invoke_get(
+            f"{self.blofin_api_base_url}/cms/share_config",
+            headers=headers,
+            model=ShareConfigResponse,
+        )
+
+    async def get_cms_color(self) -> CmsColorResponse:
+        headers = self.get_headers()
+        return await self.invoke_get(
+            f"{self.blofin_api_base_url}/cms/color",
+            headers=headers,
+            model=CmsColorResponse,
+        )
+
+    # endregion
+    ###########################################################
+    # region copy/trader
+    async def get_copy_trader_info(self, uid: int) -> CopyTraderInfoResponse:
+        payload = {
+            "uid": uid,
+        }
+        headers = self.get_headers()
+        return await self.invoke_post(
+            f"{self.blofin_api_base_url}/copy/trader/info",
+            headers=headers,
+            content=payload,
+            model=CopyTraderInfoResponse,
+        )
+
+    async def get_copy_trader_order_list(
+        self,
+        from_param: int,
+        limit_param: 0,
+        uid: int,
+    ) -> CopyTraderOrderListResponse:
+        payload = {
+            "from": from_param,
+            "limit": limit_param,
+            "uid": uid,
+        }
+        headers = self.get_headers()
+        return await self.invoke_post(
+            f"{self.blofin_api_base_url}/copy/trader/order/list",
+            headers=headers,
+            content=payload,
+            model=CopyTraderOrderListResponse,
+        )
+    
+    async def get_copy_trader_order_history(
+        self,
+        from_param: int,
+        limit_param: 0,
+        uid: int,
+    ) -> CopyTraderOrderHistoryResponse:
+        payload = {
+            "from": from_param,
+            "limit": limit_param,
+            "uid": uid,
+        }
+        headers = self.get_headers()
+        return await self.invoke_post(
+            f"{self.blofin_api_base_url}/copy/trader/order/history",
+            headers=headers,
+            content=payload,
+            model=CopyTraderOrderHistoryResponse,
+        )
+
+    # endregion
+    ###########################################################
+    # region client helper methods
+    def get_headers(self, payload=None, needs_auth: bool = False) -> dict:
+        the_timestamp = int(time.time() * 1000)
+        the_headers = {
+            # "Host": self.blofin_api_base_host,
+            "Content-Type": "application/json",
+            "Accept": "application/json",
+            "Origin": self.origin_header,
+            "X-Tz": self.timezone,
+            "Fp-Request-Id": f"{the_timestamp}.n1fDrN",
+            "Accept-Encoding": "gzip, deflate, br, zstd",
+            "User-Agent": self.user_agent,
+            "Connection": "close",
+            "appsiteid": "0",
+        }
+
+        if self.x_requested_with:
+            the_headers["X-Requested-With"] = self.x_requested_with
+
+        if needs_auth:
+            the_headers["Authorization"] = f"Bearer {self.authorization_token}"
+        return the_headers
+
+    async def invoke_get(
+        self,
+        url: str,
+        headers: dict | None = None,
+        params: dict | None = None,
+        model: Type[BlofinApiResponse] | None = None,
+        parse_float=Decimal,
+    ) -> "BlofinApiResponse":
+        """
+        Invokes the specific request to the specific url with the specific params and headers.
+        """
+        response = await self.httpx_client.get(
+            url=url,
+            headers=headers,
+            params=params,
+        )
+        return model.deserialize(response.json(parse_float=parse_float))
+
+    async def invoke_post(
+        self,
+        url: str,
+        headers: dict | None = None,
+        params: dict | None = None,
+        content: dict | str | bytes = "",
+        model: Type[BlofinApiResponse] | None = None,
+        parse_float=Decimal,
+    ) -> "BlofinApiResponse":
+        """
+        Invokes the specific request to the specific url with the specific params and headers.
+        """
+
+        if isinstance(content, dict):
+            content = json.dumps(content, separators=(",", ":"), sort_keys=True)
+
+        response = await self.httpx_client.post(
+            url=url,
+            headers=headers,
+            params=params,
+            content=content,
+        )
+        if not model:
+            return response.json()
+
+        return model.deserialize(response.json(parse_float=parse_float))
+
+    async def aclose(self) -> None:
+        await self.httpx_client.aclose()
+        logger.info("BlofinClient closed")
+        return True
+
+    def read_from_session_file(self, file_path: str) -> None:
+        """
+        Reads from session file; if it doesn't exist, creates it.
+        """
+        # check if path exists
+        target_path = Path(file_path)
+        if not target_path.exists():
+            return self._save_session_file(file_path=file_path)
+
+        aes = AESCipher(key=f"bf_{self.account_name}_bf", fav_letter=self._fav_letter)
+        content = aes.decrypt(target_path.read_text()).decode("utf-8")
+        json_data: dict = json.loads(content)
+
+        self.authorization_token = json_data.get(
+            "authorization_token",
+            self.authorization_token,
+        )
+        self.timezone = json_data.get("timezone", self.timezone)
+        self.user_agent = json_data.get("user_agent", self.user_agent)
+
+    def _save_session_file(self, file_path: str) -> None:
+        """
+        Saves current information to the session file.
+        """
+
+        json_data = {
+            "authorization_token": self.authorization_token,
+            "timezone": self.timezone,
+            "user_agent": self.user_agent,
+        }
+        aes = AESCipher(key=f"bf_{self.account_name}_bf", fav_letter=self._fav_letter)
+        target_path = Path(file_path)
+        target_path.write_text(aes.encrypt(json.dumps(json_data)))
+
+    # endregion
+    ###########################################################

trd_utils-0.0.6/trd_utils/exchanges/blofin/blofin_types.py

@@ -0,0 +1,144 @@
+# from typing import Any, Optional
+# from decimal import Decimal
+# from datetime import datetime, timedelta
+# import pytz
+
+from decimal import Decimal
+from typing import Any
+from trd_utils.types_helper import BaseModel
+
+# from trd_utils.common_utils.float_utils import (
+#     dec_to_str,
+#     dec_to_normalize,
+# )
+
+
+
+class BlofinApiResponse(BaseModel):
+    code: int = None
+    timestamp: int = None
+    msg: str = None
+
+    def __str__(self):
+        return f"code: {self.code}; timestamp: {self.timestamp}"
+
+    def __repr__(self):
+        return f"code: {self.code}; timestamp: {self.timestamp}"
+
+
+###########################################################
+
+class PnlShareListInfo(BaseModel):
+    background_color: str = None
+    background_img_up: str = None
+    background_img_down: str = None
+
+class ShareConfigResult(BaseModel):
+    pnl_share_list: list[PnlShareListInfo] = None
+
+class ShareConfigResponse(BlofinApiResponse):
+    data: ShareConfigResult = None
+
+class CmsColorResult(BaseModel):
+    color: str = None
+    city: str = None
+    country: str = None
+    ip: str = None
+
+class CmsColorResponse(BlofinApiResponse):
+    data: CmsColorResult = None
+
+###########################################################
+
+
+class CopyTraderInfoResult(BaseModel):
+    aum: str = None
+    can_copy: bool = None
+    copier_whitelist: bool = None
+    follow_state: int = None
+    followers: int = None
+    followers_max: int = None
+    forbidden_follow_type: int = None
+    hidden_all: bool = None
+    hidden_order: bool = None
+    joined_date: int = None
+    max_draw_down: Decimal = None
+    nick_name: str = None
+    order_amount_limit: None
+    profile: str = None
+    profit_sharing_ratio: Decimal = None
+    real_pnl: Decimal = None
+    roi_d7: Decimal = None
+    self_introduction: str = None
+    sharing_period: str = None
+    source: int = None
+    uid: int = None
+    whitelist_copier: bool = None
+    win_rate: Decimal = None
+
+    def get_profile_url(self) -> str:
+        return f"https://blofin.com/copy-trade/details/{self.uid}"
+
+class CopyTraderInfoResponse(BlofinApiResponse):
+    data: CopyTraderInfoResult = None
+
+class CopyTraderSingleOrderInfo(BaseModel):
+    id: int = None
+    symbol: str = None
+    leverage: int = None
+    order_side: str = None
+    avg_open_price: str = None
+    quantity: str = None
+    quantity_cont: None
+    open_time: int = None
+    close_time: Any = None
+    avg_close_price: Decimal = None
+    real_pnl: Any = None
+    close_type: Any = None
+    roe: Decimal = None
+    followers_profit: Decimal = None
+    followers: Any = None
+    order_id: Any = None
+    sharing: Any = None
+    order_state: None
+    trader_name: None
+    mark_price: None
+    tp_trigger_price: None
+    tp_order_type: None
+    sl_trigger_price: None
+    sl_order_type: None
+    margin_mode: str = None
+    time_in_force: None
+    position_side: str = None
+    order_category: None
+    price: None
+    fill_quantity: None
+    fill_quantity_cont: None
+    pnl: None
+    cancel_source: None
+    order_type: None
+    order_open_state: None
+    amount: None
+    filled_amount: None
+    create_time: None
+    update_time: None
+    open_fee: None
+    close_fee: None
+    id_md5: None
+    tp_sl: None
+    trader_uid: None
+    available_quantity: None
+    available_quantity_cont: None
+    show_in_kline: None
+    unrealized_pnl: None
+    unrealized_pnl_ratio: None
+    broker_id: None
+    position_change_history: None
+    user_id: None
+
+class CopyTraderOrderListResponse(BlofinApiResponse):
+    data: list[CopyTraderSingleOrderInfo] = None
+
+
+class CopyTraderOrderHistoryResponse(BlofinApiResponse):
+    data: list[CopyTraderSingleOrderInfo] = None

{trd_utils-0.0.4/trd_utils → trd_utils-0.0.6/trd_utils/exchanges}/bx_ultra/bx_types.py

@@ -1,10 +1,11 @@
 from typing import Any, Optional
-from ..types_helper import BaseModel
 from decimal import Decimal
 from datetime import datetime, timedelta
 import pytz
 
-from .common_utils import (
+from trd_utils.types_helper import BaseModel
+
+from trd_utils.common_utils.float_utils import (
     dec_to_str,
     dec_to_normalize,
 )

{trd_utils-0.0.4/trd_utils → trd_utils-0.0.6/trd_utils/exchanges}/bx_ultra/bx_ultra_client.py

@@ -4,6 +4,7 @@ import asyncio
 from decimal import Decimal
 import json
 import logging
+from typing import Type
 import uuid
 
 import httpx
@@ -11,8 +12,8 @@ import httpx
 import time
 from pathlib import Path
 
-from .common_utils import do_ultra_ss
-from .bx_types import (
+from trd_utils.exchanges.bx_ultra.bx_utils import do_ultra_ss
+from trd_utils.exchanges.bx_ultra.bx_types import (
     AssetsInfoResponse,
     ContractOrdersHistoryResponse,
     ContractsListResponse,
@@ -26,8 +27,11 @@ from .bx_types import (
     UserFavoriteQuotationResponse,
     ZenDeskABStatusResponse,
     ZoneModuleListResponse,
+    BxApiResponse,
 )
-from ..cipher import AESCipher
+from trd_utils.cipher import AESCipher
+
+from trd_utils.exchanges.exchange_base import ExchangeBase
 
 PLATFORM_ID_ANDROID = "10"
 PLATFORM_ID_WEB = "30"
@@ -44,25 +48,16 @@ TG_APP_VERSION = "5.0.15"
 logger = logging.getLogger(__name__)
 
 
-class BXUltraClient:
+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"
-
     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"
 
-    device_id: str = None
-    trace_id: str = None
-    app_version: str = "4.28.3"
-    platform_id: str = "10"
-    install_channel: str = "officialAPK"
-    channel_header: str = "officialAPK"
     origin_header: str = "https://\u0062ing\u0078.co\u006d"
-    authorization_token: str = None
     app_id: str = "30004"
     main_app_id: str = "10009"
     trade_env: str = "real"
@@ -71,12 +66,6 @@ class BXUltraClient:
     device_brand: str = "SM-N976N"
     platform_lang: str = "en"
     sys_lang: str = "en"
-    user_agent: str = "okhttp/4.12.0"
-    x_requested_with: str = None
-    httpx_client: httpx.AsyncClient = None
-    account_name: str = "default"
-
-    _fav_letter: str = "^"
 
     # endregion
     ###########################################################
@@ -89,6 +78,7 @@ class BXUltraClient:
         app_version: str = ANDROID_APP_VERSION,
         http_verify: bool = True,
         fav_letter: str = "^",
+        sessions_dir: str = "sessions",
     ):
         self.httpx_client = httpx.AsyncClient(
             verify=http_verify, http2=True, http1=False
@@ -99,13 +89,15 @@ class BXUltraClient:
         self.app_version = app_version
         self._fav_letter = fav_letter
 
-        self.read_from_session_file(f"{self.account_name}.bx")
+        self.read_from_session_file(f"{sessions_dir}/{self.account_name}.bx")
 
     # endregion
     ###########################################################
     # region api/coin/v1
     async def get_zone_module_info(
-        self, only_one_position: int = 0, biz_type: int = 10
+        self,
+        only_one_position: int = 0,
+        biz_type: int = 10,
     ) -> ZoneModuleListResponse:
         """
         Fetches and returns zone module info from the API.
@@ -118,101 +110,113 @@ class BXUltraClient:
         }
         headers = self.get_headers(params)
         headers["Only_one_position"] = f"{only_one_position}"
-        response = await self.httpx_client.get(
+        return await self.invoke_get(
             f"{self.we_api_base_url}/coin/v1/zone/module-info",
             headers=headers,
             params=params,
+            model=ZoneModuleListResponse,
         )
-        return ZoneModuleListResponse.deserialize(response.json(parse_float=Decimal))
 
     async def get_user_favorite_quotation(
-        self, only_one_position: int = 0, biz_type: int = 1
-    ):
+        self,
+        only_one_position: int = 0,
+        biz_type: int = 1,
+    ) -> UserFavoriteQuotationResponse:
         params = {
             "bizType": f"{biz_type}",
         }
         headers = self.get_headers(params)
         headers["Only_one_position"] = f"{only_one_position}"
-        response = await self.httpx_client.get(
+        return await self.invoke_get(
             f"{self.we_api_base_url}/coin/v1/user/favorite/quotation",
             headers=headers,
             params=params,
-        )
-        return UserFavoriteQuotationResponse.deserialize(
-            response.json(parse_float=Decimal)
+            model=UserFavoriteQuotationResponse,
         )
 
-    async def get_quotation_rank(self, only_one_position: int = 0, order_flag: int = 0):
+    async def get_quotation_rank(
+        self,
+        only_one_position: int = 0,
+        order_flag: int = 0,
+    ) -> QuotationRankResponse:
         params = {
             "orderFlag": f"{order_flag}",
         }
         headers = self.get_headers(params)
         headers["Only_one_position"] = f"{only_one_position}"
-        response = await self.httpx_client.get(
+        return await self.invoke_get(
             f"{self.we_api_base_url}/coin/v1/rank/quotation-rank",
             headers=headers,
             params=params,
+            model=QuotationRankResponse,
         )
-        return QuotationRankResponse.deserialize(response.json(parse_float=Decimal))
 
-    async def get_hot_search(self, only_one_position: int = 0, biz_type: int = 30):
+    async def get_hot_search(
+        self,
+        only_one_position: int = 0,
+        biz_type: int = 30,
+    ) -> HotSearchResponse:
         params = {
             "bizType": f"{biz_type}",
         }
         headers = self.get_headers(params)
         headers["Only_one_position"] = f"{only_one_position}"
-        response = await self.httpx_client.get(
+        return await self.invoke_get(
             f"{self.we_api_base_url}/coin/v1/quotation/hot-search",
             headers=headers,
             params=params,
+            model=HotSearchResponse,
         )
-        return HotSearchResponse.deserialize(response.json(parse_float=Decimal))
 
-    async def get_homepage(self, only_one_position: int = 0, biz_type: int = 30):
+    async def get_homepage(
+        self,
+        only_one_position: int = 0,
+        biz_type: int = 30,
+    ) -> HomePageResponse:
         params = {
             "biz-type": f"{biz_type}",
         }
         headers = self.get_headers(params)
         headers["Only_one_position"] = f"{only_one_position}"
-        response = await self.httpx_client.get(
+        return await self.invoke_get(
             f"{self.we_api_base_url}/coin/v1/discovery/homepage",
             headers=headers,
             params=params,
+            model=HomePageResponse,
         )
-        return HomePageResponse.deserialize(response.json(parse_float=Decimal))
 
     # endregion
     ###########################################################
     # region customer
-    async def get_zendesk_ab_status(self):
+    async def get_zendesk_ab_status(self) -> ZenDeskABStatusResponse:
         headers = self.get_headers()
-        response = await self.httpx_client.get(
+        return await self.invoke_get(
             f"{self.we_api_base_url}/customer/v1/zendesk/ab-status",
             headers=headers,
+            model=ZenDeskABStatusResponse,
         )
-        return ZenDeskABStatusResponse.deserialize(response.json(parse_float=Decimal))
 
     # endregion
     ###########################################################
     # region platform-tool
     async def get_hint_list(self) -> HintListResponse:
         headers = self.get_headers()
-        response = await self.httpx_client.get(
+        return await self.invoke_get(
             f"{self.we_api_base_url}/platform-tool/v1/hint/list",
             headers=headers,
+            model=HintListResponse,
         )
-        return HintListResponse.deserialize(response.json(parse_float=Decimal))
 
     # endregion
     ###########################################################
     # region asset-manager
     async def get_assets_info(self) -> AssetsInfoResponse:
         headers = self.get_headers(needs_auth=True)
-        response = await self.httpx_client.get(
+        return await self.invoke_get(
             f"{self.we_api_base_url}/asset-manager/v1/assets/account-total-overview",
             headers=headers,
+            model=AssetsInfoResponse,
         )
-        return AssetsInfoResponse.deserialize(response.json(parse_float=Decimal))
 
     # endregion
     ###########################################################
@@ -236,12 +240,12 @@ class BXUltraClient:
         if margin_coin_name:
             params["marginCoinName"] = margin_coin_name
         headers = self.get_headers(params, needs_auth=True)
-        response = await self.httpx_client.get(
+        return await self.invoke_get(
             f"{self.we_api_base_url}/v4/contract/order/hold",
             headers=headers,
             params=params,
+            model=ContractsListResponse,
         )
-        return ContractsListResponse.deserialize(response.json(parse_float=Decimal))
 
     async def get_contract_order_history(
         self,
@@ -263,13 +267,11 @@ class BXUltraClient:
             params["fromOrderNo"] = f"{from_order_no}"
 
         headers = self.get_headers(params, needs_auth=True)
-        response = await self.httpx_client.get(
+        return await self.invoke_get(
             f"{self.we_api_base_url}/v2/contract/order/history",
             headers=headers,
             params=params,
-        )
-        return ContractOrdersHistoryResponse.deserialize(
-            response.json(parse_float=Decimal)
+            model=ContractOrdersHistoryResponse,
         )
 
     async def get_today_contract_earnings(
@@ -282,7 +284,7 @@ class BXUltraClient:
         """
         Fetches today's earnings from the contract orders.
         NOTE: This method is a bit slow due to the API rate limiting.
-        NOTE: If the user has not opened ANY contract orders today, 
+        NOTE: If the user has not opened ANY contract orders today,
             this method will return None.
         """
         return await self._get_period_contract_earnings(
@@ -303,7 +305,7 @@ class BXUltraClient:
         """
         Fetches this week's earnings from the contract orders.
         NOTE: This method is a bit slow due to the API rate limiting.
-        NOTE: If the user has not opened ANY contract orders this week, 
+        NOTE: If the user has not opened ANY contract orders this week,
             this method will return None.
         """
         return await self._get_period_contract_earnings(
@@ -324,7 +326,7 @@ class BXUltraClient:
         """
         Fetches this month's earnings from the contract orders.
         NOTE: This method is a bit slow due to the API rate limiting.
-        NOTE: If the user has not opened ANY contract orders this week, 
+        NOTE: If the user has not opened ANY contract orders this week,
             this method will return None.
         """
         return await self._get_period_contract_earnings(
@@ -368,11 +370,11 @@ class BXUltraClient:
             if result.get_orders_len() < page_size:
                 break
             await asyncio.sleep(delay_per_fetch)
-        
+
         if not has_earned_any:
             return None
         return total_earnings
-    
+
     # endregion
     ###########################################################
     # region copy-trade-facade
@@ -392,13 +394,11 @@ class BXUltraClient:
             "copyTradeLabelType": f"{copy_trade_label_type}",
         }
         headers = self.get_headers(params)
-        response = await self.httpx_client.get(
+        return await self.invoke_get(
             f"{self.we_api_base_url}/copy-trade-facade/v2/real/trader/positions",
             headers=headers,
             params=params,
-        )
-        return CopyTraderTradePositionsResponse.deserialize(
-            response.json(parse_float=Decimal)
+            model=CopyTraderTradePositionsResponse,
         )
 
     async def search_copy_traders(
@@ -430,25 +430,25 @@ class BXUltraClient:
             "nickName": nick_name,
         }
         headers = self.get_headers(payload)
-        response = await self.httpx_client.post(
+        return await self.invoke_post(
             f"{self.we_api_base_url}/v6/copy-trade/search/search",
             headers=headers,
             params=params,
-            content=json.dumps(payload, separators=(",", ":"), sort_keys=True),
+            content=payload,
+            model=SearchCopyTradersResponse,
         )
-        return SearchCopyTradersResponse.deserialize(response.json(parse_float=Decimal))
 
     # endregion
     ###########################################################
     # region welfare
     async def do_daily_check_in(self):
         headers = self.get_headers(needs_auth=True)
-        response = await self.httpx_client.post(
+        return await self.invoke_post(
             f"{self.original_base_host}/api/act-operation/v1/welfare/sign-in/do",
             headers=headers,
             content="",
+            model=None,
         )
-        return response.json()
 
     # endregion
     ###########################################################
@@ -484,7 +484,7 @@ class BXUltraClient:
                 payload_data=payload,
             ),
             "Timestamp": f"{the_timestamp}",
-            # 'Accept-Encoding': 'gzip, deflate',
+            'Accept-Encoding': 'gzip, deflate',
             "User-Agent": self.user_agent,
             "Connection": "close",
             "appsiteid": "0",
@@ -497,6 +497,51 @@ class BXUltraClient:
             the_headers["Authorization"] = f"Bearer {self.authorization_token}"
         return the_headers
 
+    async def invoke_get(
+        self,
+        url: str,
+        headers: dict | None = None,
+        params: dict | None = None,
+        model: Type[BxApiResponse] | None = None,
+        parse_float=Decimal,
+    ) -> "BxApiResponse":
+        """
+        Invokes the specific request to the specific url with the specific params and headers.
+        """
+        response = await self.httpx_client.get(
+            url=url,
+            headers=headers,
+            params=params,
+        )
+        return model.deserialize(response.json(parse_float=parse_float))
+
+    async def invoke_post(
+        self,
+        url: str,
+        headers: dict | None = None,
+        params: dict | None = None,
+        content: dict | str | bytes = "",
+        model: Type[BxApiResponse] | None = None,
+        parse_float=Decimal,
+    ) -> "BxApiResponse":
+        """
+        Invokes the specific request to the specific url with the specific params and headers.
+        """
+
+        if isinstance(content, dict):
+            content = json.dumps(content, separators=(",", ":"), sort_keys=True)
+
+        response = await self.httpx_client.post(
+            url=url,
+            headers=headers,
+            params=params,
+            content=content,
+        )
+        if not model:
+            return response.json()
+
+        return model.deserialize(response.json(parse_float=parse_float))
+
     async def aclose(self) -> None:
         await self.httpx_client.aclose()
         logger.info("BXUltraClient closed")

trd_utils-0.0.4/trd_utils/bx_ultra/common_utils.py → trd_utils-0.0.6/trd_utils/exchanges/bx_ultra/bx_utils.py

@@ -1,9 +1,7 @@
 import hashlib
 import json
 import uuid
-from decimal import Decimal
 
-default_quantize = Decimal("1.00")
 
 default_e: str = (
     "\u0039\u0035\u0064\u0036\u0035\u0063\u0037\u0033\u0064\u0063\u0035"
@@ -17,12 +15,6 @@ long_accept_header1: str = (
     + "image/webp,image/apng,*/*;q=0.8,application/signed-exchange;v=b3;q=0.7"
 )
 
-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 do_ultra_ss(
     e_param: str,
     se_param: str,

trd_utils-0.0.6/trd_utils/exchanges/exchange_base.py

@@ -0,0 +1,76 @@
+
+from decimal import Decimal
+from typing import Any
+from abc import ABC
+
+import httpx
+
+
+class ExchangeBase(ABC):
+    ###########################################################
+    # region client parameters
+    user_agent: str = "okhttp/4.12.0"
+    x_requested_with: str = None
+    httpx_client: httpx.AsyncClient = None
+    account_name: str = "default"
+    sessions_dir: str = "sessions"
+
+    authorization_token: str = None
+    device_id: str = None
+    trace_id: str = None
+    app_version: str = "4.28.3"
+    platform_id: str = "10"
+    install_channel: str = "officialAPK"
+    channel_header: str = "officialAPK"
+
+    _fav_letter: str = "^"
+    # endregion
+    ###########################################################
+    # region client helper methods
+    def get_headers(self, payload=None, needs_auth: bool = False) -> dict:
+        pass
+
+    async def invoke_get(
+        self,
+        url: str,
+        headers: dict | None,
+        params: dict | None,
+        model: Any,
+        parse_float=Decimal,
+    ) -> Any:
+        """
+        Invokes the specific request to the specific url with the specific params and headers.
+        """
+        pass
+
+    async def invoke_post(
+        self,
+        url: str,
+        headers: dict | None = None,
+        params: dict | None = None,
+        content: str | bytes = "",
+        model: None = None,
+        parse_float=Decimal,
+    ):
+        """
+        Invokes the specific request to the specific url with the specific params and headers.
+        """
+        pass
+
+    async def aclose(self) -> None:
+        pass
+
+    def read_from_session_file(self, file_path: str) -> None:
+        """
+        Reads from session file; if it doesn't exist, creates it.
+        """
+        pass
+
+    def _save_session_file(self, file_path: str) -> None:
+        """
+        Saves current information to the session file.
+        """
+        pass
+
+    # endregion
+    ###########################################################

{trd_utils-0.0.4 → trd_utils-0.0.6}/trd_utils/tradingview/tradingview_client.py

@@ -1,6 +1,3 @@
-
-
-
 import json
 from typing import Optional
 
@@ -10,41 +7,45 @@ from .tradingview_types import CoinScanInfo
 class TradingViewClient:
     """TradingViewClient class to interact with TradingView API."""
 
-
     def __init__(self) -> None:
         pass
 
-    async def get_coin_scan(self,
-                            coin_filter: Optional[str] = None,
-                            limit: int = 200,
-                            offset: int = 0) -> list['CoinScanInfo']:
+    async def get_coin_scan(
+        self,
+        coin_filter: Optional[str] = None,
+        limit: int = 200,
+        offset: int = 0,
+    ) -> list["CoinScanInfo"]:
         import httpx
+
         cookies = {
-            'cookiesSettings': '{"analytics":true,"advertising":true}',
-            'cookiePrivacyPreferenceBannerProduction': 'accepted',
+            "cookiesSettings": '{"analytics":true,"advertising":true}',
+            "cookiePrivacyPreferenceBannerProduction": "accepted",
         }
 
         headers = {
-            'accept': 'application/json',
-            'accept-language': 'en-US,en;q=0.9',
-            'cache-control': 'no-cache',
-            'content-type': 'text/plain;charset=UTF-8',
-            'origin': 'https://www.tradingview.com',
-            'pragma': 'no-cache',
-            'priority': 'u=1, i',
-            'referer': 'https://www.tradingview.com/',
-            'sec-ch-ua': '"Not)A;Brand";v="99", "Google Chrome";v="127", "Chromium";v="127"',
-            'sec-ch-ua-mobile': '?0',
-            'sec-ch-ua-platform': '"Windows"',
-            'sec-fetch-dest': 'empty',
-            'sec-fetch-mode': 'cors',
-            'sec-fetch-site': 'same-site',
-            'user-agent': ('Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 ' +
-                '(KHTML, like Gecko) Chrome/127.0.0.0 Safari/537.36'),
+            "accept": "application/json",
+            "accept-language": "en-US,en;q=0.9",
+            "cache-control": "no-cache",
+            "content-type": "text/plain;charset=UTF-8",
+            "origin": "https://www.tradingview.com",
+            "pragma": "no-cache",
+            "priority": "u=1, i",
+            "referer": "https://www.tradingview.com/",
+            "sec-ch-ua": '"Not)A;Brand";v="99", "Google Chrome";v="127", "Chromium";v="127"',
+            "sec-ch-ua-mobile": "?0",
+            "sec-ch-ua-platform": '"Windows"',
+            "sec-fetch-dest": "empty",
+            "sec-fetch-mode": "cors",
+            "sec-fetch-site": "same-site",
+            "user-agent": (
+                "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 "
+                + "(KHTML, like Gecko) Chrome/127.0.0.0 Safari/537.36"
+            ),
         }
 
         params = {
-            'label-product': 'screener-coin',
+            "label-product": "screener-coin",
         }
 
         data = {
@@ -83,19 +84,14 @@ class TradingViewClient:
                 "Volatility.D",
             ],
             "ignore_unknown_fields": False,
-            "options": {
-                "lang": "en"
-            },
-            "range":[
+            "options": {"lang": "en"},
+            "range": [
                 offset,
                 offset + limit,
             ],
-            "sort": {
-                "sortBy": "crypto_total_rank",
-                "sortOrder": "asc"
-            },
+            "sort": {"sortBy": "crypto_total_rank", "sortOrder": "asc"},
             "symbols": {},
-            "markets": ["coin"]
+            "markets": ["coin"],
         }
 
         if coin_filter:
@@ -103,13 +99,13 @@ class TradingViewClient:
                 {
                     "left": "base_currency,base_currency_desc",
                     "operation": "match",
-                    "right": f"{coin_filter}"
+                    "right": f"{coin_filter}",
                 }
             ]
         data = json.dumps(data)
         async with httpx.AsyncClient() as client:
             response = await client.post(
-                'https://scanner.tradingview.com/coin/scan',
+                "https://scanner.tradingview.com/coin/scan",
                 params=params,
                 cookies=cookies,
                 headers=headers,
@@ -127,4 +123,4 @@ class TradingViewClient:
                 continue
             all_infos.append(CoinScanInfo._parse(current_data.get("d", [])))
 
-        return all_infos
+        return all_infos

{trd_utils-0.0.4 → trd_utils-0.0.6}/trd_utils/types_helper/base_model.py

@@ -1,10 +1,9 @@
 import json
 from typing import (
-    Optional,
     Union,
     get_type_hints,
     Any,
-    get_args as get_type_args
+    get_args as get_type_args,
 )
 
 from trd_utils.html_utils.html_formats import camel_to_snake
@@ -20,6 +19,7 @@ ULTRA_LIST_ENABLED: bool = False
 # attribute names are converted to snake_case.
 SET_CAMEL_ATTR_NAMES = False
 
+
 def get_my_field_types(cls):
     type_hints = {}
     for current_cls in cls.__class__.__mro__:
@@ -28,24 +28,31 @@ def get_my_field_types(cls):
         type_hints.update(get_type_hints(current_cls))
     return type_hints
 
+
 def get_real_attr(cls, attr_name):
     if cls is None:
         return None
-    
+
     if isinstance(cls, dict):
         return cls.get(attr_name, None)
 
     if hasattr(cls, attr_name):
         return getattr(cls, attr_name)
-    
+
     return None
 
+
+def is_any_type(target_type: type) -> bool:
+    return target_type == Any or target_type is type(None)
+
+
 class UltraList(list):
     def __getattr__(self, attr):
         if len(self) == 0:
             return None
         return UltraList([get_real_attr(item, attr) for item in self])
 
+
 def convert_to_ultra_list(value: Any) -> UltraList:
     if not value:
         return UltraList()
@@ -62,7 +69,7 @@ def convert_to_ultra_list(value: Any) -> UltraList:
             return tuple(convert_to_ultra_list(v) for v in value)
         elif isinstance(value, set):
             return {convert_to_ultra_list(v) for v in value}
-        
+
         for attr, attr_value in get_my_field_types(value).items():
             if isinstance(attr_value, list):
                 setattr(value, attr, convert_to_ultra_list(getattr(value, attr)))
@@ -71,6 +78,7 @@ def convert_to_ultra_list(value: Any) -> UltraList:
     except Exception:
         return value
 
+
 class BaseModel:
     def __init__(self, **kwargs):
         annotations = get_my_field_types(self)
@@ -84,17 +92,19 @@ class BaseModel:
                     # just ignore and continue
                     annotations[key] = Any
                     annotations[corrected_key] = Any
-            
+
             expected_type = annotations[corrected_key]
             if hasattr(self, "_get_" + corrected_key + "_type"):
                 try:
-                    overridden_type = getattr(self, "_get_" + corrected_key + "_type")(kwargs)
+                    overridden_type = getattr(self, "_get_" + corrected_key + "_type")(
+                        kwargs
+                    )
                     if overridden_type:
                         expected_type = overridden_type
                 except Exception:
                     pass
-            
-            is_optional_type = getattr(expected_type, '_name', None) == 'Optional'
+
+            is_optional_type = getattr(expected_type, "_name", None) == "Optional"
             # maybe in the future we can have some other usages for is_optional_type
             # variable or something like that.
             if is_optional_type:
@@ -103,11 +113,11 @@ class BaseModel:
                 except Exception:
                     # something went wrong, just ignore and continue
                     expected_type = Any
-            
+
             # Handle nested models
             if isinstance(value, dict) and issubclass(expected_type, BaseModel):
                 value = expected_type(**value)
-            
+
             elif isinstance(value, list):
                 type_args = get_type_args(expected_type)
                 if not type_args:
@@ -119,27 +129,29 @@ class BaseModel:
                     nested_type = type_args[0]
                     if issubclass(nested_type, BaseModel):
                         value = [nested_type(**item) for item in value]
-                
+
                 if ULTRA_LIST_ENABLED and isinstance(value, list):
                     value = convert_to_ultra_list(value)
-            
+
             # Type checking
-            elif expected_type != Any and not isinstance(value, expected_type):
+            elif not (is_any_type(expected_type) or isinstance(value, expected_type)):
                 try:
                     value = expected_type(value)
                 except Exception:
-                    raise TypeError(f"Field {corrected_key} must be of type {expected_type}," +
-                                    f" but it's {type(value)}")
-            
+                    raise TypeError(
+                        f"Field {corrected_key} must be of type {expected_type},"
+                        + f" but it's {type(value)}"
+                    )
+
             setattr(self, corrected_key, value)
             if SET_CAMEL_ATTR_NAMES and key != corrected_key:
                 setattr(self, key, value)
-        
+
         # Check if all required fields are present
         # for field in self.__annotations__:
         #     if not hasattr(self, field):
         #         raise ValueError(f"Missing required field: {field}")
-    
+
     @classmethod
     def deserialize(cls, json_data: Union[str, dict]):
         if isinstance(json_data, str):
@@ -147,4 +159,3 @@ class BaseModel:
         else:
             data = json_data
         return cls(**data)
-

trd_utils-0.0.4/README.md

@@ -1,3 +0,0 @@
-# Trd Utils
-
-Basic common utils for Python.

trd_utils-0.0.4/trd_utils/__init__.py

@@ -1,3 +0,0 @@
-
-
-