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