上下文管理器是一个很好的资源管理工具。
它们允许您在需要时精确地分配和释放资源。
一个大家熟知的例子是 with open()
语句:
测试代码:
with open('notes.txt', 'w') as f:
f.write('some todo...')
这将打开一个文件,并确保在程序执行离开 with 语句的上下文后自动关闭它。
它还处理异常并确保即使在出现异常的情况下也能正确关闭文件。
在内部,上面的代码翻译成这样的:
f = open('notes.txt', 'w')
try:
f.write('some todo...')
finally:
f.close()
我们可以看到使用上下文管理器和 with
语句更短更简洁。
上下文管理器的例子
- 打开和关闭文件
- 打开和关闭数据库连接
- 获取和释放锁
from threading import Lock
lock = Lock()
# 传统的方式:
lock.acquire()
lock.release()
# 好的方式:
with lock:
# do stuff
自定义上下文管理器
为了支持我们自己的类的 with
语句,我们必须实现 __enter__
和 __exit__
方法。
当执行进入 with
语句的上下文时,Python 会调用 __enter__
。
在这里应该获取并返回资源。当执行再次离开上下文时,__exit__
被调用并释放资源。
代码:
class ManagedFile:
def __init__(self, filename):
print("初始化类", filename)
self.filename = filename
def __enter__(self