Python@contextmanager

上下文管理器

上下文管理器是实现了上下文管理协议的对象,其特有的语法是“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
quer

你可能感兴趣的:(Python@contextmanager)