python argh / argparse：如何将列表作为命令传递
我试图将参数列表传递给使用argh库的python脚本。 可以采取如下输入的东西：
./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只接受第一个参数，并将其视为一个字符串。 我如何让它“预期”一个整数列表呢？
我看到这是如何在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])
上一篇: python argh/argparse: How can I pass a list as a command