with语句是Python中确保资源安全释放的核心机制,依赖__enter__和__exit__方法实现,无论正常执行或异常都能保证清理,支持文件操作、自定义类、contextlib装饰器及多管理器并行。

Python 中的 with 语句不是语法糖,而是资源安全释放的可靠机制。它背后依赖的是上下文管理器协议(__enter__ 和 __exit__ 方法),核心价值在于:**无论代码是否异常,都能确保资源被正确清理**。
文件操作:最典型的 with 应用场景
手动调用 open() 后忘记 close(),或在读取过程中出错导致文件句柄泄露,是常见隐患。用 with 可彻底规避:
-
with open("data.txt", "r") as f:执行时自动触发f.__enter__()(返回文件对象) - 缩进内代码执行完毕(无论成功或抛异常),都会调用
f.__exit__(exc_type, exc_val, exc_tb) -
__exit__返回True可抑制异常;通常返回None或False,让异常正常传播
自定义类实现上下文管理器
只要类实现了 __enter__ 和 __exit__,就能用于 with。例如管理数据库连接:
class DBConnection:
def __init__(self, url):
self.url = url
self.conn = None
def __enter__(self):
self.conn = connect(self.url) # 建立连接
return self.conn
def __exit__(self, exc_type, exc_val, exc_tb):
if self.conn:
self.conn.close() # 确保关闭,即使发生异常
return False # 不抑制异常
立即学习“Python免费学习笔记(深入)”;
使用:with DBConnection("sqlite:///app.db") as db: —— 连接在退出时必然释放。
用 contextlib 简化轻量级上下文管理
不需要写完整类时,@contextmanager 装饰器更简洁。它把生成器函数转为上下文管理器:
- yield 之前的部分相当于
__enter__ - yield 的值作为
as后的变量 - yield 之后(包括异常处理块)相当于
__exit__
示例:临时切换工作目录
from contextlib import contextmanager import os@contextmanager def cd(path): old = os.getcwd() os.chdir(path) try: yield finally: os.chdir(old)
使用
with cd("/tmp"): print(os.getcwd()) # 输出 /tmp print(os.getcwd()) # 自动切回原路径
多个上下文管理器嵌套与并行写法
Python 3.1+ 支持单行写多个管理器,等价于嵌套但更清晰:
-
with A() as a, B() as b:等价于with A() as a: with B() as b: - 所有
__enter__按顺序执行;任一失败则已进入的会按逆序调用__exit__ - 适合同时打开多个文件、组合锁与连接等场景
注意:逗号分隔写法要求每个表达式都返回有效的上下文管理器,不支持逻辑运算符混用。










