python argh / argparse:我如何传递一个列表作为命令行参数?

我想使用argh库将参数列表传递给python脚本。 可以采取如下input的东西:

./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 

我的内部代码如下所示:

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

以下是它的行为:

 $ 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 

这个问题看起来相当简单:argh只接受第一个参数,并将其视为一个string。 我如何使它“期待”一个整数列表,而不是?

我看到这是如何在optparse中完成的 ,但是(未被弃用的)argparse呢? 或者用argh的更好的装饰语法? 这似乎更加pythonic。

argparse ,你只需要使用type=int

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

示例输出:

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

编辑:我不熟悉argh ,但它似乎只是一个包装argparse ,这对我工作:

 import argh @argh.arg('-a', '--arg', nargs='+', type=int) def main(args): print args parser = argh.ArghParser() parser.add_commands([main]) parser.dispatch() 

示例输出:

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

我为了访问每个参数值,下面的代码可能会有所帮助。

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