上下文管理器
- 定义:一个类实现了
__enter__()
和__exit__()
两个方法,通过该类创建的对象我们称之为上下文管理器。 - 上下文管理器可以使用
with
语句,with
语句之所以这么强大,背后是由上下文管理器支撑的,也就是说刚才open
函数创建的文件对象就是一个上下文管理对象。 - 注意:
with
不是上下文管理器,with
管理的资源才称为上下文管理器。
with…as…原理
- 使用类定义上下文管理器需要在类上定义
__enter__
和__exit__
方法,执行: with A() as a
:
- 语句时会先执行
__enter__
方法,这个方法会返回一个操作对象赋值给后面的a
变量; - 当
with
里面的语句产生异常或正常执行完时,都要执行类中的__exit__
方法。
class Management(object):
def __init__(self, file_name, mode):
print('进入__init__方法')
print('文件名:', file_name)
print('打开模式:', mode)
def __enter__(self):
print('进入__enter__方法')
print('返回self,相当于给as后变量赋值')
return self
def __exit__(self, exc_type, exc_val, exc_tb):
print('进入__exit__方法')
print('type:', new_type)
print('value:', value)
print('trace:', trace)
self.new_close()
def new_write(self):
print('进入newWrite方法')
def new_close(self):
print('进入close方法')
def new_open(file_name, mode):
return Management(file_name, mode)
if __name__ == '__main__':
with new_open('文件', 'w') as a:
a.new_write()
--------------Out--------------
进入__init__方法
文件名: 文件
打开模式: w
进入__enter__方法
返回self,相当于给as后变量赋值
进入newWrite方法
进入__exit__方法
type: None
value: None
trace: None
进入close方法