使用with语句可确保资源正确释放,如文件操作中即使发生异常也能自动关闭;当无法使用with时,可在finally块中手动释放资源;通过自定义__enter__和__exit__方法或contextlib的@contextmanager装饰器可实现自定义资源管理,推荐优先使用with语句防止资源泄漏。

在 Python 中处理异常时,确保资源正确释放非常重要,比如文件、网络连接或数据库会话等。如果未妥善释放,可能导致资源泄漏或程序不稳定。Python 提供了多种方式来保证无论是否发生异常,资源都能被安全释放。
使用 with 语句自动管理资源
with 语句是释放资源最推荐的方式,它基于上下文管理器(context manager)机制,能确保进入和退出代码块时自动执行准备和清理操作。
常见场景如文件操作:
with open('data.txt', 'r') as f:
content = f.read()
# 即使这里发生异常,文件也会被自动关闭
只要对象实现了 __enter__ 和 __exit__ 方法,就可以用于 with 语句。除了文件,threading.Lock、数据库连接等也支持这种方式。
立即学习“Python免费学习笔记(深入)”;
在 finally 块中手动释放资源
当无法使用 with 时,可以在 try...finally 结构的 finally 块中释放资源。finally 块无论是否抛出异常都会执行。
f = None
try:
f = open('data.txt', 'r')
content = f.read()
# 可能出现异常
except FileNotFoundError:
print("文件未找到")
finally:
if f:
f.close() # 确保文件被关闭
这种方法适用于旧版本代码或自定义资源管理,但不如 with 简洁安全。
自定义上下文管理器
对于自定义资源,可以实现自己的上下文管理器,确保资源释放逻辑清晰。
class ManagedResource:
def __enter__(self):
print("获取资源")
return self
def __exit__(self, exc_type, exc_value, traceback):
print("释放资源")
if exc_type:
print(f"异常类型: {exc_type}")
return False # 不抑制异常使用示例
with ManagedResource() as res:
raise ValueError("出错了")
这样即使发生异常,__exit__ 方法仍会被调用,完成清理工作。
使用 contextlib 简化上下文管理
对于简单场景,可以用 @contextmanager 装饰器快速创建上下文管理器。
from contextlib import contextmanager@contextmanager def managed_resource(): print("获取资源") try: yield "资源" finally: print("释放资源")
使用
with managed_resource() as res: print(res) raise RuntimeError("测试异常")
yield 之前的代码相当于 __enter__,finally 块对应 __exit__,逻辑清晰且易于维护。
基本上就这些。优先使用 with 语句,配合上下文管理器或 contextlib,能有效避免资源泄漏问题。不复杂但容易忽略细节,关键是养成习惯。










