prediction-market-agent-tooling 0.57.11.dev258__py3-none-any.whl → 0.57.11.dev260__py3-none-any.whl

prediction_market_agent_tooling/abis/ownable.abi.json

@@ -0,0 +1,76 @@
+[
+    {
+      "type": "function",
+      "name": "owner",
+      "inputs": [],
+      "outputs": [
+        {
+          "name": "",
+          "type": "address",
+          "internalType": "address"
+        }
+      ],
+      "stateMutability": "view"
+    },
+    {
+      "type": "function",
+      "name": "renounceOwnership",
+      "inputs": [],
+      "outputs": [],
+      "stateMutability": "nonpayable"
+    },
+    {
+      "type": "function",
+      "name": "transferOwnership",
+      "inputs": [
+        {
+          "name": "newOwner",
+          "type": "address",
+          "internalType": "address"
+        }
+      ],
+      "outputs": [],
+      "stateMutability": "nonpayable"
+    },
+    {
+      "type": "event",
+      "name": "OwnershipTransferred",
+      "inputs": [
+        {
+          "name": "previousOwner",
+          "type": "address",
+          "indexed": true,
+          "internalType": "address"
+        },
+        {
+          "name": "newOwner",
+          "type": "address",
+          "indexed": true,
+          "internalType": "address"
+        }
+      ],
+      "anonymous": false
+    },
+    {
+      "type": "error",
+      "name": "OwnableInvalidOwner",
+      "inputs": [
+        {
+          "name": "owner",
+          "type": "address",
+          "internalType": "address"
+        }
+      ]
+    },
+    {
+      "type": "error",
+      "name": "OwnableUnauthorizedAccount",
+      "inputs": [
+        {
+          "name": "account",
+          "type": "address",
+          "internalType": "address"
+        }
+      ]
+    }
+  ]

prediction_market_agent_tooling/tools/contract.py

@@ -6,6 +6,7 @@ from contextlib import contextmanager
 
 from pydantic import BaseModel, field_validator
 from web3 import Web3
+from web3.constants import CHECKSUM_ADDRESSS_ZERO
 from web3.contract.contract import Contract as Web3Contract
 
 from prediction_market_agent_tooling.config import APIKeys, RPCConfig
@@ -78,6 +79,13 @@ class ContractBaseClass(BaseModel):
         {}
     )  # Can be used to hold values that aren't going to change after getting them for the first time, as for example `symbol` of an ERC-20 token.
 
+    @classmethod
+    def merge_contract_abis(cls, contracts: list["ContractBaseClass"]) -> ABI:
+        merged_abi: list[dict[t.Any, t.Any]] = sum(
+            (json.loads(contract.abi) for contract in contracts), []
+        )
+        return abi_field_validator(json.dumps(merged_abi))
+
     def get_web3_contract(self, web3: Web3 | None = None) -> Web3Contract:
         web3 = web3 or self.get_web3()
         return web3.eth.contract(address=self.address, abi=self.abi)
@@ -391,6 +399,13 @@ class ContractERC4626BaseClass(ContractERC20BaseClass):
 
 
 class OwnableContract(ContractBaseClass):
+    abi: ABI = abi_field_validator(
+        os.path.join(
+            os.path.dirname(os.path.realpath(__file__)),
+            "../abis/ownable.abi.json",
+        )
+    )
+
     def owner(self, web3: Web3 | None = None) -> ChecksumAddress:
         owner = Web3.to_checksum_address(self.call("owner", web3=web3))
         return owner
@@ -458,7 +473,13 @@ class ContractERC721BaseClass(ContractBaseClass):
 
 
 class ContractOwnableERC721BaseClass(ContractERC721BaseClass, OwnableContract):
-    pass
+    # We add dummy addresses below so that we can reference .abi later.
+    abi: ABI = ContractBaseClass.merge_contract_abis(
+        [
+            ContractERC721BaseClass(address=ChecksumAddress(CHECKSUM_ADDRESSS_ZERO)),
+            OwnableContract(address=ChecksumAddress(CHECKSUM_ADDRESSS_ZERO)),
+        ]
+    )
 
 
 class ContractOnGnosisChain(ContractBaseClass):
@@ -571,11 +592,13 @@ class SimpleTreasuryContract(ContractOnGnosisChain, OwnableContract):
     def set_required_nft_balance(
         self,
         api_keys: APIKeys,
+        new_required_balance: int,
         web3: Web3 | None = None,
     ) -> TxReceipt:
         return self.send(
             api_keys=api_keys,
             function_name="setRequiredNFTBalance",
+            function_params=[new_required_balance],
             web3=web3,
         )
 

{prediction_market_agent_tooling-0.57.11.dev258.dist-info → prediction_market_agent_tooling-0.57.11.dev260.dist-info}/METADATA

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

{prediction_market_agent_tooling-0.57.11.dev258.dist-info → prediction_market_agent_tooling-0.57.11.dev260.dist-info}/RECORD

@@ -13,6 +13,7 @@ prediction_market_agent_tooling/abis/omen_kleros.abi.json,sha256=QPMXrYA5UCo_wVU
 prediction_market_agent_tooling/abis/omen_oracle.abi.json,sha256=YPZ-FLvd4PA9pYd6d5mQI1TD11JnsE0LGxH4XcZZFCA,1775
 prediction_market_agent_tooling/abis/omen_realitio.abi.json,sha256=7HmFkBF_rq83UTaH2kRRsEfc_WZuf4n-qvkB4nhvweo,15953
 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/proxy.abi.json,sha256=h24GXZ6Q0bSZlwh7zOv0EiDvbqUz_PHtWfKHTyPJ1w4,644
 prediction_market_agent_tooling/abis/simpletreasury.abi.json,sha256=7-l7rntLkcFKkoN8hXywO0-h-mUZxOZlyQg1Jk7yTB0,2877
@@ -79,7 +80,7 @@ prediction_market_agent_tooling/tools/betting_strategies/utils.py,sha256=kpIb-ci
 prediction_market_agent_tooling/tools/caches/db_cache.py,sha256=aafau_n_AUbLIwkyIRiTPgKB0dmM0767mSqyPDLF2A4,10576
 prediction_market_agent_tooling/tools/caches/inmemory_cache.py,sha256=ZW5iI5rmjqeAebu5T7ftRnlkxiL02IC-MxCfDB80x7w,1506
 prediction_market_agent_tooling/tools/caches/serializers.py,sha256=upSXN5__rmRlzJ6tv1h7FodKzJu9eCkFrN_zeuwroJM,2151
-prediction_market_agent_tooling/tools/contract.py,sha256=Z3SHsC68iFj8GhU6YsFf7kqB13zKJPUvawEkvPj3B1c,28656
+prediction_market_agent_tooling/tools/contract.py,sha256=glnPQr9B0WmhLcdifD892zLPaHS5f0fZZ3dKLOUB6pI,29559
 prediction_market_agent_tooling/tools/costs.py,sha256=EaAJ7v9laD4VEV3d8B44M4u3_oEO_H16jRVCdoZ93Uw,954
 prediction_market_agent_tooling/tools/custom_exceptions.py,sha256=Fh8z1fbwONvP4-j7AmV_PuEcoqb6-QXa9PJ9m7guMcM,93
 prediction_market_agent_tooling/tools/data_models.py,sha256=jDQ7FU0QQhXlcgJh5VZZGwDTYP2OPAqKPHZFewCPAUY,732
@@ -108,8 +109,8 @@ prediction_market_agent_tooling/tools/tavily/tavily_search.py,sha256=Kw2mXNkMTYT
 prediction_market_agent_tooling/tools/transaction_cache.py,sha256=K5YKNL2_tR10Iw2TD9fuP-CTGpBbZtNdgbd0B_R7pjg,1814
 prediction_market_agent_tooling/tools/utils.py,sha256=WvuUCHgMCiMq8_wMm5PHNwvLhcdDk2zGKaAM8OUC-qY,6438
 prediction_market_agent_tooling/tools/web3_utils.py,sha256=wqUDCed3iNrn1Wao1iwGN6tzIrhpzrTRj319wlveJEo,12275
-prediction_market_agent_tooling-0.57.11.dev258.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
-prediction_market_agent_tooling-0.57.11.dev258.dist-info/METADATA,sha256=KVhSn5v8Q0-GqYQPCMiJyKb10SqKtPqz5HzaMIpaG_o,8247
-prediction_market_agent_tooling-0.57.11.dev258.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
-prediction_market_agent_tooling-0.57.11.dev258.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
-prediction_market_agent_tooling-0.57.11.dev258.dist-info/RECORD,,
+prediction_market_agent_tooling-0.57.11.dev260.dist-info/LICENSE,sha256=6or154nLLU6bELzjh0mCreFjt0m2v72zLi3yHE0QbeE,7650
+prediction_market_agent_tooling-0.57.11.dev260.dist-info/METADATA,sha256=UwIBMbs6cJebVZKvmQSaqueRxMRBqBqSihSofJu3e8U,8247
+prediction_market_agent_tooling-0.57.11.dev260.dist-info/WHEEL,sha256=IYZQI976HJqqOpQU6PHkJ8fb3tMNBFjg-Cn-pwAbaFM,88
+prediction_market_agent_tooling-0.57.11.dev260.dist-info/entry_points.txt,sha256=m8PukHbeH5g0IAAmOf_1Ahm-sGAMdhSSRQmwtpmi2s8,81
+prediction_market_agent_tooling-0.57.11.dev260.dist-info/RECORD,,