如何在Python上下文管理器中转换异常类型？

先明确上下文管理器__exit__方法的返回值规则 #

Python上下文管理器的__exit__方法返回值为布尔类型：

• 返回True：表示已处理异常，解释器不会再向上传播该异常
• 返回False（或默认返回None）：异常会继续向上传播

如果在__exit__里主动抛出新异常，无论返回值是什么，新的异常都会取代原异常继续传播。

实现方式一：手动编写上下文管理器类 #

class ExceptionWrapper:
    def __init__(self, target_exception):
        self.target_exception = target_exception

    def __enter__(self):
        # 进入上下文时无需额外操作
        pass

    def __exit__(self, exc_type, exc_val, exc_tb):
        # 仅当with块内抛出异常时执行转换逻辑
        if exc_type is not None:
            # 将原异常包装为目标异常抛出，保留异常上下文
            raise self.target_exception from exc_val

实现方式二：用contextlib.contextmanager装饰器（更简洁） #

借助contextlib的装饰器可以快速实现生成器风格的上下文管理器，逻辑和原样板代码完全对应：

from contextlib import contextmanager

@contextmanager
def ExceptionWrapper(target_exception):
    try:
        yield  # 执行with块内的业务代码
    except Exception as exc:
        # 捕获异常并转换为目标类型，保留原异常上下文
        raise target_exception from exc

使用示例 #

完全符合你期望的调用方式：

# 自定义异常示例
class SomethingError(Exception):
    pass

def something():
    # 模拟抛出异常
    raise ValueError("原始异常信息")

# 使用上下文管理器
with ExceptionWrapper(SomethingError):
    something()

运行后会抛出SomethingError，且原ValueError会作为异常的__cause__保留，和原样板代码的行为完全一致。

扩展：指定捕获的异常类型 #

如果不想捕获所有Exception，可以扩展实现，支持指定要捕获的异常类型：

from contextlib import contextmanager

@contextmanager
def ExceptionWrapper(target_exception, catch_type=Exception):
    try:
        yield
    except catch_type as exc:
        raise target_exception from exc

# 示例：仅捕获IOError及其子类
with ExceptionWrapper(SomethingError, catch_type=IOError):
    something()

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