how_to_stake.py

import argparse
import asyncio
import os
import pathlib

import pycspr
from pycspr import NodeRpcClient as NodeClient
from pycspr import NodeRpcConnectionInfo as NodeConnectionInfo
from pycspr.types.crypto import KeyAlgorithm
from pycspr.types.crypto import PrivateKey
from pycspr.types.node import Deploy


# Path to CCTL assets.
_PATH_TO_CCTL_ASSETS = pathlib.Path(os.getenv("CCTL")) / "assets"

# CLI argument parser.
_ARGS = argparse.ArgumentParser("Demo illustrating how to stake CSPR tokens as a validator.")

# CLI argument: path to validator secret key - defaults to CCTL user 1.
_ARGS.add_argument(
    "--validator-secret-key-path",
    default=_PATH_TO_CCTL_ASSETS / "nodes" / "node-1" / "keys" / "secret_key.pem",
    dest="path_to_validator_secret_key",
    help="Path to validator's secret_key.pem file.",
    type=str,
    )

# CLI argument: type of validator secret key - defaults to ED25519.
_ARGS.add_argument(
    "--validator-secret-key-type",
    default=KeyAlgorithm.ED25519.name,
    dest="type_of_validator_secret_key",
    help="Type of validator's secret key.",
    type=str,
    )

# CLI argument: path to session code wasm binary - defaults to CCTL bin/wasm/add_bid.wasm.
_ARGS.add_argument(
    "--path-to-wasm",
    default=_PATH_TO_CCTL_ASSETS / "bin" / "add_bid.wasm",
    dest="path_to_wasm",
    help="Path to add_bid.wasm file.",
    type=str,
    )

# CLI argument: amount to stake, i.e. bond, into the network.
_ARGS.add_argument(
    "--amount",
    default=int(2.5e9),
    dest="amount",
    help="Amount to bond.",
    type=int,
    )

# CLI argument: amount to charge delegators for service provision.
_ARGS.add_argument(
    "--delegation-rate",
    default=2,
    dest="delegation_rate",
    help="Amount to charge delegators for servie provision.",
    type=int,
    )

# CLI argument: name of target chain - defaults to CCTL chain.
_ARGS.add_argument(
    "--chain",
    default="cspr-dev-cctl",
    dest="chain_name",
    help="Name of target chain.",
    type=str,
    )

# CLI argument: host address of target node - defaults to CCTL node 1.
_ARGS.add_argument(
    "--node-host",
    default="localhost",
    dest="node_host",
    help="Host address of target node.",
    type=str,
    )

# CLI argument: Node API JSON-RPC port - defaults to 11101 @ CCTL node 1.
_ARGS.add_argument(
    "--node-port-rpc",
    default=11101,
    dest="node_port_rpc",
    help="Node API JSON-RPC port.  Typically 7777 on most nodes.",
    type=int,
    )


async def _main(args: argparse.Namespace):
    """Main entry point.

    :param args: Parsed command line arguments.

    """
    print("-" * 74)
    print("PYCSPR :: How To Stake")
    print("")
    print("Illustrates usage of pycspr.create_validator_auction_bid function.")
    print("-" * 74)

    # Set node client.
    client: NodeClient = _get_client(args)

    # Set validator key.
    validator: PrivateKey = pycspr.parse_private_key(
        args.path_to_validator_secret_key,
        args.type_of_validator_secret_key,
        )

    # Set deploy.
    deploy: Deploy = _get_deploy(args, validator)

    # Approve deploy.
    deploy.approve(validator)

    # Dispatch deploy to a node.
    await client.send_deploy(deploy)

    print(f"Deploy dispatched to node [{args.node_host}]: {deploy.hash.hex()}")


def _get_client(args: argparse.Namespace) -> NodeClient:
    """Returns a pycspr client instance.

    """
    return NodeClient(NodeConnectionInfo(args.node_host, args.node_port_rpc))


def _get_deploy(args: argparse.Namespace, validator: PrivateKey) -> Deploy:
    """Returns delegation deploy to be dispatched to a node.

    """
    # Set standard deploy parameters.
    deploy_params = pycspr.create_deploy_parameters(
        account=validator,
        chain_name=args.chain_name
        )

    # Set deploy.
    deploy = pycspr.create_validator_auction_bid(
        params=deploy_params,
        amount=args.amount,
        delegation_rate=args.delegation_rate,
        public_key=validator.to_public_key(),
        path_to_wasm=args.path_to_wasm
        )

    return deploy


# Entry point.
if __name__ == "__main__":
    asyncio.run(_main(_ARGS.parse_args()))