Python argparse, Positionsargumente und subparsers

Question

Ich habe folgende Schnipsel, wo ich verwende argparse mit mehreren subparsers

parser = argparse.ArgumentParser(description = "Setup the DB", 
     add_help=False) 
parser.add_argument('action', type = str, 
     choices=['drop','populate','print','create','print-metadata'], 
     help = "Specify an action", default = None) 
subparsers = parser.add_subparsers() 
drop_parser = subparsers.add_parser('drop',parents=[parser]) 
drop_parser.add_argument('-dataset-name', 
     required=True, 
     type = str, help = "Dataset Name", 
     default = None) 
..... 
args = parser.parse_args() 
..... 

Wenn ich dies laufen ich folgendes erhalten:

python .\populatedb.py drop -dataset-name foo 
populatedb.py: error: invalid choice: 'foo' (choose from 'drop', 'populate', 'print', 'create', 'print-metadata') 

Ich bin frage mich, wo der Code oben falsch läuft. Beachten Sie, dass das Argument "Aktion" ein Positionsargument ist.

Vielen Dank im Voraus. Dies ist meine erste Verwendung von Subparsern. Ich mache wahrscheinlich einen offensichtlichen Fehler.

Vielen Dank im Voraus.

Grüße,

Ranga

Answer 1

Die allgemein akzeptierte, was hier zu tun ist, nicht für die subparser ein Argument zu haben - es ist es eigenes Argument von argparse vorgesehen ist: Jetzt

parser = argparse.ArgumentParser(description = "Setup the DB", 
     add_help=False) 
subparsers = parser.add_subparsers() 
drop_parser = subparsers.add_parser('drop',parents=[parser]) 
drop_parser.add_argument('-dataset-name', 
     required=True, 
     type = str, help = "Dataset Name", 
     default = None) 
..... 
args = parser.parse_args() 

Sie wissen nicht unbedingt, welcher Parser ausgewählt wurde, aber keine Sorge, dafür gibt es auch einen eingebauten Mechanismus. Ein gemeinsamer Anwendungsfall ist mit einer Funktion, die für jeden subparser aufgerufen werden soll ...

parser = argparse.ArgumentParser(description = "Setup the DB", 
     add_help=False) 
subparsers = parser.add_subparsers() 

def drop_parser_handler(args): 
    ... 

drop_parser = subparsers.add_parser('drop',parents=[parser]) 
drop_parser.add_argument('-dataset-name', 
     required=True, 
     type = str, help = "Dataset Name", 
     default = None) 
drop_parser.set_defaults(func=drop_parser_handler) 
..... 

args = parser.parse_args() 
args.func(args) 

Natürlich brauchen Sie nicht alles, was diesen Komplex zu tun - Sie könnten nur einen konstanten Wert gesetzt:

drop_parser.set_defaults(subparser_name='drop')