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


70

我正在尝试使用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。

Answers:


94

使用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.......>)

1
太棒了!我调整了代码以包括以下行:@ argh.arg('-argB',default = [],help ='一个列表类型的arg--除非不是!',nargs ='+', type = int),而argh喜欢它。谢谢您的帮助!
安倍

是的,这是正确的方法。Argh的最新版本还支持“自然的” Python表示法:(func(argA, *argB)尽管*args映射到nargs='*'而不是nargs='+')。
Andy Mikhaylenko 2013年

@jcollado:我看到当我使用nargs =时,我无法执行-a = 1 2 3。我的意思是在参数metavar后使用'='。任何让我对每个参数使用=的解决方案。例如,在我的示例中,我想做python test.py main -a = 1 2 3
script_kiddie 2013年

@Andy Mikhaylenko上面的问题。
script_kiddie 2013年

@AndyMikhaylenko我不知道有任何简单的方法可以做到这一点。一种替代方法是:parser.add_argument('-a', '--arg', type=lambda s: [int(n) for n in s.split()])和使用引号调用脚本:python test.py --arg="1 2 3"
jcollado 2013年

1

我要访问每个参数值,以下代码可能会有所帮助。

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])
By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.