从 argparse 解包参数

Question

我一直在编写一些命令行 python 程序并使用 argparse 来完成。我一直在按如下方式构建我的代码。

def main(arg1, arg2):
    # magic
    pass

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('arg1')
    parser.add_argument('arg2')

    args = parser.parse_args()

    main(args.arg1, args.arg2)

要叫arg1和arg23次真的超级烦人。我知道必须做两次。

有什么方法可以将 parse_args 函数返回的命名空间视为元组吗？或者更好的是作为可选参数的元组和字典并进行解包？

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('arg1')
    parser.add_argument('arg2')
    parser.add_argument('--opt-arg', default='default_value')

    args, kwargs = parser.magic_method_call_that_would_make_my_life_amazing()

    # I get goosebumps just thinking about this
    main(*args, **kwargs)

Answer 1

做错了什么

if __name__ == '__main__':
    # do argparse stuff as above
    main(args)

也就是说，您为什么对提供 main() 位置参数如此着迷？

老实说，我通常在模块开头解析参数 a) [对于小脚本等]，这为我提供了一个在所有函数范围内的变量或 b) [通常] 里面 main() 如果我用你的成语:

def parse_arguments():
    parser = argparse.ArgumentParser()
    parser.add_argument('arg1')
    parser.add_argument('arg2')
    args = parser.parse_args()
    return args

def main():
    args = parse_arguments()
    # do stuff with args.arg1 and args.arg2

Answer 2

你可以看到一个类似的问题here。

编辑：寻找一种不使用内部方法的方法，我发现this discussion建议使用vars()。这很好用：

import argparse

def main(arg1, arg2):
    print arg1, arg2

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('arg1')
    parser.add_argument('arg2')

    args = parser.parse_args()
    main(**vars(args))

Answer 3

https://docs.python.org/3/library/argparse.html#the-namespace-object

This class is deliberately simple, just an object subclass with a readable string representation. If you prefer to have dict-like view of the attributes, you can use the standard Python idiom, vars():

>>>
>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo')
>>> args = parser.parse_args(['--foo', 'BAR'])
>>> vars(args)
{'foo': 'BAR'}

请注意，从 optparse 到 argparse 的一大进步或至少变化之一是位置参数（例如您的参数）被视为与可选参数相同。它们都出现在 args Namespace 对象中。在 optparse 中，位置变量只是解析已定义选项的遗留物。您可以通过省略参数并使用 parse_known_args:

在 argparse 中获得相同的效果

parser = argparse.ArgumentParser()
args, extras = parser.parse_known_args()

args 现在是一个命名空间，extras 是一个列表。然后你可以调用你的函数：

myfoo(*extras, **vars(args))

例如：

In [994]: import argparse
In [995]: def foo(*args, **kwargs):
   .....:     print(args)
   .....:     print(kwargs)
   .....:     
In [996]: parser=argparse.ArgumentParser()
In [997]: parser.add_argument('-f','--foo')
Out[997]: _StoreAction(option_strings=['-f', '--foo'], dest='foo', nargs=None, const=None, default=None, type=None, choices=None, help=None, metavar=None)
In [998]: args,extras = parser.parse_known_args(['-f','foobar','arg1','arg2'])
In [999]: args
Out[999]: Namespace(foo='foobar')
In [1000]: extras
Out[1000]: ['arg1', 'arg2']
In [1001]: foo(*extras, **vars(args))
('arg1', 'arg2')
{'foo': 'foobar'}

同一个 argparse 段落表明您可以定义自己的 Namespace class。定义一个行为类似于字典（用作 **args）和命名空间的字典并不难。 argparse 所要求的是它与 getattr 和 setattr 一起工作。

In [1002]: getattr(args,'foo')
Out[1002]: 'foobar'
In [1004]: setattr(args,'bar','ugg')
In [1005]: args
Out[1005]: Namespace(bar='ugg', foo='foobar')

另一个标准 Python 功能让我可以将 vars(args) 作为元组传递：

In [1013]: foo(*vars(args).items())
(('foo', 'foobar'), ('bar', 'ugg'))
{}

对于去年 1 月的类似回答：

在那里我给出了如何在解析后从 'optionals' 中分离出 'positionals' 的想法。

这是一个自定义命名空间 class，在其 API 中包含一种将自身作为字典返回的方法：

In [1014]: class MyNameSpace(argparse.Namespace):
   ......:     def asdict(self):
   ......:         return vars(self)
   ......:     
In [1015]: args = parser.parse_args(['-f','foobar'], namespace=MyNameSpace())
In [1016]: args
Out[1016]: MyNameSpace(foo='foobar')
In [1017]: foo(**args.asdict())
()
{'foo': 'foobar'}

另一个想法 - 使用多个 nargs (2,'*','+') 之一作为位置参数。然后在将它传递给您的函数时您只需输入一个名称。

parser.add_argument('pos',nargs='+')
args = ...
args.pos # a list, possibly empty
foo(*args.pos, **vars(args))

从 argparse 解包参数

Unpacking arguments from argparse

python

argparse