上下文管理器是Python中用于管理资源生命周期的常用机制，它可以在代码块执行前后自动执行特定的操作，最常见的应用场景就是文件操作，无需手动调用close方法就能保证文件正确关闭。
基于类的上下文管理器实现
要实现基于类的上下文管理器，需要在类中定义__enter__和__exit__两个特殊方法，这两个方法是上下文管理器的核心。
方法作用说明
__enter__方法：在进入with语句块时自动调用，它的返回值会赋值给with语句中as后面的变量__exit__方法：在退出with语句块时自动调用，无论语句块中是否出现异常都会执行，用于释放资源、处理异常等
完整实现示例
下面是一个模拟文件操作的上下文管理器实现，会在进入时打开文件，退出时自动关闭文件：
class FileManager:
def __init__(self, file_path, mode):
self.file_path = file_path
self.mode = mode
self.file = None
def __enter__(self):
# 进入with语句时执行，打开文件并返回文件对象
self.file = open(self.file_path, self.mode, encoding='utf-8')
return self.file
def __exit__(self, exc_type, exc_val, exc_tb):
# 退出with语句时执行，关闭文件
if self.file:
self.file.close()
# 如果返回True，会忽略语句块中抛出的异常；返回False则异常会继续向外抛出
# 这里返回False，让异常正常传播
return False
# 使用自定义上下文管理器
with FileManager('test.txt', 'w') as f:
f.write('这是通过自定义上下文管理器写入的内容')
__exit__方法的参数说明
__exit__方法接收三个参数，分别是exc_type（异常类型）、exc_val（异常值）、exc_tb（异常追踪信息），如果with语句块中没有抛出异常，这三个参数的值都为None。
基于contextlib的生成器实现
除了基于类的实现，Python标准库的contextlib模块提供了contextmanager装饰器，可以用生成器快速实现上下文管理器，代码更加简洁。
实现原理
被contextmanager装饰的生成器函数中，yield之前的代码相当于__enter__方法的内容，yield的返回值会作为as后面的变量值，yield之后的代码相当于__exit__方法的内容。
完整实现示例
同样实现上面的文件管理功能，用contextlib的方式代码如下：
from contextlib import contextmanager
@contextmanager
def file_manager(file_path, mode):
# 相当于__enter__方法的内容
file = open(file_path, mode, encoding='utf-8')
try:
yield file
finally:
# 相当于__exit__方法的内容，无论是否异常都会执行
file.close()
# 使用生成器实现的上下文管理器
with file_manager('test2.txt', 'w') as f:
f.write('这是通过contextlib实现的上下文管理器写入的内容')
两种实现方式的对比
两种实现方式各有适用场景，具体对比如下：
| 实现方式 | 适用场景 | 优点 | 缺点 |
|---|---|---|---|
| 基于类的实现 | 需要更复杂的资源管理逻辑，或者需要保存更多状态信息 | 逻辑清晰，状态管理方便，可扩展性强 | 代码相对繁琐，需要实现两个特殊方法 |
| 基于contextlib的实现 | 简单的资源管理场景，逻辑不复杂 | 代码简洁，实现快速，无需定义类 | 复杂逻辑下可读性差，状态管理不方便 |
注意事项
- 上下文管理器的
__exit__方法或者生成器中yield之后的代码，一定要确保资源能被正确释放，最好使用try...finally结构 - 如果需要在上下文管理器中处理异常，可以在
__exit__方法中判断exc_type参数，或者生成器的yield部分捕获异常 - 不要滥用上下文管理器，只有在需要自动管理资源生命周期的场景下使用才更有意义