上下文管理器

  • 定义:一个类实现了__enter__()__exit__()两个方法,通过该类创建的对象我们称之为上下文管理器。
  • 上下文管理器可以使用with语句,with语句之所以这么强大,背后是由上下文管理器支撑的,也就是说刚才open函数创建的文件对象就是一个上下文管理对象。
  • 注意with不是上下文管理器,with管理的资源才称为上下文管理器。

with…as…原理

  • 使用类定义上下文管理器需要在类上定义__enter____exit__方法,执行:
  • with A() as a:
  1. 语句时会先执行__enter__方法,这个方法会返回一个操作对象赋值给后面的a变量;
  2. 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方法