本文由 简悦 SimpRead 转码， 原文地址 blog.csdn.net
Python 中的上下文管理器（Context Managers）是一个允许你在执行代码块之前和之后自动执行某些操作（如资源分配和释放）的对象。它们主要通过with语句使用，这种方式让资源管理变得更加简洁和安全。最常见的上下文管理器是文件操作，但你也可以创建自定义的上下文管理器来管理任何类型的资源，如数据库连接、网络连接或任何需要显式开启和关闭的资源。
基本概念
上下文管理器通常包含两个方法：__enter__() 和 __exit__()。
__enter__()方法在进入with代码块之前被调用。该方法返回的对象会被绑定到with语句的as子句中指定的变量（如果有的话）。__exit__()方法在离开with代码块时被调用，即使是由于异常而离开。这个方法接收三个参数：exc_type（异常类型）、exc_val（异常值）和exc_tb（异常追踪信息）。如果with代码块正常执行完成，则这三个参数都会被设置为None。
使用场景
- 文件操作：自动打开和关闭文件。
- 锁操作：自动获取和释放锁。
- 数据库连接：自动建立和关闭数据库连接。
- 线程 / 进程管理：自动管理线程或进程的创建和销毁。
示例
文件操作
这是最常见的上下文管理器使用场景：
with open('example.txt', 'w') as file:
file.write('Hello, world!')
# 文件自动关闭
自定义上下文管理器
要创建自定义的上下文管理器，你需要定义一个类，实现 __enter__() 和 __exit__() 方法。
class MyContextManager:
def __enter__(self):
print("Entering context")
# 可以在这里设置初始状态，并返回一些值（如果有必要）
return self
def __exit__(self, exc_type, exc_val, exc_tb):
print("Exiting context")
# 清理资源
if exc_type is not None:
print(f"Exception occurred: {exc_type}")
# 可以在这里处理异常
# 返回True表示异常已被处理，False表示应该重新抛出异常
return False
# 使用自定义上下文管理器
with MyContextManager() as cm:
print("Inside the context")
# 抛出一个异常以测试
raise ValueError("Test exception")
# 输出将包括异常处理信息
结论
Python 的上下文管理器提供了一种非常方便的方式来管理资源，确保无论发生什么情况，资源都能被正确释放或关闭。通过使用with语句和上下文管理器，你的代码不仅更简洁，而且更安全、更易于维护。