分组argparse子解析器参数

Question

我有一个脚本，它有多个命令，每个命令通过使用add_subparser获取它自己的一组必需和/或可选参数。

=->test.py -h
usage: test.py [-h] <command> ...
positional arguments:
  <command>   Available Commands
    cmd1      Command 1
    cmd2      Command 2
    cmd3      Command 3
    cmd4      Command 4
optional arguments:
  -h, --help  show this help message and exit

=->test.py cmd1 -h
usage: test.py cmd1 [-h] --flag1 FLAG1
optional arguments:
  -h, --help     show this help message and exit
  --flag1 FLAG1  Test flag

=->test.py cmd2 -h
usage: test.py cmd2 [-h] [--flag2 FLAG2]
optional arguments:
  -h, --help     show this help message and exit
  --flag2 FLAG2  Test flag

我想以某种方式将这些命令分成组，以便用户看到如下内容:

=->test.py -h
usage: test.py [-h] <command> ...
First Group:
  cmd1      Command 1
  cmd2      Command 2
Second Group:
  cmd3      Command 3
  cmd4      Command 4
optional arguments:
  -h, --help  show this help message and exit

但是，看起来add_argument_group和add_subparser不能一起工作。

有什么办法做到这一点吗?

Answer 1

你是对的，参数组和子解析器不能一起工作。这是因为子解析器(或者更确切地说是它们的名称)不是参数。

sp = parser.add_subparsers(...)命令创建一个参数，或者严格来说是argparse.Action子类的一个实例。这是一个位置参数。add_parser命令创建一个parser对象(即调用argparse.ArgumentParser)，并将其连同其名称(和别名)添加到该操作拥有的字典中。并且这些名称填充Action的choices属性。

subparsers Action可以属于一个参数组，但由于只能有一个这样的操作，因此它不能帮助您对帮助行进行分组。

在某种程度上，您可以通过使用描述来控制帮助，并省略子解析器的帮助

import argparse
description = """
First Group:
  cmd1      Command 1
  cmd2      Command 2
Second Group:
  cmd3      Command 3
  cmd4      Command 4"""
parser = argparse.ArgumentParser(formatter_class=argparse.RawDescriptionHelpFormatter)
sp = parser.add_subparsers(title='commands',description=description)
sp.add_parser('cmd1')
sp.add_parser('cmd2')
sp.add_parser('cmd3')
sp.add_parser('cmd4')
parser.print_help()

生产

1343:~/mypy$ python stack32017020.py 
usage: stack32017020.py [-h] {cmd1,cmd2,cmd3,cmd4} ...
optional arguments:
  -h, --help            show this help message and exit
commands:
  First Group:
    cmd1      Command 1
    cmd2      Command 2
  Second Group:
    cmd3      Command 3
    cmd4      Command 4
  {cmd1,cmd2,cmd3,cmd4}

---

http://bugs.python.org/issue9341 -允许argparse子命令分组

谈论做你想做的事。我提出的补丁不是微不足道的。但是欢迎您来测试

Answer 2

我遇到了这个问题，并通过在父解析器中创建一个组和一个函数来解决这个问题，该函数为给定子解析器的每个子解析器创建一个虚拟组，并用所需组替换虚拟组的对象，同时保留其地址，因此该组确实属于每个子解析器!

下面是一个带有一点上下文的例子:

import argparse
import ctypes

parent_parser = argparse.ArgumentParser(description="Redacted")
subparsers = parent_parser.add_subparsers()
subparser_email = subparsers.add_parser("email", parents=[parent_parser], add_help=False)
subparser_gen = subparsers.add_parser("gen", parents=[parent_parser], add_help=False)
group_emails = subparser_email.add_argument_group("Main inputs")
group_gen = subparser_gen.add_argument_group("Emails generation")
group_matchers = parent_parser.add_argument_group("Matchers") # The group we want on each subparser
[...] # add the arguments to your groups here
def add_group_to_subparsers(group: argparse._ArgumentGroup, subparsers: argparse._SubParsersAction, name: str):
    for name, subparser in subparsers._name_parser_map.items():
        dummy_group = subparser.add_argument_group(name)
        ctypes.memmove(id(dummy_group), id(group), object.__sizeof__(dummy_group))
add_group_to_subparsers(group_matchers, subparsers, "Matchers")
parent_parser.parse_args()