Python@contextmanager-CSDN博客

本文链接：https://blog.csdn.net/weixin_37143690/article/details/89392578

本文介绍了Python中的上下文管理器，它用于保存和恢复全局状态，关闭文件等，通过`with`语句实现。上下文管理器需要实现`__enter__`和`__exit__`方法。此外，还讲解了`contextlib`模块，这是一个用于创建上下文管理器的内建模块，可以简化代码并提供优雅的实现方式。文中给出了一个使用`@contextmanager`装饰器的实际应用示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

上下文管理器

上下文管理器是实现了上下文管理协议的对象，其特有的语法是“with ...as”。主要用于保存和恢复各种全局状态，关闭文件等，并为try...except...finally提供了一个方便使用的封装。

上下文管理协议具体来说就是在类里面实现以下两个方法：

__enter__(): 从该方法进入运行时上下文，并返回当前对象或者与运行时上下文相关的其他对象。如果with语句有as关键词存在，返回值会绑定在as后的变量上。

__exit__(exc_type, exc_val, exc_tb): 退出运行时上下文，return True可以正常抛出异常，Flase则不会有错误信息。如果在执行with语句体时发生异常，那退出时参数会包括异常类型、异常值、异常追踪信息，否则，3个参数都是None。

class MyResource1:
    def __enter__(self):
        print('connect to resource')
        return self    # return可以是对象，然后会绑定到as后面的变量

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('close resource connection')
        # __exit__方法 return Ture 异常会报错， return Flase 不会报错

    def query(self):
        print('query data')


with MyResource1() as r:
    r.query()

运行结果：
connect to resource
query data
close resource connection