with语句是Python中确保资源安全释放的核心机制,依赖__enter__和__exit__方法实现,无论正常执行或异常都能保证清理,支持文件操作、自定义类、contextlib装饰器及多管理器并行。
Python 中的 with 语句不是语法糖,而是资源安全释放的可靠机制。它背后依赖的是上下文管理器协议(__enter__ 和 __exit__ 方法),核心价值在于:**无论代码是否异常,都能确保资源被正确清理**。
手动调用 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 # 不抑制异常
使用:with DBConnection("sqlite:///app.db") as db: —— 连接在退出时必然释放。
用 contextlib 简化轻量级上下文管理
不需要写完整类时,@contextmanager 装饰器更简洁。它把生成器函数转为上下文管理器:
__enter__
as 后的变量__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__
注意:逗号分隔写法要求每个表达式都返回有效的上下文管理器,不支持逻辑运算符混用。