包装类
这是我迄今为止发现的最可靠和DRY(不重复,无冗余)的方法,既可以显示默认值,又可以同时使用其他格式化程序,比如argparse.RawTextHelpFormatter
:
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:
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)
它能够正常工作是因为我们可以从源代码 trivially 看到 https://github.com/python/cpython/blob/v3.6.5/Lib/argparse.py#L648 :
RawTextHelpFormatter
实现了 _split_lines
ArgumentDefaultsHelpFormatter
实现了 _get_help_string
所以我们可以猜测它们将很好地一起工作。
然而,这似乎不是公共 API,formatter_class
的方法也不是公共 API,因此我不认为目前有公共 API 的方法可以这样做。 argparse
文档字符串说:
本模块中的所有其他类都被认为是实现细节。(还要注意,HelpFormatter 和 RawDescriptionHelpFormatter 只被视为对象名称时公共的 —— 格式化程序对象的 API 仍然被认为是实现细节。)
另请参见:自定义 argparse 帮助消息
在 Python 3.6.5 上进行了测试。