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
<pre class='brush:python;toolbar:false;'>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

使用方式：

立即学习“Python免费学习笔记（深入）”；

with FileHandler('test.txt', 'w') as f:
    f.write('hello')
# 离开 with 块后自动关闭文件

用 contextlib.contextmanager 装饰器简化写法

不用写完整类，用生成器函数配合@contextmanager更简洁：

• 函数中yield之前的代码相当于__enter__
• yield返回的值就是as接收的对象
• yield之后的代码相当于__exit__，会确保执行（即使出错）

例子：带日志的数据库连接管理器

标签： python app 标准库