Python基础教程（六十七）内建模块之contextlib：Python上下文管理神器：contextlib深度剖析与实战示例

最新推荐文章于 2025-09-26 16:38:16 发布

原创最新推荐文章于 2025-09-26 16:38:16 发布 · 481 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#python #数据库 #oracle

Python基础教程专栏收录该内容

99 篇文章

订阅专栏

一、contextlib核心机制解析

1. `@contextmanager`装饰器

Python的with语句依赖上下文管理器（含__enter__和__exit__方法）。contextlib的@contextmanager允许用生成器函数快速构建上下文管理器：

yield前代码 → 等效__enter__，初始化资源
yield后代码 → 等效__exit__，处理清理和异常

from contextlib import contextmanager

@contextmanager
def timer():
    start = time.time()
    try:
        yield  # 在此执行with语句块内的代码
    finally:
        print(f"耗时: {time.time() - start:.2f}s")

# 使用示例
with timer():
    time.sleep(1.5)  # 输出: 耗时: 1.50s

2. 异常处理逻辑

若with块内发生异常：

异常在yield处抛出
生成器内的finally或except捕获并处理
可通过yield返回值传递上下文资源（如文件对象）

二、4大实战工具详解与示例

1. `closing`：自动关闭资源

处理需手动调用close()的对象（如网络连接）：

from contextlib import closing
from urllib.request import urlopen

with closing(urlopen('https://python.org')) as page:
    content = page.read()
    print(f"页面长度: {len(content)}")  # 自动调用page.close()

2. `suppress`：忽略指定异常

静默处理特定异常，避免冗余try/except：

from contextlib import suppress

with suppress(FileNotFoundError):
    os.remove('temp.txt')  # 文件不存在时不报错

3. `redirect_stdout`：重定向输出

捕获或重定向print输出：

from contextlib import redirect_stdout
import io

buf = io.StringIO()
with redirect_stdout(buf):  # 将输出重定向到内存缓冲区
    print("Hello, contextlib!")
output = buf.getvalue()  # 输出: "Hello, contextlib!\n"

4. 嵌套管理器：简化多重资源管理

组合多个上下文管理器，避免深层嵌套：

from contextlib import ExitStack

with ExitStack() as stack:
    file1 = stack.enter_context(open('a.txt'))  # 自动管理多个资源
    file2 = stack.enter_context(open('b.txt'))
    # 退出时自动关闭所有文件

三、高级技巧：状态回滚与数据库事务

1. 临时状态修改与回滚

@contextmanager
def temporary_state(obj, **kwargs):
    original = {attr: getattr(obj, attr) for attr in kwargs}
    try:
        for attr, value in kwargs.items():
            setattr(obj, attr, value)
        yield
    finally:
        for attr, value in original.items():  # 回滚原始状态
            setattr(obj, attr, value)

class Config:
    mode = "default"

config = Config()
with temporary_state(config, mode="debug"):
    print(config.mode)  # 输出: debug
print(config.mode)  # 输出: default (自动回滚)

2. 数据库事务安全提交

@contextmanager
def db_transaction(connection):
    try:
        yield connection.cursor()
        connection.commit()  # 无异常则提交
    except:
        connection.rollback()  # 异常时回滚
        raise

# 使用示例
with db_transaction(db_conn) as cursor:
    cursor.execute("UPDATE users SET active=1 WHERE id=101")