我希望Python argparse抛出异常而不是使用

Question

我不认为这是可能的,但我想自己处理来自argparse的异常.

例如:

import argparse
parser = argparse.ArgumentParser()
parser.add_argument('--foo', help='foo help', required=True)
try:
    args = parser.parse_args()
except:
    do_something()

当我运行它:

$ myapp.py
usage: myapp --foo foo
myapp: error: argument --foo is required

但我希望它落入例外.

Answer 1

您可以子类化ArgumentParser并覆盖该error方法,以便在发生错误时执行不同的操作:

class ArgumentParserError(Exception): pass

class ThrowingArgumentParser(argparse.ArgumentParser):
    def error(self, message):
        raise ArgumentParserError(message)

parser = ThrowingArgumentParser()
parser.add_argument(...)
...

Answer 2

在我的例子中,argparse打印"参数太少"然后退出.在阅读了argparse代码之后,我发现在打印一些消息后它只是调用sys.exit().因为sys.exit()除了抛出SystemExit异常之外什么都不做,你可以捕获这个异常.

所以试试看看它是否适合你.

    try:
        args = parser.parse_args(args)
    except SystemExit:
        .... your handler here ...
        return