core-extra/daemon/examples/python/params.py

import argparse

DEFAULT_NODES = 2
DEFAULT_TIME = 10
DEFAULT_STEP = 1


def parse(name):
    parser = argparse.ArgumentParser(description=f"Run {name} example")
    parser.add_argument(
        "-n",
        "--nodes",
        type=int,
        default=DEFAULT_NODES,
        help="number of nodes to create in this example",
    )
    parser.add_argument(
        "-c",
        "--count",
        type=int,
        default=DEFAULT_TIME,
        help="number of time to ping node",
    )

    args = parser.parse_args()

    if args.nodes < 2:
        parser.error(f"invalid min number of nodes: {args.nodes}")
    if args.count < 1:
        parser.error(f"invalid ping count({args.count}), count must be greater than 0")

    return args