Redhat Linux 上的 argparse python 3.6.11，如何控制程序退出

如何解决Redhat Linux 上的 argparse python 3.6.11，如何控制程序退出

我正在考虑将 argparse 添加到运行 python 的现有批处理作业中。这些作业采用命令行参数，这些参数可能因作业的运行方式而异。 我想在 argparse 中使用命名参数功能，但如果参数是例如，而不是退出作业。丢失或命名错误，我想捕获该错误并将后续处理交给处理作业退出的现有 python 例程。这些例程创建日志、回退数据库事务、将文件刷新到磁盘并向负责所遇到问题的工作的人员发送电子邮件。

我尝试使用 python try except block（如下）来捕获错误，但它失败了。 我还看到了涉及子类化 argparse 的解决方案，但这会导致代码库比引入 argparse 之前复杂得多，因此会适得其反。

对于我遗漏的这个问题，有没有简单的解决方案？ 非常感谢。

try:
    args = my_parser.parse_args()
except Exception as parseError:
    do my stuff

解决方法

您需要在创建 False 对象时将 exit_on_error 选项设置为 ArgumentParser。然后它将引发异常而不是退出脚本。

my_parser = argparse.ArgumentParser(exit_on_error=False)
# Add options to my_parser
try:
    args = my_parser.parse_args()
except argparse.ArgumentError as parseError:
    # do my stuff

如果您使用的是没有此选项的旧版 argparse，您可以创建一个 ArgumentParser 的子类来覆盖 ArgumentParser.exit() 方法以引发异常退出。

class ErrorCatchingArgumentParser(argparse.ArgumentParser):
    def exit(self,status=0,message=None):
        if status:
            raise Exception(f'Exiting because of an error: {message}')
        exit(status)

my_parser = ErrorCatchingArgumentParser()
# Add options to my_parser
try:
    args = my_parser.parse_args()
except Exception as parseError:
    # do my stuff

,

在 ipython 会话中：

In [3]: parser = argparse.ArgumentParser()
In [4]: parser.parse_args(['foo'])
usage: ipython3 [-h]
ipython3: error: unrecognized arguments: foo
An exception has occurred,use %tb to see the full traceback.
SystemExit: 2

/usr/local/lib/python3.8/dist-packages/IPython/core/interactiveshell.py:3426: UserWarning: To exit: use 'exit','quit',or Ctrl-D.
  warn("To exit: use 'exit',or Ctrl-D.",stacklevel=1)
In [5]: try:
   ...:     parser.parse_args(['foo'])
   ...: except SystemExit:
   ...:     pass
   ...: 
usage: ipython3 [-h]
ipython3: error: unrecognized arguments: foo

并捕获回溯：

In [12]: import sys
In [13]: try:
    ...:     parser.parse_args(['foo'])
    ...: except SystemExit:
    ...:     tb = sys.exc_info()
    ...: 
    ...: 
usage: ipython3 [-h]
ipython3: error: unrecognized arguments: foo
In [14]: print(tb)
(<class 'SystemExit'>,SystemExit(2),<traceback object at 0x7f9ffe415cc0>)