Ottenere sottocomando selezionato con argparse
-
14-10-2019 - |
Domanda
Quando uso subcommands con pitone argparse, posso ottenere gli argomenti selezionati.
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')
Quindi args
non contiene 'foo'
. Semplicemente scrivendo sys.argv[1]
non funziona a causa dei possibili args globali. Come posso ottenere il sottocomando stessa?
Soluzione
Il fondo delle documentazione Python in argparse sub-comandi spiega come fare questo:
>>> 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')
È possibile anche utilizzare il metodo set_defaults()
fa riferimento proprio sopra l'esempio che ho trovato.
Altri suggerimenti
Ecco un esempio di layout di funzione semplice compito utilizzando subparsers.
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)
Autorizzato sotto: CC-BY-SA insieme a attribuzione
Non affiliato a StackOverflow