Python上下文管理器教程_with语句原理与实践

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,表示已处理异常,不会向上抛出;返回 FalseNone 则继续传播异常
  • 常见写法中,__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,确保异常正常抛出

使用方式和内置 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 ExitStack

filenames = ['a.txt', 'b.txt', 'c.txt'] with ExitStack() as stack: files = [stack.enter_context(open(name)) for name in filenames]

所有文件都会被自动关闭,无论中间是否出错

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