Python中的上下文管理器是一种用于管理资源的特殊对象，它能在代码块执行前后自动完成资源的初始化与清理工作，最常见的使用场景就是配合with语句操作文件、数据库连接等资源。上下文管理器的核心实现依赖于两个特殊方法：__enter__和__exit__，只要一个类定义了这两个方法，它的实例就可以作为上下文管理器使用。
自定义类实现上下文管理器
要实现自定义上下文管理器，首先需要在类中定义__enter__方法和__exit__方法，下面通过一个简单的文件操作示例来演示具体实现。
__enter__方法的作用
__enter__方法会在with语句块执行前被调用，它的返回值会赋值给with语句中as后面的变量，通常在这个方法里完成资源的初始化操作，比如打开文件、建立数据库连接等。
__exit__方法的作用
__exit__方法会在with语句块执行结束后被调用，无论语句块中是否出现异常，这个方法都会执行，通常在这个方法里完成资源的清理操作，比如关闭文件、断开数据库连接等。__exit__方法接收三个参数，分别是异常类型、异常实例和异常回溯信息，如果语句块中没有出现异常，这三个参数都为None。
完整实现示例
下面是一个自定义的文件上下文管理器示例，实现了自动打开和关闭文件的功能：
class FileContextManager:
def __init__(self, file_path, mode):
# 初始化文件路径和打开模式
self.file_path = file_path
self.mode = mode
self.file = None
def __enter__(self):
# 打开文件，返回文件对象
self.file = open(self.file_path, self.mode, encoding='utf-8')
return self.file
def __exit__(self, exc_type, exc_val, exc_tb):
# 关闭文件，释放资源
if self.file:
self.file.close()
# 如果返回True，异常会被抑制，不会向外抛出；返回False则异常会继续向外传递
# 这里返回False，让异常正常抛出
return False
# 使用自定义的上下文管理器
with FileContextManager('test.txt', 'w') as f:
f.write('这是通过自定义上下文管理器写入的内容')
__exit__方法的异常处理逻辑
__exit__方法的返回值决定了with语句块中出现的异常是否会被向外抛出。如果__exit__返回True，那么语句块中产生的异常会被抑制，不会传递到上层代码；如果返回False，异常会继续向外抛出。我们可以通过这个特性在__exit__中处理特定的异常，比如忽略某些不影响主流程的异常。
下面的示例演示了如何在__exit__中处理特定的异常：
class SafeDivisionContext:
def __enter__(self):
return self
def __exit__(self, exc_type, exc_val, exc_tb):
# 如果是除零异常，打印提示信息并抑制异常
if exc_type == ZeroDivisionError:
print('出现除零异常，已自动处理')
return True
# 其他异常正常抛出
return False
def divide(self, a, b):
return a / b
# 测试异常处理
with SafeDivisionContext() as sd:
result = sd.divide(10, 0)
print(result) # 这里不会执行，因为异常被抑制了，但是不会报错
print('程序继续执行')
使用contextmanager装饰器简化实现
除了自定义类的方式，Python标准库的contextlib模块提供了contextmanager装饰器，可以用生成器的方式更简洁地实现上下文管理器，不需要手动定义__enter__和__exit__方法。
使用contextmanager装饰器时，生成器函数中yield之前的代码相当于__enter__方法的逻辑，yield的返回值就是as后面的变量值，yield之后的代码相当于__exit__方法的逻辑。
下面是使用contextmanager实现文件上下文管理器的示例：
from contextlib import contextmanager
@contextmanager
def file_context_manager(file_path, mode):
# 相当于__enter__的逻辑，打开文件
file = open(file_path, mode, encoding='utf-8')
try:
# yield返回文件对象，赋值给as后面的变量
yield file
finally:
# 相当于__exit__的逻辑，关闭文件
file.close()
# 使用装饰器实现的上下文管理器
with file_context_manager('test2.txt', 'w') as f:
f.write('这是通过contextmanager装饰器实现的上下文管理器写入的内容')
两种实现方式的对比
自定义类实现和使用contextmanager装饰器实现各有适用场景，下面是两者的对比：
| 实现方式 | 优点 | 缺点 | 适用场景 |
|---|---|---|---|
| 自定义类实现 | 逻辑清晰，可扩展性强，能更灵活地控制__enter__和__exit__的行为 | 代码量相对较多，需要实现两个特殊方法 | 复杂的上下文管理逻辑，需要自定义更多方法或属性的场景 |
| contextmanager装饰器 | 代码简洁，用生成器的方式快速实现，不需要写类结构 | 逻辑相对隐蔽，复杂场景下可读性不如类实现 | 简单的上下文管理逻辑，只需要简单的资源获取和释放的场景 |
常见使用注意事项
- __exit__方法一定要确保资源能被正确释放，最好把清理逻辑放在try...finally结构中，避免因为异常导致清理代码不执行。
- 如果__exit__方法返回True抑制了异常，上层代码不会感知到with语句块中出现的异常，需要根据实际需求决定是否抑制异常，避免隐藏重要的错误。
- 使用contextmanager装饰器时，生成器函数中一定要用try...finally结构包裹yield语句，确保无论是否出现异常，清理逻辑都会执行。
- 上下文管理器不仅可以管理文件、数据库连接这类外部资源，也可以用于实现代码执行计时、事务回滚等逻辑，只要需要在代码块前后执行固定操作的场景都可以考虑使用。