argparser를 사용하여 플래그를 추가하는 방법은 무엇입니까?

Question

나는

def add(x,y): 
    print x+y 

def square(a): 
    print a**2 

은 내가

#! /usr/bin/python 


import argparse 
# Create Parser and Subparser 
parser = argparse.ArgumentParser(description="Example ArgumentParser") 
subparser = parser.add_subparsers(help="commands") 

# Make Subparsers 
hello_parser = subparser.add_parser('hello', help='hello func') 
hello_parser.add_argument("arg",help="string to print") 
hello_parser.set_defaults(func='hello') 

add_parser = subparser.add_parser('add', help="add func") 
add_parser.add_argument("x",type=float,help='first number') 
add_parser.add_argument("y",type=float,help='second number') 
add_parser.set_defaults(func='add') 

square_parser = subparser.add_parser('square', help="square func") 
square_parser.add_argument("a",type=float,help='number to square') 
square_parser.set_defaults(func='square') 

args = parser.parse_args() 

def hello(arg): 
    print arg 

def add(x,y): 
    print x + y 

def square(a): 
    print a**2 

if args.func == 'hello': 
    hello(args.arg) 
elif args.func == 'add': 
    add(args.x,args.y) 
elif args.func == 'square': 
    square(args.a) 

난에 플래그를 추가 할 수이 코드를 시도

./hello.py -a add 2 3 
./hello.py -s sqare 3 

해줘처럼 argparse를 사용하여 이러한 기능에 대한 플래그를 만들 수 같은 기능을 가지고 있습니다 같은 코드 ??

Answer 1

https://docs.python.org/3/library/argparse.html#sub-commands의 마지막 네 같은 일례를 갖는다

있지만 (이름 대신 실제 함수를 사용하여) add_parser.set_defaults(func=add) 같은 식. 이를 통해 스택을 args.func(args)으로 바꿀 수 있습니다.

하지만 오히려 subparsers 이상 입력의 플래그/선택적 항목 스타일을 사용하려면, 나는 다음과 같은 좋을 것 :

import argparse 

def add(x,y): 
    print x+y 

def square(a): 
    print a**2 

parser = argparse.ArgumentParser() 
parser.add_argument('--hello') 
parser.add_argument('-a', '--add', nargs=2, type=int) 
parser.add_argument('-s', '--square', type=int) 

args = parser.parse_args() 
print(args)   # good debugging tool 

if args.add is not None: 
    add(*args.add) # * expands the list into the 2 arguments 
if args.square is not None: 
    square(args.square) 
if args.hello is not None: 
    print ("hello "+args.hello) 

이

1014:~/mypy$ python stack43776406.py -a 10 11 
Namespace(add=[10, 11], hello=None, square=None) 
21 
1014:~/mypy$ python stack43776406.py --hello world -a 1 2 -s 1000 
Namespace(add=[1, 2], hello='world', square=1000) 
3 
1000000 
hello world 

생산을 만약 '-a'플래그 이미 add을 의미하므로 추가 문자열을 포함 할 필요가 없습니다. 또한 서브 파서에 필요한 별도의 'x', 'y'인수 대신 nargs=2을 사용했습니다.

is None 또는 is not None은 신고 된 값이 제공되었는지 여부를 테스트하는 편리한 방법입니다. 사용자는 None을 입력 할 수 없으므로이 기본 기본값을 테스트하는 것은 절대적으로 어렵습니다.

1014:~/mypy$ python stack43776406.py --help 
usage: stack43776406.py [-h] [--hello HELLO] [-a ADD ADD] [-s SQUARE] 

optional arguments: 
    -h, --help   show this help message and exit 
    --hello HELLO 
    -a ADD ADD, --add ADD ADD 
    -s SQUARE, --square SQUARE 

Answer 2

import argparse 

def add(x,y): 
    print x+y 

def square(a): 
    print a**2 

if __name__ == '__main__': 
    parser = argparse.ArgumentParser(description='This is a parser.') 
    parser.add_argument('command', help='The command to execute (add or square).') 
    parser.add_argument('integers', metavar='N', type=int, nargs='+', help='The arguments to the command.') 
    args = parser.parse_args() 

    if args.command == 'add': 
     add(*args.integers) 
    elif args.command == 'square': 
     square(*args.integers) 
    else: 
     print 'Command not supported.' 

출력 :

$ python hello.py add 2 3 
5 
$ python hello.py square 2 
4