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

发布于 2021-01-29 15:09:36

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

关注者
0
被浏览
96
1 个回答
  • 面试哥
    面试哥 2021-01-29
    为面试而生,有面试问题,就找面试哥。

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


知识点
面圈网VIP题库

面圈网VIP题库全新上线,海量真题题库资源。 90大类考试,超10万份考试真题开放下载啦

去下载看看