使用argparse模块里的ArgumentParser类创建命令行解析器,并且使用format_help()方法来格式化帮助文档输出,利用colorama模块来给输出内容添加颜色,同时通过重载ArgumentParser类的“_get_help_string”方法实现删除标志后的空格。
以下是代码示例:
import argparse
import colorama
class CustomArgumentParser(argparse.ArgumentParser):
def _get_help_string(self, action):
help = action.help
if '%(default)' in action.help or action.default == argparse.SUPPRESS:
if action.default == argparse.SUPPRESS:
defaulting_nargs = ''
else:
defaulting_nargs = ' (default: %(default)s)'
help = help + defaulting_nargs
return help
if __name__ == '__main__':
colorama.init()
parser = CustomArgumentParser()
parser.add_argument('-f', '--file', help=colorama.Fore.GREEN + 'path to input file' + colorama.Fore.RESET)
parser.print_help()
colorama.deinit()
输出:
usage: __main__.py [-h] [-f FILE]
optional arguments:
-h, --help show this help message and exit
-f FILE, --file FILE path to input file (default: None)