python--杂识--17--上下文管理器

Chasing__Dreams2024-05-11 19:19

方法一

contextlib.contextmanager 是 Python 标准库中的一个模块，用于创建上下文管理器。上下文管理器用于管理资源的获取和释放，例如文件操作中的打开和关闭操作。使用 contextmanager 装饰器可以将一个生成器函数转换为上下文管理器。

具体来说，contextmanager 装饰器接受一个生成器函数作为参数，生成器函数需要使用 yield 语句来定义上下文管理器的进入和退出时的行为。装饰器会将生成器函数包装成一个上下文管理器对象，使其可以使用 with 语句进行调用。

下面是一个简单的示例，演示了如何使用 contextmanager 装饰器创建一个自定义的上下文管理器：

python 复制代码

from contextlib import contextmanager

@contextmanager
def custom_context():
    # 进入上下文时的操作
    print("Entering the custom context")
    # 在这里可以执行资源的获取等操作
    
    # 通过 yield 将控制权交给 with 语句块
    yield
    
    # 退出上下文时的操作
    print("Exiting the custom context")
    # 在这里可以执行资源的释放等操作

# 使用 with 语句调用上下文管理器
with custom_context():
    # 在上下文中执行一些操作
    print("Inside the custom context")

运行以上代码，你会看到如下输出：

python 复制代码

Entering the custom context
Inside the custom context
Exiting the custom context

这个例子中，custom_context 函数是一个生成器函数，通过 yield 语句分别定义了进入和退出上下文时的操作。使用 @contextmanager 装饰器将其转换为上下文管理器后，就可以通过 with 语句来调用它。在 with 语句块中，会先执行进入上下文时的操作，然后执行 with 语句块中的代码，最后执行退出上下文时的操作。

举例：redis分布式锁

方法二：

使用__enter__()、exit() 魔法方法实现