compass_api_sdk 0.9.35__py3-none-any.whl → 2.0.21rc1__py3-none-any.whl

compass_api_sdk/sdk.py

@@ -10,21 +10,28 @@ from compass_api_sdk._hooks import SDKHooks
 from compass_api_sdk.types import OptionalNullable, UNSET
 import httpx
 import importlib
+import sys
 from typing import Any, Callable, Dict, Optional, TYPE_CHECKING, Union, cast
 import weakref
 
 if TYPE_CHECKING:
     from compass_api_sdk.aave_v3 import AaveV3
     from compass_api_sdk.aerodrome_slipstream import AerodromeSlipstream
+    from compass_api_sdk.bridge import Bridge
+    from compass_api_sdk.earn import Earn
     from compass_api_sdk.erc_4626_vaults import ERC4626Vaults
+    from compass_api_sdk.ethena import Ethena
+    from compass_api_sdk.gas_sponsorship import GasSponsorship
     from compass_api_sdk.morpho import Morpho
     from compass_api_sdk.pendle import Pendle
     from compass_api_sdk.sky import Sky
     from compass_api_sdk.smart_account import SmartAccount
-    from compass_api_sdk.token_sdk import TokenSDK
+    from compass_api_sdk.swap import Swap
+    from compass_api_sdk.token import Token
     from compass_api_sdk.transaction_bundler import TransactionBundler
     from compass_api_sdk.uniswap_v3 import UniswapV3
     from compass_api_sdk.universal import Universal
+    from compass_api_sdk.wildcat import Wildcat
 
 
 class CompassAPI(BaseSDK):
@@ -34,13 +41,19 @@ class CompassAPI(BaseSDK):
     aerodrome_slipstream: "AerodromeSlipstream"
     morpho: "Morpho"
     sky: "Sky"
-    token: "TokenSDK"
+    token: "Token"
     uniswap_v3: "UniswapV3"
     universal: "Universal"
     pendle: "Pendle"
+    swap: "Swap"
     transaction_bundler: "TransactionBundler"
     smart_account: "SmartAccount"
     erc_4626_vaults: "ERC4626Vaults"
+    ethena: "Ethena"
+    wildcat: "Wildcat"
+    gas_sponsorship: "GasSponsorship"
+    earn: "Earn"
+    bridge: "Bridge"
     _sub_sdk_map = {
         "aave_v3": ("compass_api_sdk.aave_v3", "AaveV3"),
         "aerodrome_slipstream": (
@@ -49,16 +62,22 @@ class CompassAPI(BaseSDK):
         ),
         "morpho": ("compass_api_sdk.morpho", "Morpho"),
         "sky": ("compass_api_sdk.sky", "Sky"),
-        "token": ("compass_api_sdk.token_sdk", "TokenSDK"),
+        "token": ("compass_api_sdk.token", "Token"),
         "uniswap_v3": ("compass_api_sdk.uniswap_v3", "UniswapV3"),
         "universal": ("compass_api_sdk.universal", "Universal"),
         "pendle": ("compass_api_sdk.pendle", "Pendle"),
+        "swap": ("compass_api_sdk.swap", "Swap"),
         "transaction_bundler": (
             "compass_api_sdk.transaction_bundler",
             "TransactionBundler",
         ),
         "smart_account": ("compass_api_sdk.smart_account", "SmartAccount"),
         "erc_4626_vaults": ("compass_api_sdk.erc_4626_vaults", "ERC4626Vaults"),
+        "ethena": ("compass_api_sdk.ethena", "Ethena"),
+        "wildcat": ("compass_api_sdk.wildcat", "Wildcat"),
+        "gas_sponsorship": ("compass_api_sdk.gas_sponsorship", "GasSponsorship"),
+        "earn": ("compass_api_sdk.earn", "Earn"),
+        "bridge": ("compass_api_sdk.bridge", "Bridge"),
     }
 
     def __init__(
@@ -86,7 +105,7 @@ class CompassAPI(BaseSDK):
         """
         client_supplied = True
         if client is None:
-            client = httpx.Client()
+            client = httpx.Client(follow_redirects=True)
             client_supplied = False
 
         assert issubclass(
@@ -95,7 +114,7 @@ class CompassAPI(BaseSDK):
 
         async_client_supplied = True
         if async_client is None:
-            async_client = httpx.AsyncClient()
+            async_client = httpx.AsyncClient(follow_redirects=True)
             async_client_supplied = False
 
         if debug_logger is None:
@@ -130,6 +149,7 @@ class CompassAPI(BaseSDK):
                 timeout_ms=timeout_ms,
                 debug_logger=debug_logger,
             ),
+            parent_ref=self,
         )
 
         hooks = SDKHooks()
@@ -137,12 +157,7 @@ class CompassAPI(BaseSDK):
         # pylint: disable=protected-access
         self.sdk_configuration.__dict__["_hooks"] = hooks
 
-        current_server_url, *_ = self.sdk_configuration.get_server_details()
-        server_url, self.sdk_configuration.client = hooks.sdk_init(
-            current_server_url, client
-        )
-        if current_server_url != server_url:
-            self.sdk_configuration.server_url = server_url
+        self.sdk_configuration = hooks.sdk_init(self.sdk_configuration)
 
         weakref.finalize(
             self,
@@ -154,13 +169,24 @@ class CompassAPI(BaseSDK):
             self.sdk_configuration.async_client_supplied,
         )
 
+    def dynamic_import(self, modname, retries=3):
+        for attempt in range(retries):
+            try:
+                return importlib.import_module(modname)
+            except KeyError:
+                # Clear any half-initialized module and retry
+                sys.modules.pop(modname, None)
+                if attempt == retries - 1:
+                    break
+        raise KeyError(f"Failed to import module '{modname}' after {retries} attempts")
+
     def __getattr__(self, name: str):
         if name in self._sub_sdk_map:
             module_path, class_name = self._sub_sdk_map[name]
             try:
-                module = importlib.import_module(module_path)
+                module = self.dynamic_import(module_path)
                 klass = getattr(module, class_name)
-                instance = klass(self.sdk_configuration)
+                instance = klass(self.sdk_configuration, parent_ref=self)
                 setattr(self, name, instance)
                 return instance
             except ImportError as e: