如何让自定义Python异常在未被捕获时执行特定逻辑？

你的需求很明确——模仿SystemExit的行为：异常被捕获时程序正常继续，未被捕获时触发特定操作。先说说你原来的写法为什么行不通：你在MyFatalError的__init__里直接调用sys.exit()，但__init__是在实例化异常对象的时候就执行了，不管这个异常之后会不会被捕获。所以当你raise MyFatalError()的瞬间，程序已经退出了，根本没机会进入except块。

那符合Python风格的简洁解法是什么？答案是利用sys.excepthook——这是Python解释器专门用来处理未被捕获的异常的钩子函数。我们可以替换它，针对自定义异常添加专属逻辑，其他异常还是用默认处理方式。

实现代码示例 #

import sys

class MyFatalError(Exception):
    """自定义的“致命”异常，未被捕获时执行特定逻辑"""
    pass

# 保存原始的异常钩子，避免覆盖后丢失默认行为
original_excepthook = sys.excepthook

def custom_excepthook(exc_type, exc_value, exc_traceback):
    # 检查当前未被捕获的异常是不是我们的自定义异常
    if exc_type is MyFatalError:
        print("执行未被捕获时的特定代码！")
        # 这里可以添加你需要的任何逻辑，比如清理资源、记录日志、退出程序等
        sys.exit(1)
    # 非自定义异常，交给原始钩子处理
    original_excepthook(exc_type, exc_value, exc_traceback)

# 替换默认的异常钩子
sys.excepthook = custom_excepthook

# 测试场景1：异常被捕获
try:
    raise MyFatalError("测试捕获场景")
except MyFatalError:
    print("异常被捕获，程序继续执行...")

# 测试场景2：异常未被捕获（取消注释即可看到效果）
# raise MyFatalError("测试未捕获场景")

原理说明 #

• sys.excepthook只会在异常完全未被捕获的时候触发，完美匹配你的需求：被捕获时不执行特定逻辑，未被捕获时才执行。
• 我们保存了原始的excepthook，确保其他异常的处理不受影响，这是符合Python“不要破坏默认行为”的设计原则的。
• 你可以在custom_excepthook里针对MyFatalError添加任何逻辑——比如打印错误日志、清理临时文件、调用sys.exit()退出程序，完全和SystemExit的行为对齐。

另外，还有一种不太推荐的方法是利用异常的__del__方法，但__del__的执行时机由Python的垃圾回收机制决定，不确定性很高，所以用sys.excepthook是最稳妥、最符合Python风格的方案。

内容的提问来源于stack exchange，提问作者Arminius