上下文管理器__enter__和__exit__

直接上代码：

import os

class Open(object):
    def __init__(self, file):
        self.file = file

    def __enter__(self):
        print('实例化的时候调用此代码块，若使用as则返回实例对象给as的对象')
        self.f = open(self.file)
        return self  # 返回实例是为了调用类方法

    def __exit__(self, exc_type, exc_val, exc_tb):  # 三个参数分别代表错误类型，错误值，错误追踪traceback
        print('Exit...')
        print(exc_tb)
        print(exc_type)
        print(exc_val)
        self.f.close()

    @property   # 装饰器，以属性的方式调用方法
    def say_name(self):
        return os.path.dirname(os.path.abspath(__file__))


if __name__ == '__main__':
    with Open(r'bin/aa.py') as f:  # 通过with的方式实例化类，会触发类的__enter__方法，若__enter__有返回值，则把返回值赋值给as的对象
        print('======')
        print(f.say_name)
        print('------')  # 以上代码块执行完毕触发__exit__方法
        print(abcd)  # 此处不存在变量abcd，程序会抛出异常并且终止，但是因为有__exit__方法，错误信息会被吞掉，并直接执行__exit__方法
        print('.....')  # 上一行抛异常，此处不会执行
    print('lalala')  # 上一行抛异常，此处不会执行

运行结果：

C:\Users\lwj\AppData\Local\Programs\Python\Python37\python.exe D:/02Project/pacho/12day/st_上下文管理.py
实例化的时候调用此代码块，若使用as则返回实例对象给as的对象
======
D:\02Project\pacho\12day
------
Exit...
<traceback object at 0x0000000001FFDB48>
<class 'NameError'>
name 'abcd' is not defined
Traceback (most recent call last):
  File "D:/02Project/pacho/12day/st_上下文管理.py", line 29, in <module>
    print(abcd)  # 此处不存在变量abcd，程序会抛出异常并且终止，但是__exit__方法返回为False（with中的代码块出现了异常），错误信息会被吐出，并直接执行__exit__方法
NameError: name 'abcd' is not defined

Process finished with exit code 1

 

下面说下__exit__的返回值，举例：

import os

class Open(object):
    def __init__(self, file):
        self.file = file

    def __enter__(self):
        print('实例化的时候调用此代码块，若使用as则返回实例对象给as的对象')
        self.f = open(self.file)  # 此处是为了能够调用close() 方法
        return self  # 返回实例是为了调用类方法

    def __exit__(self, exc_type, exc_val, exc_tb):  # 三个参数分别代表错误类型，错误值，错误追踪traceback
        print('Exit...')
        print(exc_tb)
        print(exc_type)
        print(exc_val)
        self.f.close()
        return True  # 如果exit返回True则吞掉所有异常并结束with语句；如果exit返回False，则吐出异常

    @property   # 装饰器，以属性的方式调用方法
    def say_name(self):
        return os.path.dirname(os.path.abspath(__file__))


if __name__ == '__main__':
    with Open(r'bin/aa.py') as f:  # 通过with的方式实例化类，会触发类的__enter__方法，若__enter__有返回值，则把返回值赋值给as的对象，即 f = 实例.__enter__()
        print('======')
        print(f.say_name)
        print('------')  # 以上代码块执行完毕触发__exit__方法
        print(abcd)  # 此处不存在变量abcd，程序会抛出异常并且终止，但是因为有__exit__方法，错误信息会被吞掉，并直接执行__exit__方法
        print('.....')  # 上一行抛异常，则直接执行exit方法，此行不会被执行，with语句结束
    print('lalala')  # 上一行抛异常，因为exit返回为True，此行会被执行

执行的结果:

C:\Users\lwj\AppData\Local\Programs\Python\Python37\python.exe D:/02Project/pacho/12day/st_上下文管理.py
实例化的时候调用此代码块，若使用as则返回实例对象给as的对象
======
D:\02Project\pacho\12day
------
Exit...
<traceback object at 0x000000000200DB08>
<class 'NameError'>
name 'abcd' is not defined
lalala

Process finished with exit code 0