/Python/Use argparse to call function and use argument in function

Use argparse to call function and use argument in function

# Parse the subcommand argument first
parser = ArgumentParser(add_help=False)
parser.add_argument("function", 
                    nargs="?",
                    choices=['function1', 'function2', 'function2'],
                    )
parser.add_argument('--help', action='store_true')
args, sub_args = parser.parse_known_args(['--help'])

# Manually handle help
if args.help:
    # If no subcommand was specified, give general help
    if args.function is None: 
        print parser.format_help()
        sys.exit(1)
    # Otherwise pass the help option on to the subcommand
    sub_args.append('--help')

# Manually handle the default for "function"
function = "function1" if args.function is None else args.function

# Parse the remaining args as per the selected subcommand
parser = ArgumentParser(prog="%s %s" % (os.path.basename(sys.argv[0]), function))
if function == "function1":
    parser.add_argument('-a','--a')
    parser.add_argument('-b','--b')
    parser.add_argument('-c','--c')
    args = parser.parse_args(sub_args)
    function1(args.a, args.b, args.c)
elif function == "function2":
    ...
elif function == "function3":
    ...

Source: stackoverflow.com

Similar codes

0 vote156 views1 answer
Slicing time series
Python
0 vote152 views1 answer
Scipy check normal distribution
Python
0 vote200 views3 answers
How to create a sub project in django
Python
1 vote198 views1 answer
Pip install mysqldb
Python
1 vote149 views1 answer
How to take input complex number in python
Python
6 votes202 views2 answers
Flask get ip address of request
Python Flask
5 votes217 views3 answers
Sys module in python
Python
115 votes167 views6 answers
Python user input
Python
77 votes150 views6 answers
Random from list python
Python
86 votes158 views4 answers
How to delete a file in python
Python