argparse接受一切

时间:2012-03-10 01:52:40

标签: python argparse

有没有办法让argparse.ArgumentParser在读取未知选项时不会引发异常,而是将所有未知选项的值放在字典中,将那些没有值的列表放在列表中?

例如,假设在prog.py的解析器中没有定义参数,我传递了两个参数:

./prog.py --foo bar --baz

我想要以下内容:

parsed = parser.parse_args()
vals = parsed.unknown_with_vals
novals = parsed.unknown_without_vals

print(vals)
#{'foo' : 'bar'}
print(novals)
#['baz']

可以这样做吗?

1 个答案:

答案 0 :(得分:9)

known, unknown_args = parser.parse_known_args(...)

正如@ben w在评论中指出的,你如何解析unknown_args,例如,使用以下语法:

unknown_args = *(with_val / without_val) EOS
with_val = OPT 1*VALUE
without_val = OPT
OPT = <argument that starts with "--">
VALUE = <argument that doesn't start with "--">

或者作为正则表达式:

(O V+ | O)* $

注意:在这种情况下,禁止使用孤儿值。

Example

d = {}
for arg in unknown_args:
    if arg.startswith('--'): # O
        opt = arg
        d[opt] = []
    else: # V
        d[opt].append(arg) #NOTE: produces NameError if an orphan encountered

with_vals = {k: v for k, v in d.items() if v}
without_vals = [k for k, v in d.items() if not v]