如何在Python中利用__enter__与__exit__自定义上下文管理器？-Python教程-PHP中文网

__enter__和__exit__易出错因签名与返回值有严格约定：__enter__须返回as绑定对象，__exit__必须接收3个异常参数且返回布尔值——漏参、返非布尔值或误抛异常均导致with行为失控。

如何在python中利用__enter__与__exit__自定义上下文管理器？

为什么直接写 `enter` 和 `exit` 容易出错？

因为这两个方法的签名和返回值有隐含约定：__enter__ 应该返回要绑定到 as 后变量的对象（或 None），而 __exit__ 必须接收 3 个参数（exc_type, exc_value, traceback）且返回布尔值——返回 True 才会抑制异常。漏掉参数、返回非布尔值、或在 __exit__ 里误抛异常，都会导致 with 块行为失控。

常见错误现象包括：

TypeError: __exit__() takes 1 positional argument but 4 were given —— 忘记加那三个异常参数
AttributeError: __exit__ —— 拼错方法名（比如写成 __exxit__）
异常没被捕获，或本该吞掉的异常却冒出来了 —— __exit__ 返回了 None 或 False

如何让 `exit` 正确处理异常并释放资源？

__exit__ 的核心职责不是“处理异常”，而是“决定是否吞掉它”。资源清理（如关闭文件、回滚事务）必须放在 __exit__ 里无条件执行，不能依赖 if exc_type is not None: 来触发。

实操建议：

所有清理逻辑（如 self.file.close()）写在 __exit__ 开头，不加条件判断
只在最后用 return True 显式抑制异常；否则默认返回 None（等价于 False），异常继续上抛
如果需要记录异常但不抑制，别在 __exit__ 里 print() 或 logging.error() 后忘了 return False

示例片段：

def __exit__(self, exc_type, exc_value, traceback):
    self.file.close()  # 总是关闭
    if exc_type is not None:
        logging.warning("File operation failed: %s", exc_value)
    return False  # 不抑制异常

什么时候该用类定义上下文管理器，而不是 `@contextlib.contextmanager`？

当需要在实例中持久保存状态（比如连接池计数器、嵌套锁标记、多次 with 块间共享的缓冲区），就必须用类。装饰器方式生成的上下文管理器每次调用都是全新函数作用域，无法跨 with 块维持状态。

Python Testing

Python 测试速查：运行 pytest、使用 mock/patch、参数化、fixtures、异步、覆盖率测试。

下载

典型使用场景：

数据库连接池：__enter__ 获取连接，__exit__ 归还连接，同时更新内部 self._in_use 计数
带重入保护的锁：用 self._lock_depth 记录嵌套层数，避免重复释放
需要延迟初始化的资源：第一次 __enter__ 才创建对象，后续复用

注意：@contextlib.contextmanager 写起来快，但它的生成器函数里不能安全地 raise 异常再 yield —— 这会导致 StopIteration 被误判为上下文退出。

如何测试自定义上下文管理器是否真正可靠？

不能只测“正常流程”，必须覆盖异常路径。重点验证三件事：资源是否总被释放、异常是否按预期传播、多次嵌套是否稳定。

实操建议：

用 pytest.raises() 配合 with 块，确认异常确实冒出（或被吞掉）
在 __exit__ 里手动 raise ValueError，看是否会引发 RuntimeError: generator didn't stop（仅限装饰器方式）
检查 __enter__ 返回值类型是否与 as 绑定目标一致（比如返回 self 却期望得到一个 file 对象）
对带状态的管理器，做连续两次 with 块，验证内部计数/标志是否正确重置