prediction-market-agent-tooling 0.69.15.dev1135__py3-none-any.whl → 0.69.15.dev1137__py3-none-any.whl

prediction_market_agent_tooling/abis/processor.abi.json

@@ -0,0 +1,16 @@
+[
+  {
+    "constant": true,
+    "inputs": [],
+    "name": "processor",
+    "outputs": [
+      {
+        "name": "impl",
+        "type": "address"
+      }
+    ],
+    "payable": false,
+    "stateMutability": "view",
+    "type": "function"
+  }
+]

prediction_market_agent_tooling/tools/contract.py

@@ -12,6 +12,7 @@ from web3 import Web3
 from web3.constants import CHECKSUM_ADDRESSS_ZERO, HASH_ZERO
 from web3.contract.contract import Contract as Web3Contract
 from web3.exceptions import ContractCustomError, ContractLogicError
+from web3.types import BlockIdentifier
 
 from prediction_market_agent_tooling.chains import POLYGON_CHAIN_ID
 from prediction_market_agent_tooling.config import APIKeys, RPCConfig
@@ -97,6 +98,7 @@ class ContractBaseClass(BaseModel):
         function_name: str,
         function_params: t.Optional[list[t.Any] | dict[str, t.Any]] = None,
         web3: Web3 | None = None,
+        block_identifier: BlockIdentifier | None = None,
     ) -> t.Any:
         """
         Used for reading from the contract.
@@ -108,6 +110,7 @@ class ContractBaseClass(BaseModel):
             contract_abi=self.abi,
             function_name=function_name,
             function_params=function_params,
+            block_identifier=block_identifier,
         )
 
     def send(
@@ -196,6 +199,22 @@ class ContractProxyBaseClass(ContractBaseClass):
         return Web3.to_checksum_address(address)
 
 
+class ContractProcessorBaseClass(ContractBaseClass):
+    """
+    Contract base class for processor contracts.
+    """
+
+    abi: ABI = abi_field_validator(
+        os.path.join(
+            os.path.dirname(os.path.realpath(__file__)), "../abis/processor.abi.json"
+        )
+    )
+
+    def processor(self, web3: Web3 | None = None) -> ChecksumAddress:
+        address = self.call("processor", web3=web3)
+        return Web3.to_checksum_address(address)
+
+
 class ContractERC20BaseClass(ContractBaseClass):
     """
     Contract base class extended by ERC-20 standard methods.
@@ -274,8 +293,17 @@ class ContractERC20BaseClass(ContractBaseClass):
             web3=web3,
         )
 
-    def balanceOf(self, for_address: ChecksumAddress, web3: Web3 | None = None) -> Wei:
-        balance = Wei(self.call("balanceOf", [for_address], web3=web3))
+    def balanceOf(
+        self,
+        for_address: ChecksumAddress,
+        web3: Web3 | None = None,
+        block_identifier: BlockIdentifier | None = None,
+    ) -> Wei:
+        balance = Wei(
+            self.call(
+                "balanceOf", [for_address], web3=web3, block_identifier=block_identifier
+            )
+        )
         return balance
 
     def balance_of_in_tokens(
@@ -948,6 +976,7 @@ def uni_implementation_address(
     # TODO: Fix the above, and afterwards assert that only 1 imp address is returned from this function. Or prove that this could indeed happen (although we are very pretty sure it shouldn't).
     addresses = [
         implementation_proxy_address(contract_address, web3),
+        processor_proxy_address(contract_address, web3),
         minimal_proxy_address(contract_address, web3),
         seer_minimal_proxy_address(contract_address, web3),
         eip_1967_proxy_address(contract_address, web3),
@@ -979,6 +1008,29 @@ def implementation_proxy_address(
         return None
 
 
+def processor_proxy_address(
+    contract_address: ChecksumAddress, web3: Web3
+) -> ChecksumAddress | None:
+    if not contract_implements_function(
+        contract_address, "processor", web3, look_for_proxy_contract=False
+    ):
+        return None
+    try:
+        return ContractProcessorBaseClass(address=contract_address).processor(web3)
+    except (ContractCustomError, ContractLogicError, tenacity.RetryError) as e:
+        if isinstance(e, tenacity.RetryError) and not isinstance(
+            e.last_attempt.exception(), (ContractCustomError, ContractLogicError)
+        ):
+            raise
+
+        # For example https://gnosisscan.io/address/0x3221a28ed2b2e955da64d1d299956f277562c95c#code,
+        # it has the processor method, but it's only for admins.
+        logger.warning(
+            f"Failed to get processor for {contract_address=} even though it has the method: {e}"
+        )
+        return None
+
+
 def minimal_proxy_address(
     contract_address: ChecksumAddress, web3: Web3
 ) -> ChecksumAddress | None:

prediction_market_agent_tooling/tools/web3_utils.py

@@ -16,7 +16,14 @@ from safe_eth.safe.safe import SafeV141
 from web3 import Web3
 from web3.constants import HASH_ZERO
 from web3.contract.contract import ContractFunction as Web3ContractFunction
-from web3.types import AccessList, AccessListEntry, Nonce, TxParams, TxReceipt
+from web3.types import (
+    AccessList,
+    AccessListEntry,
+    BlockIdentifier,
+    Nonce,
+    TxParams,
+    TxReceipt,
+)
 
 from prediction_market_agent_tooling.gtypes import (
     ABI,
@@ -106,11 +113,12 @@ def call_function_on_contract(
     contract_abi: ABI,
     function_name: str,
     function_params: Optional[list[Any] | dict[str, Any]] = None,
+    block_identifier: Optional[BlockIdentifier] = None,
 ) -> Any:
     contract = web3.eth.contract(address=contract_address, abi=contract_abi)
     output = contract.functions[function_name](
         *parse_function_params(function_params)
-    ).call()
+    ).call(block_identifier=block_identifier)
     return output
 
 

{prediction_market_agent_tooling-0.69.15.dev1135.dist-info → prediction_market_agent_tooling-0.69.15.dev1137.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: prediction-market-agent-tooling
-Version: 0.69.15.dev1135
+Version: 0.69.15.dev1137
 Summary: Tools to benchmark, deploy and monitor prediction market agents.
 License-File: LICENSE
 Author: Gnosis

{prediction_market_agent_tooling-0.69.15.dev1135.dist-info → prediction_market_agent_tooling-0.69.15.dev1137.dist-info}/RECORD

@@ -16,6 +16,7 @@ prediction_market_agent_tooling/abis/omen_realitio.abi.json,sha256=7HmFkBF_rq83U
 prediction_market_agent_tooling/abis/omen_thumbnailmapping.abi.json,sha256=u1-3B8FB3Ys9KVJCH-lw9ArkicdxbNMf34dV-VEGMMU,930
 prediction_market_agent_tooling/abis/ownable.abi.json,sha256=DeTy_7VmsMhFl7jwI8MIlmjy2jORauYxrGm7wC_Alxw,1528
 prediction_market_agent_tooling/abis/ownable_erc721.abi.json,sha256=9sxm588MAQmqCV_S0D3eYC7l9grbeALsd0Da_AHxdEI,8506
+prediction_market_agent_tooling/abis/processor.abi.json,sha256=VGMyBblTPO3I6AoKQFMgTylgJLl6-6FquUbejICDEAw,242
 prediction_market_agent_tooling/abis/proxy.abi.json,sha256=h24GXZ6Q0bSZlwh7zOv0EiDvbqUz_PHtWfKHTyPJ1w4,644
 prediction_market_agent_tooling/abis/seer_gnosis_router.abi.json,sha256=DyADzOXhy9MDS31ReVrG7ibpWbw1jVy19nExZ80xfRY,6839
 prediction_market_agent_tooling/abis/seer_market_factory.abi.json,sha256=g7RVxZVUWlTXIgTV2W6kO4twQM909Qv58zAr7Dk4XIc,13553
@@ -92,7 +93,7 @@ prediction_market_agent_tooling/tools/betting_strategies/utils.py,sha256=MpS3FOM
 prediction_market_agent_tooling/tools/caches/db_cache.py,sha256=V6o6UdesjkKzSJMhqkUtD76cJGPaNhuwA4OL2chIYSI,13801
 prediction_market_agent_tooling/tools/caches/inmemory_cache.py,sha256=ZW5iI5rmjqeAebu5T7ftRnlkxiL02IC-MxCfDB80x7w,1506
 prediction_market_agent_tooling/tools/caches/serializers.py,sha256=fwyIHUsL7u9NY7ZihTcUdqsvl2psatcbTfgWHTSez3U,2427
-prediction_market_agent_tooling/tools/contract.py,sha256=ZPQT3Kk9-q-ns7DO3MjQ_OLPUgeDjUTFwKP7L0vKKqc,36035
+prediction_market_agent_tooling/tools/contract.py,sha256=XbdegY4dVdiJkNbtXLKtR2GsqGLMYbvc5lTpIIOcWLw,37785
 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=DN_8cPrr4jWVpXdS4D0j1QB19nB8fxDoSheo2BFMc8M,14523
@@ -136,9 +137,9 @@ prediction_market_agent_tooling/tools/tokens/token_utils.py,sha256=fhs-FH9m9IbzG
 prediction_market_agent_tooling/tools/tokens/usd.py,sha256=DPO-4HBTy1-TZHKL_9CnHQ8p9W7ivPfcRlUkRO0nKJ8,3259
 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.15.dev1135.dist-info/METADATA,sha256=1H0d5JE8lKB6PgIB_tlN93PQHxM1fcPdnTtINOj2_Rg,8899
-prediction_market_agent_tooling-0.69.15.dev1135.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
-prediction_market_agent_tooling-0.69.15.dev1135.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.69.15.dev1135.dist-info/licenses/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.69.15.dev1135.dist-info/RECORD,,
+prediction_market_agent_tooling/tools/web3_utils.py,sha256=1G8uu33p4zNhGKSiyyueROtMau4YoMR3N1QDRFQ3TjE,14098
+prediction_market_agent_tooling-0.69.15.dev1137.dist-info/METADATA,sha256=0p53RdWnPkbg0lmhBEQ0YaxRvqLGQXWZVhoSJHkhQbs,8899
+prediction_market_agent_tooling-0.69.15.dev1137.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+prediction_market_agent_tooling-0.69.15.dev1137.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.69.15.dev1137.dist-info/licenses/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.69.15.dev1137.dist-info/RECORD,,