完全接受 0、1 或 2 个位置参数 [重复]

Question

我正在使用 argparse 包来解析命令行参数。现在我想准确地接受 0、1 或 2 个字符串并将它们放入一个列表中。

我目前的做法是这样的：

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('strs', nargs='*')

print(parser.parse_args([]).strs)
print(parser.parse_args(['AAA']).strs)
print(parser.parse_args(['AAA', 'BBB']).strs)
print(parser.parse_args(['AAA', 'BBB', 'CCC']).strs)

对于第一个、第二个和第三个parse_args()，结果是预期的。对于最后一个，我目前的方法是if len(parse_args.strs) > 2: raise。有没有更好的方法可以让 argparse 为我检查？

Answer 1

argparse 中没有办法接受nargs 的范围，只有1、2、'+'、'*' 之类的值。

您可以编写一个自定义的Action 类来检查nargs='*' 传递的值的数量。但我认为在解析后执行该测试同样容易。

如果您真的想深入了解，我可以找到一个探索添加范围 nargs 选项的错误/问题。

如果您没有其他位置，您可以定义 3 个位置，其中 2 个使用 ?。您甚至可以将它们组合成一个具有通用名称和“附加”操作的列表。我会探索的。

In [573]: p=argparse.ArgumentParser()
In [574]: p.add_argument('str',action='append')
In [575]: p.add_argument('str',nargs='?',action='append')
In [576]: p.add_argument('str',nargs='?',action='append')

In [577]: p.print_help()
usage: ipython3 [-h] str [str] [str]

positional arguments:
  str
  str
  str

optional arguments:
  -h, --help  show this help message and exit

In [578]: p.parse_args('1 2 3'.split())
Out[578]: Namespace(str=['1', '2', '3'])
In [579]: p.parse_args('1 3'.split())
Out[579]: Namespace(str=['1', '3', None])
In [580]: p.parse_args('1'.split())
Out[580]: Namespace(str=['1', None, None])

得到default 无。我可以用

来纠正

p.add_argument('str', nargs='?', action='append', default=argparse.SUPPRESS)

In [586]: p.parse_args('1 2'.split())
Out[586]: Namespace(str=['1', '2'])
In [588]: p.parse_args('1 2 3 4'.split())
usage: ipython3 [-h] str [str] [str]
ipython3: error: unrecognized arguments: 4

（哎呀，这是为 1-3 参数编码的，但你明白了）。