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)
参考: