首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >python argh/argparse:如何将列表作为命令行参数传递?

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

Stack Overflow用户
提问于 2012-02-22 23:45:47
回答 2查看 58K关注 0票数 73

我正在尝试使用argh库将一组参数传递给python脚本。可以接受像这样的输入的东西:

代码语言:javascript
复制
./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 

我的内部代码如下所示:

代码语言:javascript
复制
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()

下面是它的行为方式:

代码语言:javascript
复制
$ 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只接受第一个参数,并将其视为字符串。我如何让它“期望”一个整数列表呢?

我看到了how this is done in optparse,但是(不推荐使用的) argparse呢?或者使用argh更好的修饰语法?这些看起来更像蟒蛇。

EN

回答 2

Stack Overflow用户

回答已采纳

发布于 2012-02-22 23:55:21

argparse中,您只需使用type=int

代码语言:javascript
复制
import argparse

parser = argparse.ArgumentParser()
parser.add_argument('-a', '--arg', nargs='+', type=int)
print parser.parse_args()

输出示例:

代码语言:javascript
复制
$ python test.py -a 1 2 3
Namespace(arg=[1, 2, 3])

编辑:我不熟悉argh,但它似乎只是argparse的一个包装器,这对我很有效:

代码语言:javascript
复制
import argh

@argh.arg('-a', '--arg', nargs='+', type=int)
def main(args):
    print args

parser = argh.ArghParser()
parser.add_commands([main])
parser.dispatch()

输出示例:

代码语言:javascript
复制
$ python test.py main -a 1 2 3
Namespace(arg=[1, 2, 3], function=<function main at 0x.......>)
票数 98
EN

Stack Overflow用户

发布于 2016-09-14 17:54:44

为了能够访问每个参数值,下面的代码可能会有所帮助。

代码语言:javascript
复制
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])
票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/9398065

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档