Uzyskaj wybraną podkomendę z argparse

Question

Kiedy używam podkomend z python argparse, mogę pobrać wybrane argumenty.

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') 

Więc args nie zawiera 'foo'. Po prostu pisanie sys.argv[1] nie działa z powodu możliwych globalnych argumentów. Jak mogę uzyskać samą podkomendę?

Answer 1

Samo dno Python docs on argparse sub-commands wyjaśnia, jak to zrobić:

>>> 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') 

Można również użyć metody set_defaults() odwołuje się tuż nad przykład znalazłem.

Answer 2

Oto przykład prostego układu funkcji zadania.

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)