我有以下python程序:
#!/usr/bin/env python
import argparse
parser = argparse.ArgumentParser()
parser.add_argument('arg', choices=['foo', 'bar', 'baz'], default='foo', nargs='*')
args = parser.parse_args()
print(args)
如果我这样调用程序:
./prog.py
输出
Namespace(arg='foo')
但是如果我用foo
作为参数调用程序:
./prog.py foo
输出
Namespace(arg=['foo'])
问题
如何将arg
的默认值变为list
?
我已尝试
我已尝试设置default=['foo']
,但结果是:
prog.py: error: argument arg: invalid choice: ['foo'] (choose from 'foo', 'bar', 'baz')
答案 0 :(得分:6)
这是旧的,但是开放的错误/问题
的副本 http://bugs.python.org/issue9625(argparse: Problem with defaults for variable nargs when using choices
)
positional
*
得到一些特殊处理。如果您不提供值,则默认值始终通过choices
测试。
将其与optional
In [138]: p=argparse.ArgumentParser()
In [139]: a=p.add_argument('--arg',choices=['foo','bar','baz'],nargs='*')
In [140]: p.parse_args([])
Out[140]: Namespace(arg=None)
In [141]: a.default=['foo']
In [142]: p.parse_args([])
Out[142]: Namespace(arg=['foo'])
默认是在没有测试的情况下接受的:
In [143]: a.default=['xxx']
In [144]: p.parse_args([])
Out[144]: Namespace(arg=['xxx'])
相关代码是:
def _get_values(self, action, arg_strings):
...
# when nargs='*' on a positional, if there were no command-line
# args, use the default if it is anything other than None
elif (not arg_strings and action.nargs == ZERO_OR_MORE and
not action.option_strings):
if action.default is not None:
value = action.default
else:
value = arg_strings
self._check_value(action, value)
建议的错误/问题补丁对此代码块进行了一些小改动。