prediction-market-agent-tooling 0.69.6.dev1095__py3-none-any.whl → 0.69.7.dev1098__py3-none-any.whl

prediction_market_agent_tooling/tools/contract.py

@@ -909,84 +909,77 @@ def contract_implements_function(
 
     # If not found directly and we should check proxies
     if not implements and look_for_proxy_contract:
-        # Case 1: Check if it's a standard proxy (has implementation() function)
-        if contract_implements_function(
-            contract_address, "implementation", web3, look_for_proxy_contract=False
-        ):
-            # Get the implementation address and check the function there
-            implementation_address = ContractProxyOnGnosisChain(
-                address=contract_address
-            ).implementation()
+        imp_address = uni_implementation_address(contract_address, web3)
+        if imp_address is not None:
             implements = contract_implements_function(
-                implementation_address,
+                imp_address,
                 function_name=function_name,
                 web3=web3,
                 function_arg_types=function_arg_types,
                 look_for_proxy_contract=False,
             )
-        else:
-            # Case 2: Check if it's a minimal proxy contract
-            implements = minimal_proxy_implements_function(
-                contract_address=contract_address,
-                function_name=function_name,
-                web3=web3,
-                function_arg_types=function_arg_types,
-            ) or seer_minimal_proxy_implements_function(
-                contract_address=contract_address,
-                function_name=function_name,
-                web3=web3,
-                function_arg_types=function_arg_types,
-            )
 
     return implements
 
 
-def minimal_proxy_implements_function(
-    contract_address: ChecksumAddress,
-    function_name: str,
-    web3: Web3,
-    function_arg_types: list[str] | None = None,
-) -> bool:
+def uni_implementation_address(
+    contract_address: ChecksumAddress, web3: Web3
+) -> ChecksumAddress | None:
+    """
+    There are multiple ways how proxies can be implemented.
+    This function enumerates them and returns the one that succeeds, or None.
+    """
+    address = implementation_proxy_address(contract_address, web3)
+    if address is not None:
+        return address
+
+    address = minimal_proxy_address(contract_address, web3)
+    if address is not None:
+        return address
+
+    address = seer_minimal_proxy_address(contract_address, web3)
+    if address is not None:
+        return address
+
+    return None
+
+
+def implementation_proxy_address(
+    contract_address: ChecksumAddress, web3: Web3
+) -> ChecksumAddress | None:
+    if not contract_implements_function(
+        contract_address, "implementation", web3, look_for_proxy_contract=False
+    ):
+        return None
+    return ContractProxyOnGnosisChain(address=contract_address).implementation()
+
+
+def minimal_proxy_address(
+    contract_address: ChecksumAddress, web3: Web3
+) -> ChecksumAddress | None:
     try:
         # Read storage slot 0 which should contain the implementation address in minimal proxies
         raw_slot_0 = web3.eth.get_storage_at(contract_address, 0)
         singleton_address = eth_abi.decode(["address"], raw_slot_0)[0]
-        # Recurse into singleton
-        return contract_implements_function(
-            Web3.to_checksum_address(singleton_address),
-            function_name=function_name,
-            web3=web3,
-            function_arg_types=function_arg_types,
-            look_for_proxy_contract=False,
-        )
+        return Web3.to_checksum_address(singleton_address)
     except DecodingError:
         logger.info(f"Error decoding contract address for singleton")
-        return False
+        return None
 
 
-def seer_minimal_proxy_implements_function(
-    contract_address: ChecksumAddress,
-    function_name: str,
-    web3: Web3,
-    function_arg_types: list[str] | None = None,
-) -> bool:
+def seer_minimal_proxy_address(
+    contract_address: ChecksumAddress, web3: Web3
+) -> ChecksumAddress | None:
     try:
         # Read address between specific indices to find logic contract
         bytecode = web3.eth.get_code(contract_address)
         logic_contract_address = bytecode[11:31]
         if not Web3.is_address(logic_contract_address):
-            return False
-
-        return contract_implements_function(
-            Web3.to_checksum_address(logic_contract_address),
-            function_name=function_name,
-            web3=web3,
-            function_arg_types=function_arg_types,
-            look_for_proxy_contract=False,
-        )
+            return None
+        return Web3.to_checksum_address(logic_contract_address)
     except DecodingError:
         logger.info("Error decoding contract address on seer minimal proxy")
-        return False
+        return None
 
 
 def init_collateral_token_contract(

{prediction_market_agent_tooling-0.69.6.dev1095.dist-info → prediction_market_agent_tooling-0.69.7.dev1098.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.69.6.dev1095
+Version: 0.69.7.dev1098
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 Author: Gnosis
 Requires-Python: >=3.10,<3.13

{prediction_market_agent_tooling-0.69.6.dev1095.dist-info → prediction_market_agent_tooling-0.69.7.dev1098.dist-info}/RECORD

@@ -91,7 +91,7 @@ prediction_market_agent_tooling/tools/betting_strategies/utils.py,sha256=MpS3FOM
 prediction_market_agent_tooling/tools/caches/db_cache.py,sha256=rZIGhgijquwwPtp_qncSAPR1SDF2XxIVZL1ir0fgzWw,12127
 prediction_market_agent_tooling/tools/caches/inmemory_cache.py,sha256=ZW5iI5rmjqeAebu5T7ftRnlkxiL02IC-MxCfDB80x7w,1506
 prediction_market_agent_tooling/tools/caches/serializers.py,sha256=vFDx4fsPxclXp2q0sv27j4al_M_Tj9aR2JJP-xNHQXA,2151
-prediction_market_agent_tooling/tools/contract.py,sha256=Agd3eNsr9PW6t2jtGUzcQlmudNKpjvQkSfGEcq28i8k,33813
+prediction_market_agent_tooling/tools/contract.py,sha256=zZ4EDlTdKuPhUWDzwHJ1YrY3fIqVBTO3N2R99tFxPVQ,33293
 prediction_market_agent_tooling/tools/contract_utils.py,sha256=9X9raICUZkPDShilt02aYzS_ILZ62u0vG5081uWLdqk,2152
 prediction_market_agent_tooling/tools/costs.py,sha256=EaAJ7v9laD4VEV3d8B44M4u3_oEO_H16jRVCdoZ93Uw,954
 prediction_market_agent_tooling/tools/cow/cow_order.py,sha256=kdvhWVTb31oHVspQFowDNEIoDtx8hwGTlKsWYGRh3oQ,14050
@@ -136,8 +136,8 @@ prediction_market_agent_tooling/tools/tokens/usd.py,sha256=DPO-4HBTy1-TZHKL_9CnH
 prediction_market_agent_tooling/tools/transaction_cache.py,sha256=K5YKNL2_tR10Iw2TD9fuP-CTGpBbZtNdgbd0B_R7pjg,1814
 prediction_market_agent_tooling/tools/utils.py,sha256=ruq6P5TFs8CBHxeBLj1Plpx7kuNFPpDgMsJGQgDiRNs,8785
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=CDbaidlLeQ4VHzSg150L7QNfHfGveljSePGuDVFEYqc,13963
-prediction_market_agent_tooling-0.69.6.dev1095.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.69.6.dev1095.dist-info/METADATA,sha256=HEwiOdV121R0GRKDNkEvg2oLNtk_krazGIiBrrOdgIQ,8876
-prediction_market_agent_tooling-0.69.6.dev1095.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-prediction_market_agent_tooling-0.69.6.dev1095.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.69.6.dev1095.dist-info/RECORD,,
+prediction_market_agent_tooling-0.69.7.dev1098.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.69.7.dev1098.dist-info/METADATA,sha256=FsfzI3oDm18A9Kk4d4hYcXo43Xt3J4A-oX9kiwLtXYE,8876
+prediction_market_agent_tooling-0.69.7.dev1098.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+prediction_market_agent_tooling-0.69.7.dev1098.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.69.7.dev1098.dist-info/RECORD,,