Get ausgewählten Unterbefehl mit Argarse

Question

Wenn ich Unterbefehle mit Python Argparse verwenden, kann ich die ausgewählten Argumente abrufen.

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

So args nicht 'foo' enthalten. Einfaches Schreiben sys.argv[1] funktioniert wegen der möglichen globalen Argumente nicht. Wie kann ich den Unterbefehl selbst erhalten?

Answer 1

Die ganz unten auf der Python docs on argparse sub-commands erklärt, wie dies tun:

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

Sie auch die set_defaults() Methode direkt über dem Beispiel verwiesen verwenden kann ich gefunden.

Answer 2

Hier ist ein Beispiel für einfache Aufgabe Funktionslayout.

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)