在 Python 中，资源管理（文件、数据库连接、网络套接字、锁等）的核心痛点是：无论操作是否抛出异常，都必须保证资源被正确释放。传统的try-finally写法繁琐且易出错，而上下文管理器（Context Manager）配合with语句，能以极简、优雅的方式解决这个问题 —— 它让资源的 “获取 - 使用 - 释放” 流程自动化，代码可读性和健壮性大幅提升。
一、什么是上下文管理器？
上下文管理器是实现了「上下文管理协议」的对象，核心是两个特殊方法：
__enter__(self)：进入上下文时执行，负责获取资源，返回值会被with ... as var中的var接收；__exit__(self, exc_type, exc_val, exc_tb)：退出上下文时执行，负责释放资源，无论代码块是否抛出异常都会执行。
with语句是上下文管理器的 “语法糖”，其执行流程：
- 调用
__enter__获取资源； - 执行
with代码块内的逻辑； - 调用
__exit__释放资源（即使代码块抛异常）。
二、基础用法：内置上下文管理器
Python 内置了大量现成的上下文管理器，无需自定义即可直接使用，这是最常用的场景。
1. 文件操作（最经典场景）
传统try-finally写法（繁琐且易漏close()）：
python
# 传统写法：手动控制文件关闭 f = None try: f = open("test.txt", "r", encoding="utf-8") content = f.read() print(content) finally: if f: f.close() # 必须手动关闭，否则资源泄露上下文管理器写法（优雅且安全）：
python
# with语句自动调用__enter__（打开文件）和__exit__（关闭文件） with open("test.txt", "r", encoding="utf-8") as f: content = f.read() print(content) # 代码块结束后，文件已自动关闭，无需手动操作2. 线程锁管理
多线程场景中，锁的 “获取 - 释放” 是必做操作，上下文管理器可避免忘记释放锁导致的死锁：
python
import threading lock = threading.Lock() # 传统写法：手动acquire/release lock.acquire() try: print("执行线程安全操作") finally: lock.release() # 上下文管理器写法 with lock: print("执行线程安全操作") # 自动acquire，代码块结束自动release3. 临时文件 / 目录
tempfile模块的上下文管理器可自动创建临时文件 / 目录，使用后自动删除：
python
import tempfile # 临时文件：使用后自动删除 with tempfile.TemporaryFile(mode="w+", encoding="utf-8") as temp_f: temp_f.write("临时内容") temp_f.seek(0) print(temp_f.read()) # 输出：临时内容 # 退出后临时文件已被删除 # 临时目录 with tempfile.TemporaryDirectory() as temp_dir: print(f"临时目录路径：{temp_dir}") # 退出后临时目录已被删除三、自定义上下文管理器
当内置上下文管理器满足不了需求时（如自定义数据库连接、定时器等），可通过两种方式实现自定义：
方式 1：类实现「上下文管理协议」
需手动定义__enter__和__exit__方法，适合复杂场景（需精细控制资源）。
示例：自定义数据库连接管理器
python
import pymysql class MySQLConnection: def __init__(self, host, user, password, db): self.host = host self.user = user self.password = password self.db = db self.conn = None self.cursor = None # 进入上下文：建立连接，获取游标 def __enter__(self): self.conn = pymysql.connect( host=self.host, user=self.user, password=self.password, db=self.db, charset="utf8mb4" ) self.cursor = self.conn.cursor() return self.cursor # 被with ... as var接收 # 退出上下文：释放游标、关闭连接，处理异常 def __exit__(self, exc_type, exc_val, exc_tb): # exc_type: 异常类型（无异常则为None） # exc_val: 异常实例（无异常则为None） # exc_tb: 异常追踪栈（无异常则为None） # 1. 释放游标 if self.cursor: self.cursor.close() # 2. 关闭连接 if self.conn: if exc_type: # 有异常则回滚 self.conn.rollback() else: # 无异常则提交 self.conn.commit() self.conn.close() # 3. 异常处理：返回True则抑制异常，返回False则抛出 # 此处不抑制，让上层处理 return False # 使用自定义上下文管理器 try: with MySQLConnection("localhost", "root", "123456", "test_db") as cursor: cursor.execute("SELECT * FROM user LIMIT 10") result = cursor.fetchall() print(result) except Exception as e: print(f"数据库操作异常：{e}")方式 2：contextlib.contextmanager装饰器（极简写法）
通过生成器函数实现，无需定义类和魔法方法，适合简单场景，是 Pythonic 的推荐写法。
核心规则：
- 生成器中
yield之前的代码 → 对应__enter__（获取资源）； yield的返回值 → 被with ... as var接收；yield之后的代码 → 对应__exit__（释放资源）。
示例 1：极简定时器上下文管理器
python
import time from contextlib import contextmanager @contextmanager def timer(name="操作"): # __enter__逻辑：记录开始时间 start = time.time() yield # 返回值可省略，或yield自定义值（如start） # __exit__逻辑：计算耗时 end = time.time() print(f"{name}耗时：{end - start:.2f}秒") # 使用定时器 with timer("文件读取"): with open("test.txt", "r", encoding="utf-8") as f: time.sleep(0.5) # 模拟耗时操作 f.read() # 输出：文件读取耗时：0.50秒示例 2：带参数的数据库连接管理器（装饰器版）
python
import pymysql from contextlib import contextmanager @contextmanager def mysql_conn(host, user, password, db): conn = None cursor = None try: # __enter__：建立连接 conn = pymysql.connect(host=host, user=user, password=password, db=db) cursor = conn.cursor() yield cursor # 返回游标给with语句 except Exception as e: if conn: conn.rollback() raise e # 抛出异常，让上层处理 finally: # __exit__：释放资源 if cursor: cursor.close() if conn: conn.close() # 使用 with mysql_conn("localhost", "root", "123456", "test_db") as cursor: cursor.execute("SELECT COUNT(*) FROM user") print(cursor.fetchone())四、高级用法与实用工具
1. 处理异常：抑制特定异常
__exit__返回True时，会抑制代码块抛出的异常（不向上传播）；返回False（默认）则正常抛出。
示例：抑制文件不存在的异常
python
class SafeFileOpen: def __enter__(self): return self def __exit__(self, exc_type, exc_val, exc_tb): # 仅抑制FileNotFoundError if exc_type == FileNotFoundError: print(f"文件不存在：{exc_val}") return True return False with SafeFileOpen(): with open("nonexist.txt", "r") as f: f.read() # 输出：文件不存在：[Errno 2] No such file or directory: 'nonexist.txt' # 无异常抛出，程序继续执行2. 嵌套上下文管理器
多个上下文管理器可写在同一个with语句中，用逗号分隔，等价于嵌套with：
python
# 嵌套写法 with open("input.txt", "r") as f_in: with open("output.txt", "w") as f_out: f_out.write(f_in.read()) # 简化写法（推荐） with open("input.txt", "r") as f_in, open("output.txt", "w") as f_out: f_out.write(f_in.read())3.contextlib模块的实用工具
Python 内置的contextlib提供了多个简化上下文管理的工具：
| 工具 | 作用 | 示例 |
|---|---|---|
contextlib.closing | 为只有close()方法的对象添加上下文管理 | with closing(requests.get(url)) as resp: print(resp.text) |
contextlib.suppress | 抑制指定异常（比手动判断更简洁） | with suppress(FileNotFoundError): open("nonexist.txt").read() |
contextlib.nullcontext | 空上下文管理器（占位用，无操作） | with nullcontext() if flag else open("test.txt") as f: ... |
示例：用suppress抑制异常
python
from contextlib import suppress # 抑制FileNotFoundError，无需try-except with suppress(FileNotFoundError): with open("nonexist.txt", "r") as f: print(f.read())五、最佳实践
- 优先使用内置上下文管理器：文件、锁、临时文件等场景，直接用 Python 内置实现，避免重复造轮子；
- 简单场景用
contextmanager装饰器：相比类实现，生成器写法更简洁，代码量少 50% 以上； - 不要滥用异常抑制：
__exit__返回True仅用于处理预期内的轻微异常，核心异常（如数据库连接失败）应抛出，让上层感知； - 确保资源真的释放：在
__exit__或finally中必须显式释放资源（如关闭连接、释放锁），避免 “假释放”； - 参数化上下文管理器：通过
__init__（类写法）或函数参数（装饰器写法）实现复用，适配不同场景。
六、总结
上下文管理器是 Python 中处理资源的 “优雅利器”，核心价值在于：
- 自动化：资源的获取 / 释放无需手动控制，避免遗漏；
- 简洁性：替代繁琐的
try-finally，代码更易读； - 健壮性：无论代码块是否抛异常，资源都能被正确释放。
从简单的文件操作到复杂的数据库连接池，上下文管理器都能让资源管理逻辑更清晰、更可靠 —— 这也是 Python “优雅”“简洁” 哲学的最佳体现之一。
