argparse如何优先处理特定参数 让verbosity参数无论位置均先生效
问题根因
argparse默认严格按照命令行输入的参数顺序执行对应action,所以当-v/-vv放在其他参数之后时,执行-f/-b对应的action时日志级别还没有被调整,自然不会输出对应级别的日志。
最简解决方案
用两次参数解析的方式即可解决,不需要修改原有业务action逻辑,也不需要额外维护待执行队列:
- 先创建一个仅包含verbosity参数的预解析器,优先解析所有命令行参数中的日志配置项,提前完成日志级别设置
- 再用原有逻辑正常解析所有参数执行业务action
修改后完整代码示例
# test.py import argparse import logging def foo(x): logging.info(f'hi foo {x}') def bar(x): logging.debug(f'hi bar {x}') def SetVerbosity(log_level): class Action(argparse.Action): def __call__(self, parser, namespace, values, option_string=None): logging.getLogger(values).setLevel(level=log_level) return Action class PerformFoo(argparse.Action): def __call__(self, parser, namespace, values, option_string=None): foo(values) class PerformBar(argparse.Action): def __call__(self, parser, namespace, values, option_string=None): bar(values) logging.basicConfig() # === 新增预解析逻辑 开始 === # 预解析器不添加默认的-h参数,避免和主解析器冲突 pre_parser = argparse.ArgumentParser(add_help=False) pre_parser.add_argument('-v', '--verbose', nargs='?', action=SetVerbosity(logging.INFO)) pre_parser.add_argument('-vv', nargs='?', action=SetVerbosity(logging.DEBUG)) # 仅解析识别的参数,忽略其他所有参数,不会报错 pre_parser.parse_known_args() # === 新增预解析逻辑 结束 === parser = argparse.ArgumentParser() # 主解析器中保留verbosity参数定义,不影响帮助信息展示 parser.add_argument('-v', '--verbose', nargs='?', action=SetVerbosity(logging.INFO)) parser.add_argument('-vv', nargs='?', action=SetVerbosity(logging.DEBUG)) parser.add_argument('-f', '--foo', metavar='VAL', action=PerformFoo, help='Do a foo') parser.add_argument('-b', '--bar', metavar='VAL', action=PerformBar, help='Do a bar') parser.parse_args()
方案说明
- 预解析阶段用
parse_known_args方法处理参数,不管-v/-vv在命令行的哪个位置,都会被优先识别执行,提前完成日志级别配置 - 主解析器中保留verbosity参数的定义,不会影响
--help信息的输出,重复解析相同参数也仅会重复设置一次相同的日志级别,无副作用 - 完全保留原有参数按顺序执行的逻辑,
-f和-b的交错顺序不会受到任何影响
内容的提问来源于stack exchange,提问作者user4442671
相关产品推荐
相关产品推荐

