web-dev-qa-db-fra.com

python argh / argparse: comment passer une liste en argument de ligne de commande?

J'essaie de passer une liste d'arguments à un script python en utilisant la bibliothèque argh. Quelque chose qui peut prendre des entrées comme celles-ci:

./my_script.py my-func --argA blah --argB 1 2 3 4
./my_script.py my-func --argA blah --argB 1
./my_script.py my-func --argA blah --argB

Mon code interne ressemble à ceci:

import argh

@argh.arg('--argA', default="bleh", help='My first arg')
@argh.arg('--argB', default=[], help='A list-type arg--except it\'s not!')
def my_func(args):
    "A function that does something"

    print args.argA
    print args.argB

    for b in args.argB:
        print int(b)*int(b)  #Print the square of each number in the list
    print sum([int(b) for b in args.argB])  #Print the sum of the list

p = argh.ArghParser()
p.add_commands([my_func])
p.dispatch()

Et voici comment cela se comporte:

$ python temp.py my-func --argA blooh --argB 1
blooh
['1']
1
1

$ python temp.py my-func --argA blooh --argB 10
blooh
['1', '0']
1
0
1

$ python temp.py my-func --argA blooh --argB 1 2 3
usage: temp.py [-h] {my-func} ...
temp.py: error: unrecognized arguments: 2 3

Le problème semble assez simple: argh n'accepte que le premier argument et le traite comme une chaîne. Comment puis-je lui faire "attendre" une liste d'entiers à la place?

Je vois comment cela se fait dans optparse , mais qu'en est-il de l'argparse (non obsolète)? Ou en utilisant la syntaxe décorée beaucoup plus agréable d'Argh? Ceux-ci semblent beaucoup plus Pythonic.

pythoncommand-line-argumentsargparseargh
63
Abe

Avec argparse, vous utilisez simplement type=int

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-a', '--arg', nargs='+', type=int)
print parser.parse_args()

Exemple de sortie:

$ python test.py -a 1 2 3
Namespace(arg=[1, 2, 3])

Edit: je ne connais pas argh, mais il semble que ce soit juste un wrapper autour de argparse et cela a fonctionné pour moi:

import argh

@argh.arg('-a', '--arg', nargs='+', type=int)
def main(args):
    print args

parser = argh.ArghParser()
parser.add_commands([main])
parser.dispatch()

Exemple de sortie:

$ python test.py main -a 1 2 3
Namespace(arg=[1, 2, 3], function=<function main at 0x.......>)
84
jcollado

Je commande d'avoir accès à chaque valeur de paramètre, le code suivant peut être utile.

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('-a', '--arg', nargs='+', type=int)
args = vars(parser.parse_args())

print "first parameter:" + str(args["arg"][0])
print "second parameter:" + str(args["arg"][1])
print "third parameter:" + str(args["arg"][2])
1
Luis Jose