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

import argparse

DEFAULT_NODES = 2
DEFAULT_TIME = 10
DEFAULT_STEP = 1


def parse_options(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(
        "-t",
        "--time",
        type=int,
        default=DEFAULT_TIME,
        help="example iperf run time in seconds",
    )

    options = parser.parse_args()

    if options.nodes < 2:
        parser.error(f"invalid min number of nodes: {options.nodes}")
    if options.time < 1:
        parser.error(f"invalid test time: {options.time}")

    return options