python argh/argparse:如何将列表作为命令行参数传递?
我试图使用argh库将参数列表传递给python脚本。可以接受如下输入的东西:python argh/argparse:如何将列表作为命令行参数传递?,python,command-line-arguments,argparse,argh,Python,Command Line Arguments,Argparse,Argh,我试图使用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 fi
./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只接受第一个参数,并将其视为字符串。我如何让它“期望”一个整数列表呢
我明白了,但是(不是不推荐的)argparse呢?还是使用argh更好的修饰语法?这些看起来更像python。对于
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])
$ python test.py main -a 1 2 3
Namespace(arg=[1, 2, 3], function=<function main at 0x.......>)
编辑:我不熟悉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 -a 1 2 3
Namespace(arg=[1, 2, 3])
$ python test.py main -a 1 2 3
Namespace(arg=[1, 2, 3], function=<function main at 0x.......>)
$python test.py main-a 1 2 3
命名空间(arg=[1,2,3],函数=)
为了访问每个参数值,以下代码可能会有所帮助
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])
令人惊叹的!我调整了代码以包含以下行:@argh.arg('--argB',default=[],help='A list type arg--except's not!',nargs='+',type=int),argh喜欢它。谢谢你的帮助!是的,这是正确的方法。Argh的最新版本还支持“自然”的、更具Python风格的表示法:
func(argA,*argB)
(虽然*args
映射到nargs='*'
,而不是nargs='+'
)。@jcollado:我发现当我使用nargs=,我不能做-a=1 2=3。我的意思是在参数metavar之后使用“=”。允许我为每个参数使用=的任何解决方案。对于示例中的ex,我想做的是python test.py main-a=123@AndyMikhaylenko回答上述问题。@AndyMikhaylenko我不知道有什么简单的方法可以做到这一点。一种替代方法是:parser.add_参数('-a','-arg',type=lambda s:[int(n)表示s.split()中的n)
并使用引号调用脚本:python test.py--arg=“1 2 3”