需要使用 argparse 的两个参数之一

Question

给定：

import argparse

pa = argparse.ArgumentParser()
pa.add_argument('--foo')
pa.add_argument('--bar')

print pa.parse_args('--foo 1'.split())

我该怎么做

• 至少强制要求“foo, bar”之一：--foo x、--bar y 和 --foo x --bar y 都可以
• 强制设置最多一个“foo, bar”：--foo x 或 --bar y 可以，--foo x --bar y 不行

Answer 1

我认为您正在搜索类似 mutual exclusion 的内容（至少对于您问题的第二部分）。

这样，只会接受 foo 或 bar，而不是两者。

import argparse

parser = argparse.ArgumentParser()
group = parser.add_mutually_exclusive_group(required=True)
group.add_argument('--foo',action=.....)
group.add_argument('--bar',action=.....)
args = parser.parse_args()

顺便说一句，刚刚发现another question指的是同类型的问题。

Answer 2

如果您需要一些模块未提供的检查，您可以随时手动进行：

pa = argparse.ArgumentParser()
...
args = pa.parse_args()

if args.foo is None and args.bar is None:
   pa.error("at least one of --foo and --bar required")