prediction-market-agent-tooling 0.57.11.dev258__tar.gz → 0.57.11.dev259__tar.gz

{prediction_market_agent_tooling-0.57.11.dev258 → prediction_market_agent_tooling-0.57.11.dev259}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.57.11.dev258
+Version: 0.57.11.dev259
 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.dev259/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-0.57.11.dev258 → prediction_market_agent_tooling-0.57.11.dev259}/prediction_market_agent_tooling/tools/contract.py

@@ -78,6 +78,11 @@ 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 = 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 +396,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 +470,9 @@ class ContractERC721BaseClass(ContractBaseClass):
 
 
 class ContractOwnableERC721BaseClass(ContractERC721BaseClass, OwnableContract):
-    pass
+    abi: ABI = ContractBaseClass.merge_contract_abis(
+        [ContractERC721BaseClass(address=""), OwnableContract(address="")]
+    )
 
 
 class ContractOnGnosisChain(ContractBaseClass):
@@ -571,11 +585,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 → prediction_market_agent_tooling-0.57.11.dev259}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "prediction-market-agent-tooling"
-version = "0.57.11.dev258"
+version = "0.57.11.dev259"
 description = "Tools to benchmark, deploy and monitor prediction market agents."
 authors = ["Gnosis"]
 readme = "README.md"