argparse_example.py

import argparse

from rich_argparse import ArgumentDefaultsRichHelpFormatter


def get_parser():
    parser = argparse.ArgumentParser(
        description="Example of argparse usage, with default values printed in help.",
        formatter_class=ArgumentDefaultsRichHelpFormatter,
    )
    parser.add_argument(
        "--example_string",
        default="example",
        help="path to config file",
    )
    parser.add_argument(
        "--example_bool",
        action="store_true",
        help="Save bounding box visualisation.",
    )
    parser.add_argument(
        "--example_list",
        nargs="+",
        default=[1, 2, 3],
        help="List of integers",
    )
    parser.add_argument(
        "--example_choices",
        choices=["a", "b", "c"],
        default="a",
        help="Choices",
    )
    return parser


def main():
    parser = get_parser()
    args = parser.parse_args()

    # if (
    #     bool(args.video_input) + bool(args.images_input_dir) + bool(args.videos_input_dir)
    #     != 1
    # ):
    #     parser.error(
    #         "--video_input, --video_input_dir and --images_input_dir can't come together and only one must be specified."
    #     )

save_args.py

def save_args(args: argparse.Namespace):
    # Save args to json
    import json

    with open("args.json", "w") as f:
        json.dump(args.__dict__, f, ensure_ascii=False, indent=4)