argparse：乱序处理某些参数

如何解决argparse：乱序处理某些参数

我有一个 python 应用程序，它有一些可重复的参数，交错顺序很重要。例如app -f 1 -b 2 -f 3 与 app -f 1 -f 3 -b 2 不是同一个命令。

问题是我也有详细控制参数，我希望它影响在操作中执行的代码的日志记录行为。

有没有办法让 argparse 乱序处理某些参数，以便在处理其他参数之前应用详细控制，而不管它出现在 argv 中的什么位置？

# 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,option_string=None):
        foo(values)

class PerformBar(argparse.Action):
    def __call__(self,option_string=None):
        bar(values)

parser = argparse.ArgumentParser()

parser.add_argument('-v','--verbose',nargs='?',action=SetVerbosity(logging.INFO))
parser.add_argument('-vv',action=SetVerbosity(logging.DEBUG))

parser.add_argument('-f','--foo',Metavar='VAL',action=PerformFoo,help='Do a foo')
parser.add_argument('-b','--bar',action=PerformBar,help='Do a bar')

logging.basicConfig()
parser.parse_args()

# No problem
python test.py -v -f 1 -b 2 -f 3
INFO:root:hi foo 1
INFO:root:hi foo 3

# kind of unfortunate
app -f 1 -b 2 -f 3 -v
*crickets*

作为后备，我知道我可以将 -f 和 -b 累积在一个异类的“待办事项”列表中，并在事后处理它们。但是，如果有一种相当简单的方法来解决这个问题，而无需额外的间接层，我宁愿不跳过那个额外的环节。

解决方法

看到选项时不要调用 foo 和 var。相反，在解析所有选项后收集要执行的操作列表。

类似的东西

import argparse
import logging

def foo(x):
  pass

def bar(x):
  pass

def SetVerbosity(value):
    class Action(argparse.Action):
        def __call__(self,parser,namespace,values,option_string=None):
            logging.getLogger(values).setLevel(level=value)

    return Action

class PerformFoo(argparse.Action):
    def __call__(self,option_string=None):
        namespace.to_do += [(foo,values)]

class PerformBar(argparse.Action):
    def __call__(self,option_string=None):
        namespace.to_do += [(bar,values)]

parser = argparse.ArgumentParser()

parser.add_argument('-v','--verbose',nargs='?',action=SetVerbosity(logging.INFO))
parser.add_argument('-vv',action=SetVerbosity(logging.DEBUG))

parser.add_argument('-f','--foo',metavar='VAL',action=PerformFoo,help='Do a foo')
parser.add_argument('-b','--bar',action=PerformBar,help='Do a bar')

args = parser.parse_args()

for f,v in args.to_do:
    f(v)

,

按照历史悠久的传统，我在发布问题后不久就设法找到了答案。

如果其他人在这里登陆，我设法通过创建第二个解析器并使用 parse_known_args() 调用它来实现，以便忽略非详细参数。

verbosityParser = argparse.ArgumentParser(add_help=False)
parser = argparse.ArgumentParser()

verbosityParser.add_argument('-v',action=SetVerbosity(logging.INFO))
verbosityParser.add_argument('-vv',action=SetVerbosity(logging.DEBUG))

# Still add verbosity to the "main" parser so that it shows up in help
parser.add_argument('-v',nargs='?')
parser.add_argument('-vv',nargs='?')
parser.add_argument('-f',help='Do a bar')

logging.basicConfig()
verbosityParser.parse_known_args()
parser.parse_args()