schema_to_class

Hello, my job is to make easier the writing of each class for the contracts. I read the schemas and translate it into python code.

Simple to use:

python3 schema_to_class/schema_to_class.py contracts/airdrop

Result looks like this:

class Airdrop():

        def execute_update_config(owner: str = None):
                return {'update_config': {'owner': owner}}

        def execute_register_merkle_root(merkle_root: str):
                return {'register_merkle_root': {'merkle_root': merkle_root}}

        def execute_claim(amount, proof: list, stage: int):
                return {'claim': {'amount': amount, 'proof': proof, 'stage': stage}}

        def execute_withdraw(amount, recipient: str):
                return {'withdraw': {'amount': amount, 'recipient': recipient}}

        def instantiate(anchor_token: str, gov_contract: str, owner: str):
                return {'anchor_token': anchor_token, 'gov_contract': gov_contract, 'owner': owner}

        def query_config():
                return {'config': {}}

        def query_merkle_root(stage: int):
                return {'merkle_root': {'stage': stage}}

        def query_latest_stage():
                return {'latest_stage': {}}

        def query_is_claimed(address: str, stage: int):
                return {'is_claimed': {'address': address, 'stage': stage}}

Import and play:

from sdk_wrapper import Contract, terra, store_contract
from airdrop import Airdrop

# Contract holder (able to instantiate, execute, query)
aidrop_contract = Contract()

# Store the contract
code_id = store_contract(terra, deployer, 'anchor_airdrop')

# Instantiate the contract
instantiate_msg = Airdrop.instantiate(anchor_token.address, gov_contract.address, alice.key.acc_address)
aidrop_contract.instantiate(alice, code_id, instantiate_msg)
# Now has an address: airdrop_contract.address = "terra1....."

# Execute update config
update_config_msg = Airdrop.execute_update_config(bob.key.acc_address)
aidrop_contract.execute(alice, update_config_msg)

# Query config
res = airdrop_contract.query(Airdrop.query_config())
# res = "terra1....." (bob)