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 更好的修饰语法？这些看起来更像蟒蛇。

59964

小开

最佳答案

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