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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: trd_utils
-Version: 0.0.30
+Version: 0.0.32
 Summary: Common Basic Utils for Python3. By ALiwoto.
 Keywords: utils,trd_utils,basic-utils,common-utils
 Author: ALiwoto

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

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

trd_utils-0.0.32/trd_utils/__init__.py

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

{trd_utils-0.0.30 → trd_utils-0.0.32}/trd_utils/exchanges/blofin/blofin_client.py

@@ -59,6 +59,8 @@ class BlofinClient(ExchangeBase):
         self.account_name = account_name
         self._fav_letter = fav_letter
         self.sessions_dir = sessions_dir
+        
+        super().__init__()
 
         if read_session_file:
             self.read_from_session_file(f"{sessions_dir}/{self.account_name}.bf")

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

@@ -636,6 +636,17 @@ class CopyTraderResumeResult(BaseModel):
     swap_copy_trade_label_type: int = None
     is_pro: int = None
 
+    def get_account_identity_by_filter(self, filter_text: str):
+        if not self.trader_sharing_accounts:
+            return 0
+        
+        for current in self.trader_sharing_accounts:
+            if current.display_name.lower().find(filter_text) != -1 or \
+                current.copy_trade_account_enum.lower().find(filter_text) != -1:
+                if current.api_identity:
+                    return current.api_identity
+        return 0
+
 
 class CopyTraderResumeResponse(BxApiResponse):
     data: CopyTraderResumeResult = None

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

@@ -124,14 +124,12 @@ class BXUltraClient(ExchangeBase):
         self._fav_letter = fav_letter
         self.sessions_dir = sessions_dir
 
+        super().__init__()
         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
     ###########################################################
@@ -310,11 +308,12 @@ class BXUltraClient(ExchangeBase):
             async for msg in ws:
                 try:
                     decompressed_message = gzip.decompress(msg)
-                    if decompressed_message.lower() == "ping":
+                    str_msg = decompressed_message.decode("utf-8")
+                    if str_msg.lower() == "ping":
                         await ws.send("Pong")
                         continue
 
-                    data: dict = json.loads(decompressed_message, parse_float=Decimal)
+                    data: dict = json.loads(str_msg, parse_float=Decimal)
                     if not isinstance(data, dict):
                         logger.warning(f"invalid data instance: {type(data)}")
                         continue
@@ -702,6 +701,7 @@ class BXUltraClient(ExchangeBase):
     async def get_trader_api_identity(
         self,
         uid: int | str,
+        sub_account_filter: str = "futures",
     ) -> int | str:
         global user_api_identity_cache
         api_identity = user_api_identity_cache.get(uid, None)
@@ -710,6 +710,12 @@ class BXUltraClient(ExchangeBase):
                 uid=uid,
             )
             api_identity = resume.data.api_identity
+            if not api_identity:
+                # second try: try to use one of the sub-accounts' identity
+                api_identity = resume.data.get_account_identity_by_filter(sub_account_filter)
+
+            # maybe also try to fetch it in other ways later?
+            # ...
             user_api_identity_cache[uid] = api_identity
         return api_identity
 
@@ -838,7 +844,8 @@ class BXUltraClient(ExchangeBase):
         aes = AESCipher(key=f"bx_{self.account_name}_bx", fav_letter=self._fav_letter)
         target_path = Path(file_path)
         if not target_path.exists():
-            target_path.mkdir(parents=True)
+            target_path.parent.mkdir(parents=True, exist_ok=True)
+            target_path.touch()
         target_path.write_text(aes.encrypt(json.dumps(json_data)))
 
     # endregion
@@ -848,10 +855,17 @@ class BXUltraClient(ExchangeBase):
     async def get_unified_trader_positions(
         self,
         uid: int | str,
+        api_identity: int | str | None = None,
+        sub_account_filter: str = "futures",
     ) -> UnifiedTraderPositions:
-        api_identity = await self.get_trader_api_identity(
-            uid=uid,
-        )
+        if not api_identity:
+            api_identity = await self.get_trader_api_identity(
+                uid=uid,
+                sub_account_filter=sub_account_filter,
+            )
+        
+        if not api_identity:
+            raise ValueError(f"Failed to fetch api_identity for user {uid}")
 
         result = await self.get_copy_trader_positions(
             uid=uid,

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

@@ -66,7 +66,15 @@ class ExchangeBase(ABC):
     ws_connections: list[WSConnection] = None
     # endregion
     ###########################################################
+    # region constructor method
 
+    def __init__(self):
+        self._internal_lock = asyncio.Lock()
+        self.extra_tasks = []
+        self.ws_connections = []
+
+    # endregion
+    ###########################################################
     # region abstract trading methods
 
     async def get_unified_trader_positions(
@@ -96,7 +104,6 @@ class ExchangeBase(ABC):
         )
 
     # endregion
-
     ###########################################################
     # region client helper methods
     def get_headers(self, payload=None, needs_auth: bool = False) -> dict:

{trd_utils-0.0.30 → trd_utils-0.0.32}/trd_utils/exchanges/hyperliquid/hyperliquid_client.py

@@ -47,6 +47,8 @@ class HyperLiquidClient(ExchangeBase):
         self._fav_letter = fav_letter
         self.sessions_dir = sessions_dir
 
+        super().__init__()
+
         if read_session_file:
             self.read_from_session_file(f"{sessions_dir}/{self.account_name}.hl")
 

{trd_utils-0.0.30 → trd_utils-0.0.32}/trd_utils/exchanges/okx/okx_client.py

@@ -55,6 +55,8 @@ class OkxClient(ExchangeBase):
         self._fav_letter = fav_letter
         self.sessions_dir = sessions_dir
 
+        super().__init__()
+
         if read_session_file:
             self.read_from_session_file(f"{sessions_dir}/{self.account_name}.okx")
 

trd_utils-0.0.30/trd_utils/__init__.py

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