Python上下文管理器全实例详解

沿途有光

发布时间：2026-10-13| 4人浏览过

原创

本文深入解析Python上下文管理器，通过文件操作、异常处理及SQLAlchemy会话管理等实例，详解__enter__与__exit__方法原理。介绍如何利用contextlib装饰器简化函数式上下文管理，提升代码可读性与资源安全性。

Python上下文管理器简介

Python上下文管理器

简介

近期在实际开发中频繁接触该机制，经深入调研发现其价值显著，特此整理记录以供参考。

使用场景与执行顺序

使用场景

当业务逻辑涉及临时资源的获取与释放，且需兼顾异常处理时，传统方式通常依赖try-finally语句块。例如，在打开文件进行读写操作后，必须确保文件句柄被正确关闭，无论操作是否成功。

打开文件：


f = None
try:
 print("try")
 f = open("__init__.py", "r")
 print(f.read())
except Exception as e:
 print("exception")
finally:
 if f:
  print("finally")
  f.close()

利用上下文管理器：


class OpenHandle:

 def __init__(self, filename, mode):
  self.filename = filename
  self.mode = mode

 def __enter__(self):
  self.f = open(self.filename, self.mode)
  return self.f

 def __exit__(self, exc_type, exc_val, exc_tb):
  if exc_type:
   print("exception")
  else:
   print("normal")
  self.f.close()

with OpenHandle("book.txt", "r") as f:
 print(f.read())

通过with-as语句重构代码，可将开发者注意力集中于核心业务逻辑，自动屏蔽资源获取与释放的冗余操作。此举显著提升了代码的可读性与维护性，优势明显。

执行顺序

理解上下文管理器的执行流程是掌握其用法的关键：

初始化阶段，调用handle类的__init__()方法。
执行__enter__()方法以获取资源对象，并将返回值赋给as子句后的变量。
执行业务代码逻辑。
执行__exit__方法，该方法接收三个参数：异常类型、异常对象及调用栈对象；若无异常发生，这三个参数均为None。
根据业务执行结果，抛出异常或正常结束流程。

函数式上下文管理器与SQLAlchemy应用

函数式上下文管理器

通过import contextlib模块中的contextmanager装饰器，可将普通函数转换为上下文管理器。该装饰器底层原理是返回一个实现了__enter__和__exit__方法的类实例，从而简化代码结构。


from contextlib import contextmanager

@contextmanager
def managed_resource(*args, **kwds):
 # Code to acquire resource, e.g.:
 resource = acquire_resource(*args, **kwds)
 try:
  yield resource
 finally:
  # Code to release resource, e.g.:
  release_resource(resource)

>>> with managed_resource(timeout=3600) as resource:
...  # Resource is released at the end of this block,
...  # even if code in the block raises an exception

模板代码

sqlalchemy会话上下文管理器

利用此机制管理sqlalchemy会话对象的创建与销毁，是控制数据库事务的理想选择。


class DbTransaction:

 def __init__(self, session_maker):
  self.session_maker = session_maker

 def __enter__(self):
  self.session = self.session_maker()
  return self.session

 def __exit__(self, exc_type, exc_val, exc_tb):
  if exc_type:
   self.session.rollback()
  else:
   self.session.commit()
  self.session.close()
  return False if exc_type else True

以上内容涵盖了上下文管理器的核心知识点，感谢读者的学习及对本站的支持。