如何至少需要一个参数并有可能在 Python 中使用 argparse 获取所有参数？

Question

该程序有 2 个参数需要处理：状态和密钥。 我需要有可能提供以下选项作为输入：

• prog -state state_value
• prog -key key_value
• prog -state state_value -key key_value

最接近的是使用互斥组，但它无法给出 两个参数同时作为输入。

Answer 1

正如https://stackoverflow.com/a/24911007/901925 中所述，没有内置的机制来测试这个，但是在解析后很容易进行这样的测试。假设参数有默认的default 或None，这样的测试可能是：

if args.state is None and args.key is None:
    parser.error('At least one of the arguments is required ')

---

最终argparse 可能会有这样的机制。有一个BUG请求更多的分组测试能力，http://bugs.python.org/issue11588

一个想法是添加 UsageGroups，以互斥组为模型，但具有更一般的逻辑。该代码无法下载，但我很想知道这个示例是否清晰。

parser = argparse.ArgumentParser(prog='prog')
group = parser.add_usage_group(kind='any', required=True)
group.add_argument('-s', '--state', metavar='state_value')
group.add_argument('-k', '--key', metavar='key_value')
args = parser.parse_args()
print(args)

带有错误消息：

usage: prog [-h] (-s state_value | -k key_value)
prog: error: group any(): some of the arguments [state, key] is required

用法行中的替代方法是(-s state_value, -k key_value)，因为| 已经被用于互斥的xor 关系。 usage 和错误信息可以自定义。

Answer 2

我认为这超出了 argparse 的能力。不过，您可以在之后对返回的命名空间执行简单的验证。您应该在帮助中包含至少需要一次状态或密钥。

def validate(namespace, parser):
    if not any(arg in {"state", "key"} for arg in vars(namespace).keys()):
        parser.print_help()
        parser.exit(2)

namespace = parser.parse_args()
validate(namespace, parser)

Answer 3

#!/usr/bin/python
#-*- coding:utf-8 -*-

import sys

paras = sys.argv.pop(0)
state_value = None
key_value = None
i=0
while i < len(paras):
    if paras[i] == '-state':
        state_value = paras[i]
    elif paras[i] == '-key':
        key_value = paras[i]
    i += 2

if state_value:
    # do something
    pass
if key_value:
    # do something
    pass