Python上下文管理器是一种用于管理资源生命周期的语法结构，核心作用是在代码块执行前后自动完成资源的初始化与清理工作，避免手动操作资源时出现的遗漏释放问题。无论是文件读写、网络连接还是数据库连接场景，合理使用上下文管理器都能让代码更简洁安全。
上下文管理器的基本工作原理
上下文管理器的核心是通过with语句触发，其底层依赖两个特殊方法：__enter__和__exit__。当执行with语句时，会先调用目标对象的__enter__方法完成资源初始化，然后将__enter__的返回值赋值给as后面的变量；当with代码块执行完毕，或者代码块中出现异常时，都会自动调用目标对象的__exit__方法完成资源清理。
我们常用的open函数返回的文件对象就是内置的上下文管理器，使用方式如下：
# 内置文件上下文管理器使用示例
with open("test.txt", "w", encoding="utf-8") as f:
f.write("这是一段测试内容")
# with代码块结束后，文件会自动关闭，无需手动调用f.close()
自定义上下文管理器的两种实现方式
方式一：基于类实现上下文管理器
只要一个类实现了__enter__和__exit__两个方法，就可以作为上下文管理器使用。__enter__方法没有额外参数，返回值会传递给as后的变量；__exit__方法接收三个参数，分别是异常类型、异常实例、异常回溯信息，当代码块无异常时这三个参数都为None。
下面是一个自定义的文件操作上下文管理器示例，额外增加了操作日志功能：
class FileManager:
def __init__(self, file_path, mode, encoding="utf-8"):
self.file_path = file_path
self.mode = mode
self.encoding = encoding
self.file_obj = None
def __enter__(self):
# 初始化资源，打开文件
print(f"开始打开文件：{self.file_path}")
self.file_obj = open(self.file_path, self.mode, encoding=self.encoding)
return self.file_obj
def __exit__(self, exc_type, exc_val, exc_tb):
# 清理资源，关闭文件
if self.file_obj:
self.file_obj.close()
print(f"文件已关闭：{self.file_path}")
# 如果返回True，会抑制代码块中抛出的异常，这里不抑制异常所以返回False
return False
# 使用自定义上下文管理器
with FileManager("demo.txt", "w") as f:
f.write("自定义上下文管理器测试")
方式二：使用contextlib模块装饰器实现
对于逻辑简单的上下文管理器，使用标准库的contextlib模块的@contextmanager装饰器可以更简洁地实现，不需要单独定义类。这种方式基于生成器实现，yield之前的代码相当于__enter__的逻辑，yield之后的代码相当于__exit__的逻辑，yield的返回值会传递给as后的变量。
下面是使用装饰器实现的数据库连接上下文管理器示例：
from contextlib import contextmanager
@contextmanager
def db_connection(db_name):
# 模拟数据库连接初始化
print(f"连接数据库：{db_name}")
conn = {"db_name": db_name, "status": "connected"}
try:
# yield之前的代码是__enter__逻辑，yield返回值是as后的变量
yield conn
finally:
# yield之后的代码是__exit__逻辑，无论是否异常都会执行
conn["status"] = "closed"
print(f"关闭数据库连接：{db_name}")
# 使用装饰器实现的上下文管理器
with db_connection("test_db") as conn:
print(f"当前数据库连接状态：{conn['status']}")
# 这里即使出现异常，连接也会正常关闭
上下文管理器的常见应用场景
- 文件与IO资源管理：除了内置的文件对象，自定义的网络连接、串口连接等资源都可以用上下文管理器管理，避免连接泄漏。
- 临时环境切换：比如临时修改系统环境变量、临时切换工作目录，退出
with代码块后自动恢复原有环境。 - 异常安全的事务操作：数据库事务场景中，
with代码块执行成功则提交事务，出现异常则自动回滚，保证数据一致性。 - 计时与性能统计：可以在
__enter__记录开始时间，__exit__计算耗时，方便统计代码块执行效率。
注意事项
在自定义__exit__方法时，如果代码块中可能出现异常，建议在__exit__中使用try...finally结构保证资源清理逻辑一定执行。如果__exit__返回True，则会吞掉with代码块中抛出的所有异常，除非明确需要抑制异常，否则建议返回False让异常正常抛出，方便问题排查。
另外，contextlib还提供了closing方法，可以将有close方法但没有实现上下文管理器协议的对象包装成上下文管理器，比如某些第三方库返回的资源对象，使用方式如下：
from contextlib import closing
class Resource:
def close(self):
print("资源已释放")
# 将普通有close方法的对象转为上下文管理器
with closing(Resource()) as res:
print("使用资源中")
Python上下文管理器with语句资源管理__enter___修改时间：2026-06-24 13:36:32