oli-python 0.1.0__py3-none-any.whl

oli/attestation/utils_validator.py

@@ -0,0 +1,200 @@
+class UtilsValidator:
+    def __init__(self, oli_client):
+        """
+        Initialize the DataValidator with an OLI client.
+        
+        Args:
+            oli_client: The OLI client instance
+        """
+        self.oli = oli_client
+        self.allowed_prefixes = [
+            'eip155:',  # Ethereum and EVM-compatible chains
+            'solana:',  # Solana
+            'tron:',    # TRON
+            'stellar:', # Stellar
+            'bip122:',  # Bitcoin
+            'SN_'       # Starknet
+        ]
+
+    def fix_simple_tags_formatting(self, tags: dict) -> dict:
+        """
+        Fix basic formatting in the tags dictionary. This includes:
+        - Ensuring all tag_ids and their value are lowercase
+        - Booling values are converted from strings to booleans
+        - Removing leading/trailing whitespace from string values
+        - Checksum address (string(42)) tags
+        
+        Args:
+            tags (dict): Dictionary of tags
+            
+        Returns:
+            dict: Formatted tags
+        """
+        # Convert tag_ids to lowercase
+        tags = {k.lower(): v for k, v in tags.items()}
+
+        # Convert all tag_values to lower case & strip whitespaces, then single boolean values from strings to booleans
+        for k, v in tags.items():
+            if isinstance(v, str):
+                tags[k] = v.strip().lower()
+                if tags[k] == 'true':
+                    tags[k] = True
+                elif tags[k] == 'false':
+                    tags[k] = False
+            elif isinstance(v, list):
+                tags[k] = [i.strip().lower() if isinstance(i, str) else i for i in v]
+
+        # Checksum address (string(42)) and transaction hash (string(66)) tags
+        for k, v in tags.items():
+            if k in self.oli.tag_definitions and self.oli.tag_definitions[k]['type'] == 'string(42)':
+                tags[k] = self.oli.w3.to_checksum_address(v)
+
+        return tags
+
+    def check_label_correctness(self, address: str, chain_id: str, tags: dict, ref_uid: str="0x0000000000000000000000000000000000000000000000000000000000000000", auto_fix: bool=True) -> bool:
+        """
+        Check if the label is compliant with the OLI Data Model. See OLI Github documentation for more details: https://github.com/openlabelsinitiative/OLI
+        
+        Args:
+            address (str): Address to check
+            chain_id (str): Chain ID to check
+            tags (dict): Tags to check
+            ref_uid (str): Reference UID to check
+            auto_fix (bool): If True, will attempt to fix the label automatically using the fix_simple_tags_formatting function
+            
+        Returns:
+            bool: True if the label is correct, False otherwise
+        """
+        # basic checks
+        self.checks_address(address)
+        self.checks_chain_id(chain_id)
+        self.checks_tags(tags, auto_fix=auto_fix)
+        self.checks_ref_uid(ref_uid)
+        return True
+        
+    def checks_chain_id(self, chain_id: str) -> bool:
+        """
+        Check if chain_id for a label is in CAIP-2 format.
+        
+        Args:
+            chain_id (str): Chain ID to check
+            
+        Returns:
+            bool: True if correct, False otherwise
+        """
+        # Check if the chain_id starts with any of the allowed prefixes
+        for prefix in self.allowed_prefixes:
+            if chain_id.startswith(prefix):
+                # For eip155, further validate that the rest is a number or 'any'
+                if prefix == 'eip155:':
+                    rest = chain_id[len(prefix):]
+                    if rest.isdigit():
+                        return True
+                    elif rest == 'any':
+                        print("Please ensure the label is accurate and consistent across all EVM chains before setting chain_id = 'eip155:any'.")
+                        return True
+                    else:
+                        print(f"Invalid eip155 chain_id format: {chain_id}")
+                        raise ValueError("For eip155 chains, format must be 'eip155:' followed by a number or 'any'")
+                return True
+        
+        # If we get here, the chain_id didn't match any allowed format
+        print(f"Unsupported chain ID format: {chain_id}")
+        raise ValueError("Chain ID must be in CAIP-2 format (e.g., Base -> 'eip155:8453'), see this guide on CAIP-2: https://docs.portalhq.io/resources/chain-id-formatting")
+
+    def checks_address(self, address: str) -> bool:
+        """
+        Check if address is a valid Ethereum address.
+        
+        Args:
+            address (str): Address to check
+            
+        Returns:
+            bool: True if correct, False otherwise
+        """
+        if self.oli.w3.is_address(address):
+            return True
+        else:
+            print(address)
+            raise ValueError("Address must be a valid Ethereum address in hex format")
+        
+    def checks_tags(self, tags: dict, auto_fix: bool=False) -> bool:
+        """
+        Check if tags are in the correct format.
+        
+        Args:
+            tags (dict): Tags to check
+            
+        Returns:
+            bool: True if correct, False otherwise
+        """
+        # Check if tags is a dictionary
+        if isinstance(tags, dict):
+            if auto_fix:
+                tags = self.fix_simple_tags_formatting(tags)
+            else:
+                pass
+        else:
+            print(tags)
+            raise ValueError("Tags must be a dictionary with OLI compliant tags (e.g., {'contract_name': 'example', 'is_eoa': True})")
+        
+        # Check each tag_id in the dictionary
+        for tag_id in tags.keys():
+            
+            # Check if the tag_id is in the official OLI tag list
+            if tag_id not in self.oli.tag_ids:
+                print(f"WARNING: Tag tag_id '{tag_id}' is not an official OLI tag. Please check the 'oli.tag_definitions' or https://github.com/openlabelsinitiative/OLI/blob/main/1_data_model/tags/tag_definitions.yml.")
+            
+            # Check if the tag_id is in the correct format. So far implemented [boolean, string, integer, list, float, string(42), string(66), date (YYYY-MM-DD HH:MM:SS)]
+            else:
+                if self.oli.tag_definitions[tag_id]['type'] == 'boolean' and not isinstance(tags[tag_id], bool):
+                    print(f"WARNING: Tag value for {tag_id} must be a boolean (True/False).")
+                elif self.oli.tag_definitions[tag_id]['type'] == 'string' and not isinstance(tags[tag_id], str):
+                    print(f"WARNING: Tag value for {tag_id} must be a string.")
+                elif self.oli.tag_definitions[tag_id]['type'] == 'integer' and not isinstance(tags[tag_id], int):
+                    print(f"WARNING: Tag value for {tag_id} must be an integer.")
+                elif self.oli.tag_definitions[tag_id]['type'] == 'float' and not isinstance(tags[tag_id], float):
+                    print(f"WARNING: Tag value for {tag_id} must be a float.")
+                elif self.oli.tag_definitions[tag_id]['type'] == 'list' and not isinstance(tags[tag_id], list):
+                    print(f"WARNING: Tag value for {tag_id} must be a list.")
+                elif self.oli.tag_definitions[tag_id]['type'] == 'string(42)' and not self.oli.w3.is_address(tags[tag_id]):
+                    print(f"WARNING: Tag value for {tag_id} must be a valid Ethereum address string with '0x'.")
+                elif self.oli.tag_definitions[tag_id]['type'] == 'string(66)' and not (len(tags[tag_id]) == 66 and tags[tag_id].startswith('0x')):
+                    print(f"WARNING: Tag value for {tag_id} must be a valid hex string with '0x' prefix and 64 hex characters (66 characters total).")
+                elif self.oli.tag_definitions[tag_id]['type'] == 'date (YYYY-MM-DD HH:MM:SS)' and not isinstance(tags[tag_id], str):
+                    print(f"WARNING: Tag value for {tag_id} must be a string in the format 'YYYY-MM-DD HH:MM:SS'.")
+
+            # Check if the value is in the value set
+            if tag_id in self.oli.tag_value_sets:
+                # single value
+                if tags[tag_id] not in self.oli.tag_value_sets[tag_id] and not isinstance(tags[tag_id], list):
+                    print(f"WARNING: Invalid tag value for {tag_id}: '{tags[tag_id]}'")
+                    if len(self.oli.tag_value_sets[tag_id]) < 100:
+                        print(f"Please use one of the following values for {tag_id}: {self.oli.tag_value_sets[tag_id]}")
+                    else:
+                        print(f"Please use a valid value from the predefined value_set for {tag_id}: {self.oli.tag_definitions[tag_id]['value_set']}")
+                # list of values
+                elif tags[tag_id] not in self.oli.tag_value_sets[tag_id] and isinstance(tags[tag_id], list):
+                    for i in tags[tag_id]:
+                        if i not in self.oli.tag_value_sets[tag_id]:
+                            print(f"WARNING: Invalid tag value for {tag_id}: {i}")
+                            if len(self.oli.tag_value_sets[tag_id]) < 100:
+                                print(f"Please use a list of values from the predefined value_set for {tag_id}: {self.oli.tag_value_sets[tag_id]}")
+                            else:
+                                print(f"Please use a list of values from the predefined value_set for {tag_id}: {self.oli.tag_definitions[tag_id]['value_set']}")
+
+    def checks_ref_uid(self, ref_uid: str) -> bool:
+        """
+        Check if ref_uid is a valid UID.
+        
+        Args:
+            ref_uid (str): Reference UID to check
+            
+        Returns:
+            bool: True if correct, throws error otherwise
+        """
+        if ref_uid.startswith('0x') and len(ref_uid) == 66:
+            return True
+        else:
+            print(ref_uid)
+            raise ValueError("Ref_uid must be a valid UID in hex format, leave empty if not used")

oli/core.py

@@ -0,0 +1,128 @@
+from web3 import Web3
+import eth_account
+from eth_keys import keys
+from requests import Response
+
+from oli.attestation.utils_validator import UtilsValidator
+from oli.attestation.utils_other import UtilsOther
+from oli.attestation.onchain import OnchainAttestations
+from oli.attestation.offchain import OffchainAttestations
+from oli.data.fetcher import DataFetcher
+from oli.data.graphql import GraphQLClient
+
+class OLI:
+    def __init__(self, private_key: str, is_production: bool=True, custom_rpc_url: str=None) -> None:
+        """
+        Initialize the OLI API client.
+        
+        Args:
+            private_key (str): The private key to sign attestations
+            is_production (bool): Whether to use production or testnet
+        """
+        print("Initializing OLI API client...")
+
+        # Set network based on environment
+        if is_production:
+            self.rpc = "https://mainnet.base.org"
+            self.graphql = "https://base.easscan.org/graphql"
+            self.rpc_chain_number = 8453
+            self.eas_api_url = "https://base.easscan.org/offchain/store"
+            self.eas_address = "0x4200000000000000000000000000000000000021"  # EAS contract address on mainnet
+        else:
+            self.rpc = "https://sepolia.base.org"
+            self.graphql = "https://base-sepolia.easscan.org/graphql"
+            self.rpc_chain_number = 84532
+            self.eas_api_url = "https://base-sepolia.easscan.org/offchain/store"
+            self.eas_address = "0x4200000000000000000000000000000000000021"  # EAS contract address on testnet
+        
+        # Use provided RPC endpoint if specified
+        if custom_rpc_url is not None:
+            self.rpc = custom_rpc_url
+
+        # Initialize Web3 and account
+        self.w3 = Web3(Web3.HTTPProvider(self.rpc))
+        if not self.w3.is_connected():
+            raise Exception("Failed to connect to the Ethereum node")
+            
+        # Convert the hex private key to the proper key object
+        self.private_key = private_key
+        if private_key.startswith('0x'):
+            private_key_bytes = private_key[2:]
+        else:
+            private_key_bytes = private_key
+        private_key_obj = keys.PrivateKey(bytes.fromhex(private_key_bytes))
+        
+        # Create account from private key
+        self.account = eth_account.Account.from_key(private_key_obj)
+        self.address = self.account.address
+        
+        # Label Pool Schema for OLI
+        self.oli_label_pool_schema = '0xb763e62d940bed6f527dd82418e146a904e62a297b8fa765c9b3e1f0bc6fdd68'
+        
+        # Load EAS ABI
+        self.eas_abi = '[{"inputs": [],"stateMutability": "nonpayable","type": "constructor"},{"inputs": [],"name": "AccessDenied","type": "error"},{"inputs": [],"name": "AlreadyRevoked","type": "error"},{"inputs": [],"name": "AlreadyRevokedOffchain","type": "error"},{"inputs": [],"name": "AlreadyTimestamped","type": "error"},{"inputs": [],"name": "DeadlineExpired","type": "error"},{"inputs": [],"name": "InsufficientValue","type": "error"},{"inputs": [],"name": "InvalidAttestation","type": "error"},{"inputs": [],"name": "InvalidAttestations","type": "error"},{"inputs": [],"name": "InvalidExpirationTime","type": "error"},{"inputs": [],"name": "InvalidLength","type": "error"},{"inputs": [],"name": "InvalidNonce","type": "error"},{"inputs": [],"name": "InvalidOffset","type": "error"},{"inputs": [],"name": "InvalidRegistry","type": "error"},{"inputs": [],"name": "InvalidRevocation","type": "error"},{"inputs": [],"name": "InvalidRevocations","type": "error"},{"inputs": [],"name": "InvalidSchema","type": "error"},{"inputs": [],"name": "InvalidSignature","type": "error"},{"inputs": [],"name": "InvalidVerifier","type": "error"},{"inputs": [],"name": "Irrevocable","type": "error"},{"inputs": [],"name": "NotFound","type": "error"},{"inputs": [],"name": "NotPayable","type": "error"},{"inputs": [],"name": "WrongSchema","type": "error"},{"anonymous": false,"inputs": [{"indexed": true,"internalType": "address","name": "recipient","type": "address"},{"indexed": true,"internalType": "address","name": "attester","type": "address"},{"indexed": false,"internalType": "bytes32","name": "uid","type": "bytes32"},{"indexed": true,"internalType": "bytes32","name": "schemaUID","type": "bytes32"}],"name": "Attested","type": "event"},{"anonymous": false,"inputs": [{"indexed": false,"internalType": "uint256","name": "oldNonce","type": "uint256"},{"indexed": false,"internalType": "uint256","name": "newNonce","type": "uint256"}],"name": "NonceIncreased","type": "event"},{"anonymous": false,"inputs": [{"indexed": true,"internalType": "address","name": "recipient","type": "address"},{"indexed": true,"internalType": "address","name": "attester","type": "address"},{"indexed": false,"internalType": "bytes32","name": "uid","type": "bytes32"},{"indexed": true,"internalType": "bytes32","name": "schemaUID","type": "bytes32"}],"name": "Revoked","type": "event"},{"anonymous": false,"inputs": [{"indexed": true,"internalType": "address","name": "revoker","type": "address"},{"indexed": true,"internalType": "bytes32","name": "data","type": "bytes32"},{"indexed": true,"internalType": "uint64","name": "timestamp","type": "uint64"}],"name": "RevokedOffchain","type": "event"},{"anonymous": false,"inputs": [{"indexed": true,"internalType": "bytes32","name": "data","type": "bytes32"},{"indexed": true,"internalType": "uint64","name": "timestamp","type": "uint64"}],"name": "Timestamped","type": "event"},{"inputs": [{"components": [{"internalType": "bytes32","name": "schema","type": "bytes32"},{"components": [{"internalType": "address","name": "recipient","type": "address"},{"internalType": "uint64","name": "expirationTime","type": "uint64"},{"internalType": "bool","name": "revocable","type": "bool"},{"internalType": "bytes32","name": "refUID","type": "bytes32"},{"internalType": "bytes","name": "data","type": "bytes"},{"internalType": "uint256","name": "value","type": "uint256"}],"internalType": "struct AttestationRequestData","name": "data","type": "tuple"}],"internalType": "struct AttestationRequest","name": "request","type": "tuple"}],"name": "attest","outputs": [{"internalType": "bytes32","name": "","type": "bytes32"}],"stateMutability": "payable","type": "function"},{"inputs": [{"components": [{"internalType": "bytes32","name": "schema","type": "bytes32"},{"components": [{"internalType": "address","name": "recipient","type": "address"},{"internalType": "uint64","name": "expirationTime","type": "uint64"},{"internalType": "bool","name": "revocable","type": "bool"},{"internalType": "bytes32","name": "refUID","type": "bytes32"},{"internalType": "bytes","name": "data","type": "bytes"},{"internalType": "uint256","name": "value","type": "uint256"}],"internalType": "struct AttestationRequestData","name": "data","type": "tuple"},{"components": [{"internalType": "uint8","name": "v","type": "uint8"},{"internalType": "bytes32","name": "r","type": "bytes32"},{"internalType": "bytes32","name": "s","type": "bytes32"}],"internalType": "struct Signature","name": "signature","type": "tuple"},{"internalType": "address","name": "attester","type": "address"},{"internalType": "uint64","name": "deadline","type": "uint64"}],"internalType": "struct DelegatedAttestationRequest","name": "delegatedRequest","type": "tuple"}],"name": "attestByDelegation","outputs": [{"internalType": "bytes32","name": "","type": "bytes32"}],"stateMutability": "payable","type": "function"},{"inputs": [],"name": "getAttestTypeHash","outputs": [{"internalType": "bytes32","name": "","type": "bytes32"}],"stateMutability": "pure","type": "function"},{"inputs": [{"internalType": "bytes32","name": "uid","type": "bytes32"}],"name": "getAttestation","outputs": [{"components": [{"internalType": "bytes32","name": "uid","type": "bytes32"},{"internalType": "bytes32","name": "schema","type": "bytes32"},{"internalType": "uint64","name": "time","type": "uint64"},{"internalType": "uint64","name": "expirationTime","type": "uint64"},{"internalType": "uint64","name": "revocationTime","type": "uint64"},{"internalType": "bytes32","name": "refUID","type": "bytes32"},{"internalType": "address","name": "recipient","type": "address"},{"internalType": "address","name": "attester","type": "address"},{"internalType": "bool","name": "revocable","type": "bool"},{"internalType": "bytes","name": "data","type": "bytes"}],"internalType": "struct Attestation","name": "","type": "tuple"}],"stateMutability": "view","type": "function"},{"inputs": [],"name": "getDomainSeparator","outputs": [{"internalType": "bytes32","name": "","type": "bytes32"}],"stateMutability": "view","type": "function"},{"inputs": [],"name": "getName","outputs": [{"internalType": "string","name": "","type": "string"}],"stateMutability": "view","type": "function"},{"inputs": [{"internalType": "address","name": "account","type": "address"}],"name": "getNonce","outputs": [{"internalType": "uint256","name": "","type": "uint256"}],"stateMutability": "view","type": "function"},{"inputs": [{"internalType": "address","name": "revoker","type": "address"},{"internalType": "bytes32","name": "data","type": "bytes32"}],"name": "getRevokeOffchain","outputs": [{"internalType": "uint64","name": "","type": "uint64"}],"stateMutability": "view","type": "function"},{"inputs": [],"name": "getRevokeTypeHash","outputs": [{"internalType": "bytes32","name": "","type": "bytes32"}],"stateMutability": "pure","type": "function"},{"inputs": [],"name": "getSchemaRegistry","outputs": [{"internalType": "contract ISchemaRegistry","name": "","type": "address"}],"stateMutability": "pure","type": "function"},{"inputs": [{"internalType": "bytes32","name": "data","type": "bytes32"}],"name": "getTimestamp","outputs": [{"internalType": "uint64","name": "","type": "uint64"}],"stateMutability": "view","type": "function"},{"inputs": [{"internalType": "uint256","name": "newNonce","type": "uint256"}],"name": "increaseNonce","outputs": [],"stateMutability": "nonpayable","type": "function"},{"inputs": [{"internalType": "bytes32","name": "uid","type": "bytes32"}],"name": "isAttestationValid","outputs": [{"internalType": "bool","name": "","type": "bool"}],"stateMutability": "view","type": "function"},{"inputs": [{"components": [{"internalType": "bytes32","name": "schema","type": "bytes32"},{"components": [{"internalType": "address","name": "recipient","type": "address"},{"internalType": "uint64","name": "expirationTime","type": "uint64"},{"internalType": "bool","name": "revocable","type": "bool"},{"internalType": "bytes32","name": "refUID","type": "bytes32"},{"internalType": "bytes","name": "data","type": "bytes"},{"internalType": "uint256","name": "value","type": "uint256"}],"internalType": "struct AttestationRequestData[]","name": "data","type": "tuple[]"}],"internalType": "struct MultiAttestationRequest[]","name": "multiRequests","type": "tuple[]"}],"name": "multiAttest","outputs": [{"internalType": "bytes32[]","name": "","type": "bytes32[]"}],"stateMutability": "payable","type": "function"},{"inputs": [{"components": [{"internalType": "bytes32","name": "schema","type": "bytes32"},{"components": [{"internalType": "address","name": "recipient","type": "address"},{"internalType": "uint64","name": "expirationTime","type": "uint64"},{"internalType": "bool","name": "revocable","type": "bool"},{"internalType": "bytes32","name": "refUID","type": "bytes32"},{"internalType": "bytes","name": "data","type": "bytes"},{"internalType": "uint256","name": "value","type": "uint256"}],"internalType": "struct AttestationRequestData[]","name": "data","type": "tuple[]"},{"components": [{"internalType": "uint8","name": "v","type": "uint8"},{"internalType": "bytes32","name": "r","type": "bytes32"},{"internalType": "bytes32","name": "s","type": "bytes32"}],"internalType": "struct Signature[]","name": "signatures","type": "tuple[]"},{"internalType": "address","name": "attester","type": "address"},{"internalType": "uint64","name": "deadline","type": "uint64"}],"internalType": "struct MultiDelegatedAttestationRequest[]","name": "multiDelegatedRequests","type": "tuple[]"}],"name": "multiAttestByDelegation","outputs": [{"internalType": "bytes32[]","name": "","type": "bytes32[]"}],"stateMutability": "payable","type": "function"},{"inputs": [{"components": [{"internalType": "bytes32","name": "schema","type": "bytes32"},{"components": [{"internalType": "bytes32","name": "uid","type": "bytes32"},{"internalType": "uint256","name": "value","type": "uint256"}],"internalType": "struct RevocationRequestData[]","name": "data","type": "tuple[]"}],"internalType": "struct MultiRevocationRequest[]","name": "multiRequests","type": "tuple[]"}],"name": "multiRevoke","outputs": [],"stateMutability": "payable","type": "function"},{"inputs": [{"components": [{"internalType": "bytes32","name": "schema","type": "bytes32"},{"components": [{"internalType": "bytes32","name": "uid","type": "bytes32"},{"internalType": "uint256","name": "value","type": "uint256"}],"internalType": "struct RevocationRequestData[]","name": "data","type": "tuple[]"},{"components": [{"internalType": "uint8","name": "v","type": "uint8"},{"internalType": "bytes32","name": "r","type": "bytes32"},{"internalType": "bytes32","name": "s","type": "bytes32"}],"internalType": "struct Signature[]","name": "signatures","type": "tuple[]"},{"internalType": "address","name": "revoker","type": "address"},{"internalType": "uint64","name": "deadline","type": "uint64"}],"internalType": "struct MultiDelegatedRevocationRequest[]","name": "multiDelegatedRequests","type": "tuple[]"}],"name": "multiRevokeByDelegation","outputs": [],"stateMutability": "payable","type": "function"},{"inputs": [{"internalType": "bytes32[]","name": "data","type": "bytes32[]"}],"name": "multiRevokeOffchain","outputs": [{"internalType": "uint64","name": "","type": "uint64"}],"stateMutability": "nonpayable","type": "function"},{"inputs": [{"internalType": "bytes32[]","name": "data","type": "bytes32[]"}],"name": "multiTimestamp","outputs": [{"internalType": "uint64","name": "","type": "uint64"}],"stateMutability": "nonpayable","type": "function"},{"inputs": [{"components": [{"internalType": "bytes32","name": "schema","type": "bytes32"},{"components": [{"internalType": "bytes32","name": "uid","type": "bytes32"},{"internalType": "uint256","name": "value","type": "uint256"}],"internalType": "struct RevocationRequestData","name": "data","type": "tuple"}],"internalType": "struct RevocationRequest","name": "request","type": "tuple"}],"name": "revoke","outputs": [],"stateMutability": "payable","type": "function"},{"inputs": [{"components": [{"internalType": "bytes32","name": "schema","type": "bytes32"},{"components": [{"internalType": "bytes32","name": "uid","type": "bytes32"},{"internalType": "uint256","name": "value","type": "uint256"}],"internalType": "struct RevocationRequestData","name": "data","type": "tuple"},{"components": [{"internalType": "uint8","name": "v","type": "uint8"},{"internalType": "bytes32","name": "r","type": "bytes32"},{"internalType": "bytes32","name": "s","type": "bytes32"}],"internalType": "struct Signature","name": "signature","type": "tuple"},{"internalType": "address","name": "revoker","type": "address"},{"internalType": "uint64","name": "deadline","type": "uint64"}],"internalType": "struct DelegatedRevocationRequest","name": "delegatedRequest","type": "tuple"}],"name": "revokeByDelegation","outputs": [],"stateMutability": "payable","type": "function"},{"inputs": [{"internalType": "bytes32","name": "data","type": "bytes32"}],"name": "revokeOffchain","outputs": [{"internalType": "uint64","name": "","type": "uint64"}],"stateMutability": "nonpayable","type": "function"},{"inputs": [{"internalType": "bytes32","name": "data","type": "bytes32"}],"name": "timestamp","outputs": [{"internalType": "uint64","name": "","type": "uint64"}],"stateMutability": "nonpayable","type": "function"},{"inputs": [],"name": "version","outputs": [{"internalType": "string","name": "","type": "string"}],"stateMutability": "view","type": "function"}]'
+
+        # Initialize EAS contract
+        self.eas = self.w3.eth.contract(address=self.eas_address, abi=self.eas_abi)
+    
+        # Initialize components
+        self.data_fetcher = DataFetcher(self)
+        self.tag_definitions = self.data_fetcher.get_OLI_tags()
+        self.tag_ids = list(self.tag_definitions.keys())
+        self.tag_value_sets = self.data_fetcher.get_OLI_value_sets()
+        
+        # Initialize validator
+        self.validator = UtilsValidator(self)
+        
+        # Initialize other utilities
+        self.utils_other = UtilsOther(self)
+        
+        # Initialize onchain and offchain attestations
+        self.onchain = OnchainAttestations(self)
+        self.offchain = OffchainAttestations(self)
+        
+        # Initialize GraphQL client
+        self.graphql_client = GraphQLClient(self)
+
+        print("...OLI client initialized successfully.")
+    
+    # Expose onchain attestation methods
+    def create_onchain_label(self, address: str, chain_id: str, tags: dict, ref_uid: str="0x0000000000000000000000000000000000000000000000000000000000000000", gas_limit: int=0) -> tuple[str, str]:
+        return self.onchain.create_onchain_label(address, chain_id, tags, ref_uid, gas_limit)
+    
+    def create_multi_onchain_labels(self, labels: list, gas_limit: int=0) -> tuple[str, list]:
+        return self.onchain.create_multi_onchain_labels(labels, gas_limit)
+    
+    # Expose offchain attestation methods
+    def create_offchain_label(self, address: str, chain_id: str, tags: dict, ref_uid: str="0x0000000000000000000000000000000000000000000000000000000000000000", retry: int=4) -> Response:
+        return self.offchain.create_offchain_label(address, chain_id, tags, ref_uid, retry)
+    
+    # Expose revocation methods
+    def revoke_attestation(self, uid_hex: str, onchain: bool, gas_limit: int=200000) -> str:
+        if onchain:
+            return self.onchain.revoke_attestation(uid_hex, gas_limit)
+        else:
+            return self.offchain.revoke_attestation(uid_hex, gas_limit)
+    
+    def multi_revoke_attestations(self, uids: str, onchain: bool, gas_limit: int=10000000) -> str:
+        if onchain:
+            return self.onchain.multi_revoke_attestations(uids, gas_limit)
+        else:
+            return self.offchain.multi_revoke_attestations(uids, gas_limit)
+    
+    # Expose query methods
+    def graphql_query_attestations(self, address: str=None, attester: str=None, timeCreated: int=None, revocationTime: int=None) -> dict:
+        return self.graphql_client.graphql_query_attestations(address, attester, timeCreated, revocationTime)
+    
+    def get_full_raw_export_parquet(self, file_path: str="raw_labels.parquet") -> str:
+        return self.data_fetcher.get_full_raw_export_parquet(file_path)
+    
+    def get_full_decoded_export_parquet(self, file_path: str="decoded_labels.parquet") -> str:
+        return self.data_fetcher.get_full_decoded_export_parquet(file_path)
+    
+    # Expose validation methods
+    def check_label_correctness(self, address: str, chain_id: str, tags: dict, ref_uid: str="0x0000000000000000000000000000000000000000000000000000000000000000", auto_fix: bool=True) -> bool:
+        return self.validator.check_label_correctness(address, chain_id, tags, ref_uid, auto_fix)
+    
+    def fix_simple_tags_formatting(self, tags: dict) -> dict:
+        return self.validator.fix_simple_tags_formatting(tags)

oli/data/__init__.py

@@ -0,0 +1,4 @@
+from oli.data.graphql import GraphQLClient
+from oli.data.fetcher import DataFetcher
+
+__all__ = ["GraphQLClient", "DataFetcher"]

oli/data/fetcher.py

@@ -0,0 +1,111 @@
+import requests
+import yaml
+
+class DataFetcher:
+    def __init__(self, oli_client):
+        """
+        Initialize the DataFetcher with an OLI client.
+        
+        Args:
+            oli_client: The OLI client instance
+        """
+        self.oli = oli_client
+    
+    def get_OLI_tags(self):
+        """
+        Get latest OLI tags from OLI Github repo.
+        
+        Returns:
+            dict: Dictionary of official OLI tags
+        """
+        url = "https://raw.githubusercontent.com/openlabelsinitiative/OLI/refs/heads/main/1_data_model/tags/tag_definitions.yml"
+        response = requests.get(url)
+        if response.status_code == 200:
+            y = yaml.safe_load(response.text)
+            y = {i['tag_id']: i for i in y['tags']}
+            return y
+        else:
+            raise Exception(f"Failed to fetch OLI tags from Github: {response.status_code} - {response.text}")
+
+    def get_OLI_value_sets(self) -> dict:
+        """
+        Get latest value sets for OLI tags.
+        
+        Returns:
+            dict: Dictionary of value sets with tag_id as key
+        """
+        value_sets = {}
+
+        # value sets from self.oli.tag_definitions (must be a list)
+        additional_value_sets = {i['tag_id']: i['value_set'] for i in self.oli.tag_definitions.values() if 'value_set' in i}
+        for tag_id, value_set in additional_value_sets.items():
+            if isinstance(value_set, list):
+                # convert all string values to lowercase and keep the rest as is
+                value_set = [i.lower() if isinstance(i, str) else i for i in value_set]
+                value_sets[tag_id] = value_set
+
+        # value set for owner_project
+        url = "https://api.growthepie.xyz/v1/labels/projects.json" 
+        response = requests.get(url)
+        if response.status_code == 200:
+            y = yaml.safe_load(response.text)
+            value_sets["owner_project"] = [i[0] for i in y['data']['data']]
+            value_sets["owner_project"] = [i.lower() if isinstance(i, str) else i for i in value_sets["owner_project"]]
+        else:
+            raise Exception(f"Failed to fetch owner_project value set from grwothepie projects api: {response.status_code} - {response.text}")
+
+        # value set for usage_category
+        url = "https://raw.githubusercontent.com/openlabelsinitiative/OLI/refs/heads/main/1_data_model/tags/valuesets/usage_category.yml"
+        response = requests.get(url)
+        if response.status_code == 200:
+            y = yaml.safe_load(response.text)
+            value_sets['usage_category'] = [i['category_id'] for i in y['categories']]
+            value_sets['usage_category'] = [i.lower() if isinstance(i, str) else i for i in value_sets['usage_category']]
+        else:
+            raise Exception(f"Failed to fetch usage_category value set from OLI Github: {response.status_code} - {response.text}")
+
+        return value_sets
+    
+    def get_full_raw_export_parquet(self, file_path: str="raw_labels.parquet") -> str:
+        """
+        Downloads the full raw export of all attestations in Parquet format.
+        
+        Args:
+            file_path (str): Path where the file will be saved. Defaults to "raw_labels.parquet".
+            
+        Returns:
+            str: Path to the downloaded Parquet file
+        """
+        url = "https://api.growthepie.xyz/v1/oli/labels_raw.parquet"
+        
+        response = requests.get(url, stream=True)
+        if response.status_code == 200:
+            with open(file_path, 'wb') as f:
+                f.write(response.content)
+            print(f"Downloaded and saved: {file_path}")
+            return file_path
+        else:
+            print(f"Failed to download {url}. Status code: {response.status_code}")
+            return None
+
+    def get_full_decoded_export_parquet(self, file_path: str="decoded_labels.parquet") -> str:
+        """
+        Downloads the full decoded export of all attestations in Parquet format.
+        
+        Args:
+            file_path (str): Path where the file will be saved. Defaults to "decoded_labels.parquet".
+            
+        Returns:
+            str: Path to the downloaded Parquet file
+        """
+        url = "https://api.growthepie.xyz/v1/oli/labels_decoded.parquet"
+        
+        response = requests.get(url, stream=True)
+        if response.status_code == 200:
+            with open(file_path, 'wb') as f:
+                f.write(response.content)
+            print(f"Downloaded and saved: {file_path}")
+            return file_path
+        else:
+            print(f"Failed to download {url}. Status code: {response.status_code}")
+            return None

oli/data/graphql.py

@@ -0,0 +1,87 @@
+import requests
+
+class GraphQLClient:
+    def __init__(self, oli_client):
+        """
+        Initialize the GraphQLClient with an OLI client.
+        
+        Args:
+            oli_client: The OLI client instance
+        """
+        self.oli = oli_client
+    
+    def graphql_query_attestations(self, address: str=None, attester: str=None, timeCreated: int=None, revocationTime: int=None) -> dict:
+        """
+        Queries attestations from the EAS GraphQL API based on the specified filters.
+        
+        Args:
+            address (str, optional): Ethereum address of the labeled contract
+            attester (str, optional): Ethereum address of the attester
+            timeCreated (int, optional): Filter for attestations created after this timestamp
+            revocationTime (int, optional): Filter for attestations with revocation time >= this timestamp
+            
+        Returns:
+            dict: JSON response containing matching attestation data
+        """
+        query = """
+            query Attestations($take: Int, $where: AttestationWhereInput, $orderBy: [AttestationOrderByWithRelationInput!]) {
+                attestations(take: $take, where: $where, orderBy: $orderBy) {
+                    attester
+                    decodedDataJson
+                    expirationTime
+                    id
+                    ipfsHash
+                    isOffchain
+                    recipient
+                    refUID
+                    revocable
+                    revocationTime
+                    revoked
+                    time
+                    timeCreated
+                    txid
+                }
+            }
+        """
+            
+        variables = {
+            "where": {
+                "schemaId": {
+                    "equals": self.oli.oli_label_pool_schema
+                }
+            },
+            "orderBy": [
+                {
+                "timeCreated": "desc"
+                }
+            ]
+        }
+        
+        # Add address to where clause if not None
+        if address is not None:
+            variables["where"]["recipient"] = {"equals": address}
+
+        # Add attester to where clause if not None
+        if attester is not None:
+            variables["where"]["attester"] = {"equals": attester}
+        
+        # Add timeCreated to where clause if not None, ensuring it's an int
+        if timeCreated is not None:
+            timeCreated = int(timeCreated)
+            variables["where"]["timeCreated"] = {"gt": timeCreated}
+        
+        # Add revocationTime to where clause if not None, ensuring it's an int
+        if revocationTime is not None:
+            revocationTime = int(revocationTime)
+            variables["where"]["revocationTime"] = {"gte": revocationTime}
+        
+        headers = {
+            "Content-Type": "application/json"
+        }
+        
+        response = requests.post(self.oli.graphql, json={"query": query, "variables": variables}, headers=headers)
+        
+        if response.status_code == 200:
+            return response.json()
+        else:
+            raise Exception(f"GraphQL query failed with status code {response.status_code}: {response.text}")