在 Python 编程中，文件操作、数据库连接、网络请求等场景都涉及资源管理。手动开启和关闭资源不仅繁琐，还容易因代码异常导致资源泄漏（比如忘记关闭文件句柄）。而上下文管理器（Context Manager） 正是为解决这一问题而生，它能自动完成资源的分配与释放，让代码更简洁、更健壮。
一、为什么需要上下文管理器？
先看一个常见的 “反面案例”—— 手动操作文件：
# 手动管理文件资源，存在风险
file = None
try:
file = open("data.txt", "r")
content = file.read()
print(content)
except FileNotFoundError:
print("文件不存在")
finally:
# 必须手动关闭文件，否则资源泄漏
if file:
file.close()
这段代码的问题很明显：
（1）需用try-finally确保资源关闭，代码冗余；
（2）若忘记写close()，或finally块中出现异常，仍会导致资源泄漏。
而用上下文管理器的with语句重写后，代码瞬间简洁：
# 上下文管理器自动管理资源
try:
with open("data.txt", "r") as file:
content = file.read()
print(content)
except FileNotFoundError:
print("文件不存在")
无需手动调用close()——with语句会在代码块执行完毕后（无论正常结束还是抛出异常），自动触发资源释放。这就是上下文管理器的核心价值：将 “资源管理” 与 “业务逻辑” 解耦。
二、上下文管理器的核心原理
上下文管理器的本质是实现了两个特殊方法的对象：
（1）__enter__(self)：进入with代码块时调用，返回的对象会赋值给as后的变量（可选）；
（2）__exit__(self, exc_type, exc_val, exc_tb)：离开with代码块时调用，负责释放资源（即使代码块抛出异常）。
执行流程拆解
（1）执行with后的表达式（如open("data.txt")），创建上下文管理器对象；
（2）调用管理器的__enter__()方法，获取资源并返回；
（3）执行with代码块内的业务逻辑（如读取文件）；
（4）无论代码块是否抛出异常，都会调用管理器的__exit__()方法释放资源；
（5）若代码块正常执行：exc_type, exc_val, exc_tb均为None；
（6）若代码块抛出异常：这三个参数会携带异常信息，__exit__可选择处理异常（返回True表示已处理，不向外传播；返回False则继续传播）。
三、实现上下文管理器的两种方式
Python 提供了两种实现上下文管理器的途径：类实现和装饰器实现，分别适用于复杂场景和简单场景。
方式 1：基于类实现（灵活可控）
若需自定义资源的分配与释放逻辑（如管理数据库连接），可通过类实现__enter__和__exit__。
示例：自定义数据库连接管理器
class DBConnection:
def __init__(self, host, port):
self.host = host
self.port = port
self.conn = None
# 进入with块：建立数据库连接
def __enter__(self):
print(f"连接数据库 {self.host}:{self.port}")
self.conn = "模拟数据库连接对象" # 实际场景替换为真实连接
return self.conn # 赋值给as后的变量
# 离开with块：关闭数据库连接
def __exit__(self, exc_type, exc_val, exc_tb):
print(f"关闭数据库连接 {self.host}:{self.port}")
if self.conn:
self.conn = None # 实际场景替换为conn.close()
# 处理异常（可选）：若返回True，异常不向外传播
if exc_type:
print(f"数据库操作异常：{exc_val}")
return True # 表示已处理异常
# 使用自定义上下文管理器
with DBConnection("localhost", 3306) as conn:
print(f"使用连接执行操作：{conn}")
# 代码块结束后，自动调用__exit__关闭连接
执行结果：
连接数据库 localhost:3306
使用连接执行操作：模拟数据库连接对象
关闭数据库连接 localhost:3306
方式 2：基于contextlib装饰器（简洁高效）
对于简单场景（无需复杂的类逻辑），Python 标准库contextlib的contextmanager装饰器可将生成器函数转为上下文管理器，大幅简化代码。
核心逻辑：
（1）生成器中yield之前的代码，对应__enter__的功能（分配资源）；
（2）yield返回的值，对应__enter__的返回值（赋值给as变量）；
（3）yield之后的代码，对应__exit__的功能（释放资源）。
示例 1：简化文件操作
from contextlib import contextmanager
@contextmanager
def open_file(file_path, mode):
# 1. 分配资源：对应__enter__
file = open(file_path, mode)
try:
yield file # 返回资源给as变量，暂停执行
finally:
# 2. 释放资源：对应__exit__（无论是否异常都会执行）
file.close()
print("文件已关闭")
# 使用装饰器实现的上下文管理器
with open_file("data.txt", "w") as f:
f.write("Hello, Context Manager!")
示例 2：临时切换目录
有时需要临时切换工作目录执行操作，之后恢复原目录，用contextmanager可轻松实现：
import os
from contextlib import contextmanager
@contextmanager
def change_dir(temp_dir):
# 1. 保存原目录（分配“原目录”资源）
original_dir = os.getcwd()
try:
# 切换到临时目录
os.chdir(temp_dir)
yield # 无需返回值，as可省略
finally:
# 2. 恢复原目录（释放资源）
os.chdir(original_dir)
print(f"已恢复到原目录：{original_dir}")
# 使用：临时切换到"/tmp"目录
with change_dir("/tmp"):
print(f"当前临时目录：{os.getcwd()}") # 输出 /tmp
# 代码块结束后，自动恢复原目录
四、常见内置上下文管理器
Python 标准库中已内置多个实用的上下文管理器，无需重复造轮子：
| 上下文管理器 | 用途 | 示例代码 |
|---|---|---|
open() |
文件读写，自动关闭文件句柄 | with open("a.txt") as f: f.read() |
threading.Lock() |
线程锁，自动释放锁 | with lock: 执行线程安全操作 |
decimal.localcontext() |
临时修改小数精度 | with localcontext() as ctx: ctx.prec = 2 |
sqlite3.connect() |
SQLite 数据库连接，自动关闭 | with sqlite3.connect("db.db") as conn: ... |
五、总结
上下文管理器是 Python 的 “优雅语法糖”，核心价值在于自动管理资源，避免泄漏。使用时需记住：
（1）简单场景用contextlib.contextmanager装饰器，代码更简洁；
（2）复杂场景（需自定义异常处理、多资源管理）用类实现__enter__/__exit__；
（3）优先使用 Python 内置的上下文管理器（如open()），避免重复开发。