Python上下文管理器怎么写_with语句实战说明【指导】-Python教程-PHP中文网

Python上下文管理器怎么写_with语句实战说明【指导】

舞夢輝影

发布： 2025-12-18 21:59:02

原创

280人浏览过

Python的with语句通过上下文管理器实现资源自动清理，核心是定义__enter__和__exit__方法；可手动实现类或用@contextmanager装饰器简化；内置如open()、threading.Lock()等均支持，适用于“获取-使用-释放”场景。

python上下文管理器怎么写_with语句实战说明【指导】

Python的with语句靠上下文管理器实现资源自动清理，核心是定义__enter__和__exit__两个方法。

自定义上下文管理器的写法

手动实现一个类，必须包含这两个特殊方法：

__enter__：进入with块时执行，返回值绑定到as后的变量
__exit__(exc_type, exc_value, traceback)：退出with块时执行，无论是否发生异常都会调用；三个参数用于判断异常类型，正常退出时全为None

例如，写一个简单的文件操作管理器：

class FileHandler:
    def __init__(self, filename, mode):
        self.filename = filename
        self.mode = mode
        self.file = None
<pre class='brush:python;toolbar:false;'>def __enter__(self):
    self.file = open(self.filename, self.mode)
    return self.file

def __exit__(self, exc_type, exc_value, traceback):
    if self.file:
        self.file.close()
    # 返回 False 表示不压制异常，让异常继续向上抛出
    return False

登录后复制

使用方式：

立即学习“Python免费学习笔记（深入）”；

with FileHandler('test.txt', 'w') as f:
    f.write('hello')
# 离开 with 块后自动关闭文件

登录后复制

用 contextlib.contextmanager 装饰器简化写法

不用写完整类，用生成器函数配合@contextmanager更简洁：

函数中yield之前的代码相当于__enter__
yield返回的值就是as接收的对象
yield之后的代码相当于__exit__，会确保执行（即使出错）

例子：带日志的数据库连接管理器

ProcessOn

免费在线流程图思维导图，专业强大的作图工具，支持多人实时在线协作

925

查看详情

from contextlib import contextmanager
<p>@contextmanager
def db_connection(db_url):
conn = None
try:
print("连接数据库...")
conn = connect_to_db(db_url)  # 假设这个函数存在
yield conn
finally:
if conn:
print("关闭数据库连接...")
conn.close()</p>

登录后复制

使用：

with db_connection('sqlite:///app.db') as db:
    db.execute("SELECT * FROM users")
# 退出时自动关闭连接，出错也不漏关

登录后复制

常见内置上下文管理器和典型场景

Python标准库中很多对象原生支持with：

open()：文件读写，自动关闭
threading.Lock()：加锁/解锁，避免忘记释放
tempfile.TemporaryDirectory() 和 TemporaryFile()：临时资源自动清理
unittest.mock.patch()：测试中临时替换对象，退出自动还原

关键原则：只要涉及“获取-使用-释放”三步操作，且容易遗漏最后一步（比如忘记close()、unlock()、rollback()），就适合用上下文管理器封装。

错误处理与异常传播要点

__exit__方法的返回值决定异常是否被压制：

返回True：吞掉异常，后续代码继续执行
返回False或None（默认）：异常照常抛出

如果想在清理时记录异常但不压制，可以这样写：

def __exit__(self, exc_type, exc_value, traceback):
    if exc_type is not None:
        print(f"捕获到异常: {exc_type.__name__}: {exc_value}")
    self.cleanup()
    # 不 return True，让异常继续传播

登录后复制

注意：__exit__本身不应抛出新异常，否则会覆盖原始异常。

以上就是Python上下文管理器怎么写_with语句实战说明【指导】的详细内容，更多请关注php中文网其它相关文章！