Python在类中实现上下文管理器及优化方案

一，Python中with语句：上下文管理协议
类中实现上下文管理器需要实现两个方法：
1、def enter(self) (必须return self)
2、def exit(self)
如下代码实例：
class A（）：
def_enter_(self): #打开文件 获取资源
print(‘start’)
return self
def opr(self):
print(‘处理’）
def_exit_(self，exc_type,exc_val,exc_tb): #释放资源
print(‘end’)
with A() as a:
a.opr()

结果顺序：start ，处理 ，end

二、使用contextlib简化上下文管理器
import contextlib

@contextlib.contextmannger
def file_open():
#yield上方相当于_enter_()函数
print（‘open file’）
yield{}
#yield下方相当于_exit_()函数
print()
with file_open(‘test.text’) as f:
print(‘file opration’)

（此处使用yield{}可以省掉类中多写的两个方法）