包装器类
这是我迄今为止发现的既可以显示默认值又可以使用其他格式化程序(如argparse)的最可靠的DRY方法。同时使用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文档字符串说:
此模块中的所有其他类都被视为实现细节。
(还要注意,HelpFormatter和RawDescriptionHelpFormatter只是
作为对象名被认为是公共的——格式化器对象的API是
仍然被认为是一个实现细节。)
参见:自定义argparse帮助消息
在Python 3.6.5上测试。