Python上下文管理器与with用法详解

Python的with语句背后是强大而优雅的上下文管理器机制，它通过自动调用__enter__和__exit__方法，确保资源（如文件、网络连接、锁等）在进入和退出代码块时得到安全、可靠的初始化与清理——无论是否发生异常，都能避免资源泄漏；本文深入剖析其工作原理，详解异常抑制逻辑，并手把手带你实现自定义上下文管理器，同时介绍contextlib模块提供的@contextmanager等高效写法，助你写出更健壮、简洁、Pythonic的资源管理代码。

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
<pre class="brush:php;toolbar:false"><code>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，确保异常正常抛出</code>

使用方式和内置 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
<p>@contextmanager
def db_connection(db_url):
conn = connect_to_db(db_url)  # 假设这是连接函数
try:
yield conn
finally:
conn.close()  # 确保关闭，哪怕执行中抛异常
</p>

使用起来同样清晰：

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 ExitStack
<p>filenames = ['a.txt', 'b.txt', 'c.txt']
with ExitStack() as stack:
files = [stack.enter_context(open(name)) for name in filenames]</p><h1>所有文件都会被自动关闭，无论中间是否出错</h1><p></p>

掌握上下文管理器不是为了炫技，而是写出更健壮、更易维护的代码。它把“一定会做的事”从手动重复中解放出来，交给语言机制保障。

到这里，我们也就讲完了《Python上下文管理器与with用法详解》的内容了。个人认为，基础知识的学习和巩固，是为了更好的将其运用到项目中，欢迎关注golang学习网公众号，带你了解更多关于的知识点！