[英]python argh/argparse: How can I pass a list as a command-line argument?
I'm trying to pass a list of arguments to a python script using the argh library. 我正在尝试使用argh库将参数列表传递给python脚本。 Something that can take inputs like these:
可以采取以下输入的东西:
./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
My internal code looks like this: 我的内部代码如下所示:
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()
And here's how it behaves: 这是它的行为方式:
$ 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
The problem seems pretty straightforward: argh is only accepting the first argument, and treating it as a string. 问题似乎很简单:argh只接受第一个参数,并将其视为一个字符串。 How do I make it "expect" a list of integers instead?
我如何让它“期望”整数列表呢?
I see how this is done in optparse , but what about the (not-deprecated) argparse? 我在optparse中看到了这是怎么做的 ,但是(未弃用的)argparse呢? Or using argh's much nicer decorated syntax?
或者使用argh更好的装饰语法? These seem much more pythonic.
这些似乎更加pythonic。
With argparse
, you just use type=int
使用
argparse
,您只需使用type=int
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('-a', '--arg', nargs='+', type=int)
print parser.parse_args()
Example output: 示例输出:
$ python test.py -a 1 2 3
Namespace(arg=[1, 2, 3])
Edit: I'm not familiar with argh
, but it seems to be just a wrapper around argparse
and this worked for me: 编辑:我不熟悉
argh
,但它似乎只是argparse
一个包装,这对我argh
:
import argh
@argh.arg('-a', '--arg', nargs='+', type=int)
def main(args):
print args
parser = argh.ArghParser()
parser.add_commands([main])
parser.dispatch()
Example output: 示例输出:
$ python test.py main -a 1 2 3
Namespace(arg=[1, 2, 3], function=<function main at 0x.......>)
I order to have access to each parameter value, the following code may be helpful. 我命令有权访问每个参数值,以下代码可能会有所帮助。
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])
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.