Argparse：在“--help”中包含默认值的方法？

Question

假设我有以下 argparse sn-p:

diags.cmdln_parser.add_argument( '--scan-time',
                     action  = 'store',
                     nargs   = '?',
                     type    = int,
                     default = 5,
                     help    = "Wait SCAN-TIME seconds between status checks.")

目前，--help 返回：

usage: connection_check.py [-h]
                             [--version] [--scan-time [SCAN_TIME]]

          Test the reliability/uptime of a connection.



optional arguments:
-h, --help            show this help message and exit
--version             show program's version number and exit
--scan-time [SCAN_TIME]
                    Wait SCAN-TIME seconds between status checks.

我更喜欢这样的：

--scan-time [SCAN_TIME]
                    Wait SCAN-TIME seconds between status checks.
                    (Default = 5)

查看帮助格式化程序代码发现了有限的选项。有没有一种聪明的方法让argparse 以类似的方式打印--scan-time 的默认值，或者我应该只继承help 格式化程序？

Answer 1

使用argparse.ArgumentDefaultsHelpFormatter formatter：

parser = argparse.ArgumentParser(
    # ... other options ...
    formatter_class=argparse.ArgumentDefaultsHelpFormatter)

引用documentation：

另一个可用的格式化程序类ArgumentDefaultsHelpFormatter 将添加有关每个参数的默认值的信息。

请注意，这仅适用于定义了帮助文本的参数；如果参数没有help 值，则没有帮助消息添加有关默认值到的信息。

您的扫描时间选项的确切输出将变为：

  --scan-time [SCAN_TIME]
                        Wait SCAN-TIME seconds between status checks.
                        (default: 5)

Answer 2

将'%(default)s'添加到帮助参数以控制显示的内容。

parser.add_argument("--type", default="toto", choices=["toto","titi"],
                              help = "type (default: %(default)s)")

注意事项：

• 它是 %+ 默认括号 + 格式字符（不要与我们在 format 或 f-string 中找到的大括号 {default} 混淆）
• 不要忘记在末尾添加类型表示的“说明符字符”（例如，s 用于字符串，d 用于整数，f 用于浮点数等）
• 您还可以添加常用的“printf”格式说明符（如浮点数、前导零等）

您可以参考printf documentation了解更多详情。

Answer 3

包装类

这是迄今为止我发现的最可靠和最简单的方法，既可以显示默认值，又可以同时使用其他格式化程序，例如 argparse.RawTextHelpFormatter：

#!/usr/bin/env python3

import argparse

class ArgumentParserWithDefaults(argparse.ArgumentParser):
    def add_argument(self, *args, help=None, default=None, **kwargs):
        if help is not None:
            kwargs['help'] = help
        if default is not None and args[0] != '-h':
            kwargs['default'] = default
            if help is not None:
                kwargs['help'] += ' Default: {}'.format(default)
        super().add_argument(*args, **kwargs)

parser = ArgumentParserWithDefaults(
    formatter_class=argparse.RawTextHelpFormatter
)
parser.add_argument('-a', default=13, help='''my help
for a''')
parser.add_argument('-b', default=42, help='''my help
for b''')
parser.add_argument('--no-default', help='''my help
for no-default''')
parser.add_argument('--no-help', default=101)

parser.print_help()
print()
print(parser.parse_args())

输出：

usage: main.py [-h] [-a A] [-b B] [--no-default NO_DEFAULT]
               [--no-help NO_HELP]

optional arguments:
  -h, --help            show this help message and exit
  -a A                  my help
                        for a Default: 13
  -b B                  my help
                        for b Default: 42
  --no-default NO_DEFAULT
                        my help
                        for no-default
  --no-help NO_HELP

Namespace(a=13, b=42, no_default=None, no_help=101)

ArgumentDefaultsHelpFormatter + RawTextHelpFormatter 多重继承

多重继承才行，但好像不是公共API：

#!/usr/bin/env python3

import argparse

class RawTextArgumentDefaultsHelpFormatter(
        argparse.ArgumentDefaultsHelpFormatter,
        argparse.RawTextHelpFormatter
    ):
        pass

parser = argparse.ArgumentParser(
    formatter_class=RawTextArgumentDefaultsHelpFormatter
)
parser.add_argument('-a', default=13, help='''my help
for a''')
parser.add_argument('-b', default=42, help='''my help
for b''')
parser.print_help()

输出：

usage: a.py [-h] [-a A] [-b B]

optional arguments:
  -h, --help  show this help message and exit
  -a A        my help
              for a (default: 13)
  -b B        my help
              for b (default: 42)

它可以正常工作，因为我们可以从https://github.com/python/cpython/blob/v3.6.5/Lib/argparse.py#L648 的来源中看到：

• RawTextHelpFormatter 实现 _split_lines
• ArgumentDefaultsHelpFormatter 实现 _get_help_string

所以我们可以猜测它们会很好地协同工作。

但是，这似乎不是公共 API，formatter_class 的方法也不是，所以我认为目前没有公共 API 方法可以做到这一点。 argparse docstring 说：

此模块中的所有其他类都被视为实现细节。 （另请注意，HelpFormatter 和 RawDescriptionHelpFormatter 仅 被视为公共对象名称 - 格式化程序对象的 API 是 仍被视为实现细节。）

另见：Customize argparse help message

在 Python 3.6.5 上测试。

Answer 4

能够自动在帮助输出中包含default值通常很有用，但仅限于那些明确指定的值（使用default=..） .已经提到的方法在这方面存在一些不足：

• ArgumentDefaultsHelpFormattermethod 为未明确指定默认值的每个参数打印出(default: None)，为“标志”(action='store_true') 打印出(default: False)。这会使帮助输出变得混乱。为避免这种情况，需要为每个此类参数手动添加default=argparse.SUPPRESS。

• '%(default)s' method 需要手动将其添加到我们确实希望在帮助中打印的所有参数的help 字符串中。

这两种方法最终都需要人工干预才能仅打印出“正确”的默认值。 一种自动执行此操作的方法是增加 ArgumentDefaultsHelpFormatter 以忽略 Nones 和 Falses 默认值：

class ExplicitDefaultsHelpFormatter(argparse.ArgumentDefaultsHelpFormatter):
    def _get_help_string(self, action):
        if action.default in (None, False):
            return action.help
        return super()._get_help_string(action)

用它代替ArgumentDefaultsHelpFormatter:

parser = argparse.ArgumentParser(
        formatter_class=ExplicitDefaultsHelpFormatter
                )

这将只打印帮助输出中明确设置的default 值。

注意：如果参数的默认值显式设置为None 或False，则不会在此类的帮助中显示；如果您希望在帮助输出中使用该参数，请将 %(default)s 字符串添加到 help。