[python] argparse로 선택된 부속 명령 가져 오기

Program / 글쓴이

python argparse와 함께 하위 명령을 사용하면 선택한 인수를 얻을 수 있습니다.

parser = argparse.ArgumentParser()
parser.add_argument('-g', '--global')
subparsers = parser.add_subparsers()   
foo_parser = subparsers.add_parser('foo')
foo_parser.add_argument('-c', '--count')
bar_parser = subparsers.add_parser('bar')
args = parser.parse_args(['-g, 'xyz', 'foo', '--count', '42'])
# args => Namespace(global='xyz', count='42')

을 args포함하지 않습니다 'foo'. sys.argv[1]가능한 전역 인수 때문에 글쓰기 가 제대로 작동하지 않습니다. 하위 명령 자체를 어떻게 얻을 수 있습니까?

답변

argparse 부속 명령에 대한 Python 문서의 맨 아래에는 이를 수행하는 방법이 설명되어 있습니다.

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('-g', '--global')
>>> subparsers = parser.add_subparsers(dest="subparser_name") # this line changed
>>> foo_parser = subparsers.add_parser('foo')
>>> foo_parser.add_argument('-c', '--count')
>>> bar_parser = subparsers.add_parser('bar')
>>> args = parser.parse_args(['-g', 'xyz', 'foo', '--count', '42'])
>>> args
Namespace(count='42', global='xyz', subparser_name='foo')

set_defaults()내가 찾은 예제 바로 위에 참조 된 방법을 사용할 수도 있습니다 .

답변

ArgumentParser.add_subparsers이 dest형식 인수가 설명 :

dest-부속 명령 이름이 저장 될 속성의 이름; 기본적으로 None값이 저장되지 않습니다

하위 파서를 사용하는 간단한 작업 함수 레이아웃의 아래 예에서 선택한 하위 파서는입니다 parser.parse_args().subparser.

import argparse


def task_a(alpha):
    print('task a', alpha)


def task_b(beta, gamma):
    print('task b', beta, gamma)


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    subparsers = parser.add_subparsers(dest='subparser')

    parser_a = subparsers.add_parser('task_a')
    parser_a.add_argument(
        '-a', '--alpha', dest='alpha', help='Alpha description')

    parser_b = subparsers.add_parser('task_b')
    parser_b.add_argument(
        '-b', '--beta', dest='beta', help='Beta description')
    parser_b.add_argument(
        '-g', '--gamma', dest='gamma', default=42, help='Gamma description')

    kwargs = vars(parser.parse_args())
    globals()[kwargs.pop('subparser')](**kwargs)

답변