Python上下文管理器实战_with语句管理资源解析【指导】-Python教程-PHP中文网

Python上下文管理器实战_with语句管理资源解析【指导】

舞夢輝影

发布： 2025-12-23 22:55:19

原创

327人浏览过

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

python上下文管理器实战_with语句管理资源解析【指导】

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
<pre class="brush:php;toolbar:false;">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: —— 连接在退出时必然释放。

AutoGLM沉思

智谱AI推出的具备深度研究和自主执行能力的AI智能体

239

查看详情

用 contextlib 简化轻量级上下文管理

不需要写完整类时，@contextmanager 装饰器更简洁。它把生成器函数转为上下文管理器：

yield 之前的部分相当于 __enter__
yield 的值作为 as 后的变量
yield 之后（包括异常处理块）相当于 __exit__

示例：临时切换工作目录

from contextlib import contextmanager
import os
<p>@contextmanager
def cd(path):
old = os.getcwd()
os.chdir(path)
try:
yield
finally:
os.chdir(old)</p><h1>使用</h1><p>with cd("/tmp"):
print(os.getcwd())  # 输出 /tmp
print(os.getcwd())  # 自动切回原路径

登录后复制