我希望 Python argparse 抛出异常而不是使用
问题描述:
我不认为这是可能的,但我想自己处理来自 argparse 的异常.
I don't think this is possible, but I want to handle exceptions from argparse myself.
例如:
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
但我希望它陷入异常.
答
您可以子类化 ArgumentParser
并覆盖 error
方法以在发生错误时执行不同的操作:
You can subclass ArgumentParser
and override the error
method to do something different when an error occurs:
class ArgumentParserError(Exception): pass
class ThrowingArgumentParser(argparse.ArgumentParser):
def error(self, message):
raise ArgumentParserError(message)
parser = ThrowingArgumentParser()
parser.add_argument(...)
...