nargs 取决于另一个设置？

Question

我正在尝试编写一个支持任意位操作的程序：位图的 AND、OR、NOT 和 COUNT。用法是您运行 program.py --and f1.bit f2.bit 并将结果打印到标准输出。

问题是我希望解析器处理所有警告。具体来说，我希望 nargs 取决于设置的模式 - 如果设置为 COUNT 或 NOT，则需要一个文件，如果设置为 OR 或 AND，则需要两个文件。这是一些（非工作）示例代码：

#!/usr/bin/env python

import argparse

def main(mode, fnames):
    pass

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('-O', '--or',
        nargs=2,
        action='store_const', const='or'
    )
    args = parser.parse_args()

    import pprint
    pprint.pprint(args.__dict__)

    #main(**args.__dict__)

我得到的错误是：

Traceback (most recent call last):
  File "bitmaptool.py", line 12, in <module>
    action='store_const', const='or'
  File "/usr/lib/python3.7/argparse.py", line 1362, in add_argument
    action = action_class(**kwargs)
TypeError: __init__() got an unexpected keyword argument 'nargs'

注释掉nargs 有帮助，留下nargs 但注释掉action 也有帮助 - 但我两者都想要。我需要手动实现它还是有什么技巧或其他库可以让我到达那里？

编辑我想通过显示我需要手动编写的代码来阐明我在寻找什么：

if __name__ == '__main__':
    parser = argparse.ArgumentParser(argument_default=argparse.SUPPRESS)
    parser.add_argument('-O', '--or', nargs=2)
    parser.add_argument('-A', '--and', nargs=2)
    parser.add_argument('-M', '--minus', nargs=2)
    parser.add_argument('-C', '--count', nargs=1)
    parser.add_argument('-N', '--not', nargs=1)
    parser.add_argument('-o', '--output', default='/dev/stdout')
    args = parser.parse_args().__dict__

    mode = None
    files = []
    for current_mode in ['or', 'and', 'not', 'count']:
        if current_mode in args:
            if mode is not None:
                sys.exit('ERROR: more than one mode was specified')
            mode = current_mode
            files = args[mode]

    if mode is None:
        sys.stderr.write('ERROR: no mode was specified\n\n')
        parser.print_help()
        sys.exit(1)

    import pprint
    pprint.pprint(args)

有没有更优雅的方式到达那里？

Answer 1

store_const 从不 获取参数，它实际上将您所说的内容存储为const 或None。因为它是一个常数，而不是一个变量。来自argparse's action documentation，重点是我的：

'store_const' - 存储由 const 关键字参数指定的值。 'store_const' 操作最常与指定某种标志的可选参数一起使用。

您应该将操作更改为实际存储传递的文件名的内容。根据argparse's nargs documentation 和示例，您实际上根本不需要指定action，默认（action='store'）就足够了。

文档示例：

>>> parser = argparse.ArgumentParser()
>>> parser.add_argument('--foo', nargs=2) #this line
>>> parser.add_argument('bar', nargs=1)
>>> parser.parse_args('c --foo a b'.split())
Namespace(bar=['c'], foo=['a', 'b'])

---

EDIT 用于问题的编辑版本 - mutually exclusive group 将确保仅指定一个参数（当然来自该组）：

if __name__ == '__main__':
    parser = argparse.ArgumentParser(argument_default=argparse.SUPPRESS)
    parser.add_argument('-o', '--output', default='/dev/stdout')

    group = parser.add_mutually_exclusive_group(required=True)
    group.add_argument('-O', '--or', nargs=2)
    group.add_argument('-A', '--and', nargs=2)
    group.add_argument('-M', '--minus', nargs=2)
    group.add_argument('-C', '--count', nargs=1)
    group.add_argument('-N', '--not', nargs=1)

    args = parser.parse_args().__dict__

    import pprint
    pprint.pprint(args)