如何检测Python代码是否抛出异常并实现无异常执行自动化验证

针对你的需求——自动化验证大型代码块无未处理异常，同时避免生产环境的异常处理开销，下面是几个实用方案：

方案1：过滤sys.settrace的异常事件，仅追踪自身代码 #

sys.settrace会捕获所有异常事件（包括库内部已处理的），但你可以通过栈帧的文件路径过滤，只关注自己项目代码抛出的异常：

import sys
import os

# 替换为你的项目代码根目录
MY_PROJECT_ROOT = os.path.abspath("/path/to/your/project")

def trace_handler(frame, event, arg):
    if event == "exception":
        exc_type, exc_val, exc_tb = arg
        # 检查异常所在栈帧是否属于项目代码
        frame_path = os.path.abspath(frame.filename)
        if frame_path.startswith(MY_PROJECT_ROOT):
            # 这里可以记录异常、启动调试器或标记测试失败
            print(f"检测到项目代码异常: {exc_type.__name__}: {exc_val}")
            import pdb; pdb.post_mortem(exc_tb)
            sys.exit(1)
    return trace_handler

def run_my_code_test():
    sys.settrace(trace_handler)
    # 执行你的目标代码块
    your_large_code_block()
    sys.settrace(None)

这个逻辑和调试器忽略库内部异常的原理一致——通过路径区分用户代码和第三方库代码，只处理前者的异常事件。

方案2：直接捕获顶层未处理异常（最简便） #

实际上，第三方库内部的异常会被库自身的try-except捕获，不会传播到代码顶层。你只需在测试环境中用try-except包裹代码块，捕获那些真正未被处理的异常即可，测试阶段的异常处理开销完全不影响生产环境：

import traceback
import pdb

def test_my_code():
    try:
        # 执行目标代码块
        your_large_code_block()
    except Exception:
        # 打印完整栈信息，或启动调试器定位问题
        traceback.print_exc()
        pdb.post_mortem()
        raise  # 让测试框架（如pytest/unittest）标记测试失败

这个方法最直接，完全匹配你手动用调试器验证的逻辑——只有未被处理的异常才会触发捕获，库内部的已处理异常会被自动忽略。

方案3：用调试器API自动化断点 #

如果你想完全复刻调试器的异常断点体验，可以使用pdb或其他调试器的编程接口，设置仅在用户代码中触发异常断点：

import pdb

def test_with_debugger():
    debugger = pdb.Pdb()
    # 设置仅在用户代码中触发异常断点
    debugger.set_break("exception", only_my_code=True)
    # 自动运行代码，遇到用户代码异常时进入调试
    debugger.run("your_large_code_block()")

这个方案适合需要深度调试的场景，但不同调试器的API细节可能略有差异。

总结 #

• 追求精准控制选方案1，和调试器逻辑对齐；
• 追求简便高效选方案2，能覆盖绝大多数回归测试场景；
• 需要完全复刻调试器体验选方案3。

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