在 Python 中调用 atexit 回调时如何查找退出代码或原因？

Question

我想知道 Python 脚本是否正确终止。为此，我使用atexit，但问题是我不知道如何区分atexit 是使用sys.exit(0) 还是非零或异常调用的。

推理：如果程序正常结束，它什么也不做，但如果程序以异常结束或返回不同于零的错误代码（退出状态）我想触发一些动作。

如果您想知道为什么我不使用 try/finally 是因为我想为导入公共模块的十几个脚本添加相同的行为。我不想修改所有这些，而是​​想将 atexit() hack 添加到正在导入的模块中，并在所有这些模块中免费获得此行为。

Answer 1

您可以使用sys.excepthook 和猴子补丁sys.exit() 解决此问题：

import atexit
import sys

class ExitHooks(object):
    def __init__(self):
        self.exit_code = None
        self.exception = None

    def hook(self):
        self._orig_exit = sys.exit
        sys.exit = self.exit
        sys.excepthook = self.exc_handler

    def exit(self, code=0):
        self.exit_code = code
        self._orig_exit(code)

    def exc_handler(self, exc_type, exc, *args):
        self.exception = exc

hooks = ExitHooks()
hooks.hook()

def foo():
    if hooks.exit_code is not None:
        print("death by sys.exit(%d)" % hooks.exit_code)
    elif hooks.exception is not None:
        print("death by exception: %s" % hooks.exception)
    else:
        print("natural death")
atexit.register(foo)

# test
sys.exit(1)