Python的with语句基于上下文管理器,核心是__enter__和__exit__方法:前者在进入时执行并返回对象,后者在退出时无论异常与否都执行清理;__exit__返回True可抑制异常,否则异常继续传播。

Python 的 with 语句背后是上下文管理器(Context Manager),它让资源管理更安全、简洁。核心在于自动执行“进入”和“退出”逻辑,比如打开文件后必定关闭,哪怕中间出错也不遗漏。
上下文管理器的本质:__enter__ 和 __exit__
一个对象只要实现了 __enter__ 和 __exit__ 这两个特殊方法,就是上下文管理器。__enter__ 在 with 开始时调用,通常返回要操作的对象(如文件句柄);__exit__ 在 with 块结束时调用,无论是否发生异常,都负责清理工作。
-
__exit__(self, exc_type, exc_value, traceback)的三个参数用于判断异常:如果没异常,三者都是None;有异常时分别对应异常类型、值和回溯信息 - 若
__exit__返回True,表示已处理异常,不会向上抛出;返回False或None则继续传播异常 - 常见写法中,
__exit__通常只做清理(如close()),不主动吞异常,除非你明确想屏蔽某类错误
手动实现一个文件上下文管理器
理解原理最直接的方式是自己写一个。下面是一个简化版的文件管理器,模拟内置 open() 的行为:
class MyOpen:
def __init__(self, name, mode='r'):
self.name = name
self.mode = mode
self.file = None
def __enter__(self):
self.file = open(self.name, self.mode)
return self.file
def __exit__(self, exc_type, exc_value, traceback):
if self.file:
self.file.close()
# 不返回 True,确保异常正常抛出
立即学习“Python免费学习笔记(深入)”;
使用方式和内置 open 一样:
with MyOpen('data.txt', 'w') as f:
f.write('Hello, context!')
# 退出 with 后,文件已自动关闭
用 contextlib 简化上下文管理器编写
手写类稍显繁琐,contextlib 提供了更轻量的方式。最常用的是 @contextmanager 装饰器,用生成器函数定义上下文逻辑:
- 函数中
yield之前的部分相当于__enter__ -
yield返回的值就是as后绑定的对象 - yield 之后的部分相当于
__exit__,会确保执行(即使前面出错)
例如,为数据库连接写一个上下文管理器:
from contextlib import contextmanager@contextmanager def db_connection(db_url): conn = connect_to_db(db_url) # 假设这是连接函数 try: yield conn finally: conn.close() # 确保关闭,哪怕执行中抛异常
使用起来同样清晰:
with db_connection('sqlite:///app.db') as conn:
conn.execute('INSERT INTO logs VALUES (?)', ['start'])
多个上下文管理器的嵌套与合并
Python 3.1+ 支持在单个 with 语句中管理多个资源,用逗号分隔:
with open('input.txt') as fin, open('output.txt', 'w') as fout:
fout.write(fin.read().upper())
这等价于嵌套写法,但更扁平、可读性更好。注意:它们按顺序进入,逆序退出(类似栈)。如果某个 __enter__ 失败,前面已成功进入的会按相反顺序调用 __exit__ 清理。
对于动态数量的上下文管理器(比如不确定要打开几个文件),可用 contextlib.ExitStack:
from contextlib import ExitStackfilenames = ['a.txt', 'b.txt', 'c.txt'] with ExitStack() as stack: files = [stack.enter_context(open(name)) for name in filenames]
所有文件都会被自动关闭,无论中间是否出错
立即学习“Python免费学习笔记(深入)”;
掌握上下文管理器不是为了炫技,而是写出更健壮、更易维护的代码。它把“一定会做的事”从手动重复中解放出来,交给语言机制保障。










