contextlib 是 Python 标准库中的一个模块,提供了一些用于创建上下文管理器(context manager)的实用工具。contextlib.contextmanager 是 contextlib 模块中的一个装饰器,用于将一个生成器函数转换为上下文管理器。通过使用这个装饰器,可以更方便地创建上下文管理器,而不必编写完整的类。 contextlib.contextmanager 装饰器: 允许将一个生成器函数转换为上下文管理器。 被装饰的生成器函数的 yield 语句之前的代码用于进入上下文,yield 语句之后的代码用于退出上下文。yield 的值通常是由 with 语句中的变量接收的。
import contextlib@contextlib.contextmanager
def my_context():print("Entering the context")yield "Hello, context!"print("Exiting the context")# my_context函数中需要yield语句。# 使用上下文管理器,yield值将在 with语句块中被赋给as后面的变量value
with my_context() as value:print("Inside the context:", value)
参考:
https://docs.python.org/zh-cn/3.10/library/contextlib.html