Python进阶学习笔记——上下文管理器

上下文管理器

• 定义：一个类实现了__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方法