trd-utils 0.0.19__tar.gz → 0.0.21__tar.gz

{trd_utils-0.0.19 → trd_utils-0.0.21}/PKG-INFO

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

{trd_utils-0.0.19 → trd_utils-0.0.21}/pyproject.toml

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

trd_utils-0.0.21/trd_utils/__init__.py

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

trd_utils-0.0.21/trd_utils/common_utils/wallet_utils.py

@@ -0,0 +1,26 @@
+"""
+General usage wallet-related utils code.
+"""
+
+
+def shorten_wallet_address(
+    address: str,
+    start_chars: int = 7,
+    end_chars: int = 6,
+):
+    """
+    Shortens an Ethereum address by keeping a specific number of characters
+    from the beginning and end, separated by '...'.
+
+    Args:
+        address (str): The Ethereum address to shorten
+        start_chars (int): Number of characters to keep from the beginning (default: 7)
+        end_chars (int): Number of characters to keep from the end (default: 6)
+
+    Returns:
+        str: The shortened address
+    """
+    if not address or len(address) <= start_chars + end_chars:
+        return address
+
+    return f"{address[:start_chars]}...{address[-end_chars:]}"

{trd_utils-0.0.19 → trd_utils-0.0.21}/trd_utils/exchanges/base_types.py

@@ -1,10 +1,9 @@
-
-
 from datetime import datetime
 from decimal import Decimal
 
 from trd_utils.types_helper.base_model import BaseModel
 
+
 class UnifiedPositionInfo(BaseModel):
     # The id of the position.
     position_id: str = None
@@ -40,20 +39,22 @@ class UnifiedPositionInfo(BaseModel):
 
     def __str__(self):
         parts = []
-        
+
         # Add position pair and ID
-        parts.append(f"Position: {self.position_pair or 'Unknown'} (ID: {self.position_id or 'N/A'})")
-        
+        parts.append(
+            f"Position: {self.position_pair or 'Unknown'} (ID: {self.position_id or 'N/A'})"
+        )
+
         # Add side and leverage
         side_str = f"Side: {self.position_side or 'Unknown'}"
         if self.position_leverage is not None:
             side_str += f", {self.position_leverage}x"
         parts.append(side_str)
-        
+
         # Add margin mode if available
         if self.margin_mode:
             parts.append(f"Margin: {self.margin_mode}")
-        
+
         # Add open price if available
         price_str = "Open price: "
         if self.open_price is not None:
@@ -63,23 +64,25 @@ class UnifiedPositionInfo(BaseModel):
         else:
             price_str += "N/A"
         parts.append(price_str)
-        
+
         # Add open time if available
         if self.open_time:
             parts.append(f"Opened: {self.open_time.strftime('%Y-%m-%d %H:%M:%S')}")
-        
+
         # Add PNL if available
         if self.position_pnl is not None:
             parts.append(f"PNL: {self.position_pnl}")
-        
+
         return " | ".join(parts)
 
     def __repr__(self):
         return self.__str__()
 
+
 class UnifiedTraderPositions(BaseModel):
     positions: list[UnifiedPositionInfo] = None
 
+
 class UnifiedTraderInfo(BaseModel):
     # Trader's id. Either int or str. In DEXes (such as HyperLiquid),
     # this might be wallet address of the trader.
@@ -94,13 +97,15 @@ class UnifiedTraderInfo(BaseModel):
     # Trader's win-rate. Not all exchanges might support this field.
     win_rate: Decimal = None
 
+    def get_win_rate_str(self) -> str:
+        return str(round(self.win_rate, 2)) if self.win_rate is not None else "N/A"
+
     def __str__(self):
         return (
-            f"Trader: {self.trader_name} (ID: {self.trader_id})"
-            f"{' | Win Rate: ' + str(round(self.win_rate, 2))}"
-            f"{' | Profile: ' + self.trader_url}"
+            f"{self.trader_name} ({self.trader_id})"
+            f" | Win Rate: {self.get_win_rate_str()}"
+            f" | Profile: {self.trader_url}"
         )
-    
+
     def __repr__(self):
         return self.__str__()
-    

{trd_utils-0.0.19 → trd_utils-0.0.21}/trd_utils/exchanges/hyperliquid/hyperliquid_client.py

@@ -8,11 +8,15 @@ import httpx
 from pathlib import Path
 
 from trd_utils.cipher import AESCipher
+from trd_utils.common_utils.wallet_utils import shorten_wallet_address
+from trd_utils.exchanges.base_types import UnifiedPositionInfo, UnifiedTraderInfo, UnifiedTraderPositions
 from trd_utils.exchanges.exchange_base import ExchangeBase
 from trd_utils.exchanges.hyperliquid.hyperliquid_types import HyperLiquidApiResponse, TraderPositionsInfoResponse
 
 logger = logging.getLogger(__name__)
 
+BASE_PROFILE_URL = "https://hypurrscan.io/address/"
+
 
 class HyperLiquidClient(ExchangeBase):
     ###########################################################
@@ -63,7 +67,7 @@ class HyperLiquidClient(ExchangeBase):
             f"{self.hyperliquid_api_base_host}/info",
             headers=headers,
             content=payload,
-            model=TraderPositionsInfoResponse,
+            model_type=TraderPositionsInfoResponse,
         )
 
     #endregion
@@ -78,7 +82,7 @@ class HyperLiquidClient(ExchangeBase):
     #         f"{self.hyperliquid_api_base_url}/another-thing/info",
     #         headers=headers,
     #         content=payload,
-    #         model=CopyTraderInfoResponse,
+    #         model_type=CopyTraderInfoResponse,
     #     )
 
     # endregion
@@ -138,3 +142,48 @@ class HyperLiquidClient(ExchangeBase):
 
     # endregion
     ###########################################################
+    # region unified methods
+    async def get_unified_trader_positions(
+        self,
+        uid: int | str,
+    ) -> UnifiedTraderPositions:
+        result = await self.get_trader_positions_info(
+            uid=uid,
+        )
+        unified_result = UnifiedTraderPositions()
+        unified_result.positions = []
+        for position_container in result.asset_positions:
+            position = position_container.position
+            unified_pos = UnifiedPositionInfo()
+            unified_pos.position_id = position.get_position_id()
+            unified_pos.position_pnl = round(position.unrealized_pnl, 3)
+            unified_pos.position_side = position.get_side()
+            unified_pos.margin_mode = position.leverage.type
+            unified_pos.position_leverage = Decimal(position.leverage.value)
+            unified_pos.position_pair = f"{position.coin}/USDT"
+            unified_pos.open_time = None # hyperliquid doesn't provide this...
+            unified_pos.open_price = position.entry_px
+            unified_pos.open_price_unit = "USDT"
+            unified_result.positions.append(unified_pos)
+
+        return unified_result
+
+    async def get_unified_trader_info(
+        self,
+        uid: int | str,
+    ) -> UnifiedTraderInfo:
+        if not isinstance(uid, str):
+            uid = str(uid)
+        # sadly hyperliquid doesn't really have an endpoint to fetch information
+        # so we have to somehow *fake* these...
+        # maybe in future try to find a better way?
+        unified_info = UnifiedTraderInfo()
+        unified_info.trader_id = uid
+        unified_info.trader_name = shorten_wallet_address(uid)
+        unified_info.trader_url = f"{BASE_PROFILE_URL}{uid}"
+        unified_info.win_rate = None
+
+        return unified_info
+
+    # endregion
+    ###########################################################

{trd_utils-0.0.19 → trd_utils-0.0.21}/trd_utils/exchanges/hyperliquid/hyperliquid_types.py

@@ -58,6 +58,17 @@ class PositionInfo(BaseModel):
         In any case, we will have to somehow fake it in order to be able to compare
         it with other positions...
         """
+        entry = self.entry_px
+        if entry > 100:
+            entry = round(entry, 1)
+        elif entry > 10:
+            entry = round(entry, 2)
+        elif entry > 1:
+            entry = round(entry, 3)
+        elif entry > 0.1:
+            entry = round(entry, 4)
+        elif entry > 0.01:
+            entry = round(entry, 5)
         return (
             f"{self.coin}-{self.leverage.value}{self.entry_px}"
         ).encode("utf-8").hex()

trd_utils-0.0.19/trd_utils/__init__.py

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