prediction-market-agent-tooling 0.57.11.dev247__tar.gz → 0.57.11.dev254__tar.gz

{prediction_market_agent_tooling-0.57.11.dev247 → prediction_market_agent_tooling-0.57.11.dev254}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: prediction-market-agent-tooling
-Version: 0.57.11.dev247
+Version: 0.57.11.dev254
 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.dev254/prediction_market_agent_tooling/abis/simpletreasury.abi.json

@@ -0,0 +1,156 @@
+[
+  {
+    "inputs": [
+      {
+        "internalType": "address",
+        "name": "_nftContract",
+        "type": "address"
+      }
+    ],
+    "stateMutability": "nonpayable",
+    "type": "constructor"
+  },
+  {
+    "inputs": [
+      {
+        "internalType": "address",
+        "name": "owner",
+        "type": "address"
+      }
+    ],
+    "name": "OwnableInvalidOwner",
+    "type": "error"
+  },
+  {
+    "inputs": [
+      {
+        "internalType": "address",
+        "name": "account",
+        "type": "address"
+      }
+    ],
+    "name": "OwnableUnauthorizedAccount",
+    "type": "error"
+  },
+  {
+    "anonymous": false,
+    "inputs": [
+      {
+        "indexed": true,
+        "internalType": "address",
+        "name": "previousOwner",
+        "type": "address"
+      },
+      {
+        "indexed": true,
+        "internalType": "address",
+        "name": "newOwner",
+        "type": "address"
+      }
+    ],
+    "name": "OwnershipTransferred",
+    "type": "event"
+  },
+  {
+    "anonymous": false,
+    "inputs": [
+      {
+        "indexed": true,
+        "internalType": "address",
+        "name": "holder",
+        "type": "address"
+      },
+      {
+        "indexed": false,
+        "internalType": "uint256",
+        "name": "amount",
+        "type": "uint256"
+      }
+    ],
+    "name": "WithdrawnByNFTHolder",
+    "type": "event"
+  },
+  {
+    "inputs": [],
+    "name": "nftContract",
+    "outputs": [
+      {
+        "internalType": "contract IERC721",
+        "name": "",
+        "type": "address"
+      }
+    ],
+    "stateMutability": "view",
+    "type": "function"
+  },
+  {
+    "inputs": [],
+    "name": "owner",
+    "outputs": [
+      {
+        "internalType": "address",
+        "name": "",
+        "type": "address"
+      }
+    ],
+    "stateMutability": "view",
+    "type": "function"
+  },
+  {
+    "inputs": [],
+    "name": "renounceOwnership",
+    "outputs": [],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
+  {
+    "inputs": [],
+    "name": "requiredNFTBalance",
+    "outputs": [
+      {
+        "internalType": "uint256",
+        "name": "",
+        "type": "uint256"
+      }
+    ],
+    "stateMutability": "view",
+    "type": "function"
+  },
+  {
+    "inputs": [
+      {
+        "internalType": "uint256",
+        "name": "_newBalance",
+        "type": "uint256"
+      }
+    ],
+    "name": "setRequiredNFTBalance",
+    "outputs": [],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
+  {
+    "inputs": [
+      {
+        "internalType": "address",
+        "name": "newOwner",
+        "type": "address"
+      }
+    ],
+    "name": "transferOwnership",
+    "outputs": [],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
+  {
+    "inputs": [],
+    "name": "withdraw",
+    "outputs": [],
+    "stateMutability": "nonpayable",
+    "type": "function"
+  },
+  {
+    "stateMutability": "payable",
+    "type": "receive"
+  }
+]

{prediction_market_agent_tooling-0.57.11.dev247 → prediction_market_agent_tooling-0.57.11.dev254}/prediction_market_agent_tooling/tools/contract.py

@@ -390,7 +390,13 @@ class ContractERC4626BaseClass(ContractERC20BaseClass):
         return self.convertToShares(amount, web3=web3)
 
 
-class ContractOwnableERC721BaseClass(ContractBaseClass):
+class OwnableContract(ContractBaseClass):
+    def owner(self, web3: Web3 | None = None) -> ChecksumAddress:
+        owner = Web3.to_checksum_address(self.call("owner", web3=web3))
+        return owner
+
+
+class ContractERC721BaseClass(ContractBaseClass):
     abi: ABI = abi_field_validator(
         os.path.join(
             os.path.dirname(os.path.realpath(__file__)),
@@ -417,8 +423,8 @@ class ContractOwnableERC721BaseClass(ContractBaseClass):
         balance: int = self.call("balanceOf", [owner], web3=web3)
         return balance
 
-    def ownerOf(self, tokenId: int, web3: Web3 | None = None) -> ChecksumAddress:
-        owner = Web3.to_checksum_address(self.call("ownerOf", [tokenId], web3=web3))
+    def owner_of(self, token_id: int, web3: Web3 | None = None) -> ChecksumAddress:
+        owner = Web3.to_checksum_address(self.call("ownerOf", [token_id], web3=web3))
         return owner
 
     def name(self, web3: Web3 | None = None) -> str:
@@ -451,6 +457,10 @@ class ContractOwnableERC721BaseClass(ContractBaseClass):
         )
 
 
+class ContractOwnableERC721BaseClass(ContractERC721BaseClass, OwnableContract):
+    pass
+
+
 class ContractOnGnosisChain(ContractBaseClass):
     """
     Contract base class with Gnosis Chain configuration.
@@ -541,6 +551,51 @@ class DebuggingContract(ContractOnGnosisChain):
         )
 
 
+class SimpleTreasuryContract(ContractOnGnosisChain, OwnableContract):
+    # Contract ABI taken from built https://github.com/gnosis/labs-contracts.
+    abi: ABI = abi_field_validator(
+        os.path.join(
+            os.path.dirname(os.path.realpath(__file__)),
+            "../abis/simpletreasury.abi.json",
+        )
+    )
+
+    address: ChecksumAddress = Web3.to_checksum_address(
+        "0x624ad0db52e6b18afb4d36b8e79d0c2a74f3fc8a"
+    )
+
+    def required_nft_balance(self, web3: Web3 | None = None) -> int:
+        min_num_of_nfts: int = self.call("requiredNFTBalance", web3=web3)
+        return min_num_of_nfts
+
+    def set_required_nft_balance(
+        self,
+        api_keys: APIKeys,
+        web3: Web3 | None = None,
+    ) -> TxReceipt:
+        return self.send(
+            api_keys=api_keys,
+            function_name="setRequiredNFTBalance",
+            web3=web3,
+        )
+
+    def nft_contract(self, web3: Web3 | None = None) -> ContractERC721BaseClass:
+        nft_contract_address: ChecksumAddress = self.call("nftContract", web3=web3)
+        contract = ContractERC721BaseClass(address=nft_contract_address)
+        return contract
+
+    def withdraw(
+        self,
+        api_keys: APIKeys,
+        web3: Web3 | None = None,
+    ) -> TxReceipt:
+        return self.send(
+            api_keys=api_keys,
+            function_name="withdraw",
+            web3=web3,
+        )
+
+
 class AgentCommunicationContract(ContractOnGnosisChain):
     # Contract ABI taken from built https://github.com/gnosis/labs-contracts.
     abi: ABI = abi_field_validator(

{prediction_market_agent_tooling-0.57.11.dev247 → prediction_market_agent_tooling-0.57.11.dev254}/pyproject.toml

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