Python - Passa argomenti a diversi metodi da Argparse
Domanda
Sto scrivendo uno script Python relativamente semplice che supporta un paio di comandi diversi. I diversi comandi supportano diverse opzioni e voglio essere in grado di passare le opzioni analizzate da ArgParse al metodo corretto per il comando specificato.
La stringa di utilizzo sembra così:
usage: script.py [-h]
{a, b, c}
...
script.py: error: too few arguments
Posso facilmente chiamare il metodo appropriato:
def a():
...
def b():
...
def c():
...
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.set_defaults(method = a)
...
arguments = parser.parse_args()
arguments.method()
Tuttavia, devo trasmettere argomenti a questi metodi e tutti accettano diverse serie di argomenti.
Attualmente, ho appena superato il Namespace
Oggetto restituito da Argparse, come così:
def a(arguments):
arg1 = getattr(arguments, 'arg1', None)
...
Questo sembra un po 'imbarazzante e rende i metodi un po' più difficili da riutilizzare perché devo passare argomenti come un dict
o spazio dei nomi piuttosto che come soliti parametri.
Vorrei in qualche modo di definire i metodi con i parametri (come faresti una funzione normale) ed essere ancora in grado di chiamarli dinamicamente mentre passano parametri appropriati. Così:
def a(arg1, arg2):
...
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.set_defaults(method = a)
...
arguments = parser.parse_args()
arguments.method() # <<<< Arguments passed here somehow
Qualche idea?
Soluzione
Ho trovato una soluzione piuttosto bella:
import argparse
def a(arg1, arg2, **kwargs):
print arg1
print arg2
if __name__ == "__main__":
parser = argparse.ArgumentParser()
parser.set_defaults(method = a)
parser.add_argument('arg1', type = str)
parser.add_argument('arg2', type = str)
arguments = parser.parse_args()
arguments.method(**vars(arguments))
Naturalmente c'è un problema minore se gli argomenti del metodo si scontrano con i nomi degli argomenti che Argparse usa, anche se penso che ciò sia preferibile a passare l'oggetto dello spazio dei nomi e usare getattr
.
Altri suggerimenti
Probabilmente stai cercando di ottenere la funzionalità fornita da sub-comandi:http://docs.python.org/dev/library/argparse.html#sub-commands
Non sono sicuro di quanto sia pratico, ma usando ispezionare Puoi tralasciare il parametro estraneo ** Kwargs sulle tue funzioni, come: così:
import argparse
import inspect
def sleep(seconds=0):
print "sleeping", seconds, "seconds"
def foo(a, b=2, **kwargs):
print "a=",a
print "b=",b
print "kwargs=",kwargs
parser = argparse.ArgumentParser()
subparsers = parser.add_subparsers(title="subcommand")
parser_sleep = subparsers.add_parser('sleep')
parser_sleep.add_argument("seconds", type=int, default=0)
parser_sleep.set_defaults(func=sleep)
parser_foo = subparsers.add_parser('foo')
parser_foo.add_argument("-a", type=int, default=101)
parser_foo.add_argument("-b", type=int, default=201)
parser_foo.add_argument("--wacky", default=argparse.SUPPRESS)
parser_foo.set_defaults(func=foo)
args = parser.parse_args()
arg_spec = inspect.getargspec(args.func)
if arg_spec.keywords:
## convert args to a dictionary
args_for_func = vars(args)
else:
## get a subset of the dictionary containing just the arguments of func
args_for_func = {k:getattr(args, k) for k in arg_spec.args}
args.func(**args_for_func)
Esempi:
$ python test.py sleep 23
sleeping 23 seconds
$ python test.py foo -a 333 -b 444
a= 333
b= 444
kwargs= {'func': <function foo at 0x10993dd70>}
$ python test.py foo -a 333 -b 444 --wacky "this is wacky"
a= 333
b= 444
kwargs= {'func': <function foo at 0x10a321d70>, 'wacky': 'this is wacky'}
Divertiti!