lifecycle.py

import os
import logging
import json

from kwil import Kwil


def run():
    # assume that account has enough fund
    # here we use test_db.kf as our dataset schema, we'll use test_db.json(compiled schema)

    # create client
    client = Kwil(Kwil.GRPCProvider(os.getenv("KWIL_PROVIDER", "grpc.kwil.com:80")),
                  Kwil.load_wallet(os.getenv("KWIL_CLI_PRIVATE_KEY", "EMPTY_KEY")))

    db_name = "testdb"  # The name of the database, from the schema file
    # create dataset
    with open("./testdb.json", "r") as f:
        schema_json = f.read()
        schema = json.loads(schema_json)
        db_name = schema["name"]
        try:
            client.deploy_database(schema_json.encode("utf-8"))
        except Exception as e:
            logging.exception(e)

    # list dataset
    dbs = client.list_databases()
    print("datasets after create: ", dbs)

    # execute an action
    db_id = Kwil.generate_dbi(client.wallet.address, db_name)

    action = "create_user"
    tx_receipt = client.execute_action(
        db_id,
        action,
        # `create_user` is defined in test_db.kf
        [{"$id": 1, "$username": "aha", "$age": 18}],
    )
    print("create user result: ", tx_receipt)

    # call a view action
    action = "view_user_info"
    tx_receipt = client.call_action(db_id, action, {})
    print("user info:", tx_receipt)

    # drop dataset
    tx_receipt = client.drop_database(db_name)
    print("drop dataset result: ", tx_receipt)

    # list dataset
    dbs = client.list_databases()
    print("datasets after delete: ", dbs)


if __name__ == "__main__":
    run()