Python上下文管理器怎么写_with语句实战说明【指导】
Python的with语句通过上下文管理器实现资源自动清理，核心是定义__enter__和__exit__方法；可手动实现类或用@contextmanager装饰器简化；内置如open()、threading.Lock()等均支持，适用于“获取-使用-释放”场景。
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
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使用方式：
with FileHandler('test.txt', 'w') as f:
f.write('hello')
# 离开 with 块后自动关闭文件
用 contextlib.contextmanager 装饰器简化写法
不用写完整类，用生成器函数配合@contextmanager更简洁：
- 函数中
yield之前的代码相当于__enter__ -
yield返回的值就是as接收的对象 -
yield之后的代码相当于__exit__，会确保执行（即使出错）
例子：带日志的数据库连接管理器
from contextlib import contextmanager@contextmanager def db_connection(db_url): conn = None try: print("连接数据库...") conn = connect_to_db(db_url) # 假设这个函数存在 yield conn finally: if conn: print("关闭数据库连接...") conn.close()
使用：
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__本身不应抛出新异常，否则会覆盖原始异常。
