上下文管理协议with_open,__enter__和__exit__（三十八）

在操作文件对象的时候可以这么写

with open('a.txt') as f:
    '代码块'

上述叫做上下文管理协议，即with语句，为了让一个对象兼容with语句，必须在这个对象的类中声明__enter__和__exit__方法

with obj as f ==========等同于 f = obj.__enter__()

class Foo:
    def __init__(self,name):
        self.name = name

    def __enter__(self):
        print("出现with语句，对象的__enter__被触发，有返回值则赋值给as声明的变量")
        return self
    def __exit__(self, exc_type, exc_val, exc_tb):
        print("with中代码块执行完毕时执行")

with Foo('a.txt') as f:
    print("执行代码块.....")
    print(f,f.name)
'''
出现with语句，对象的__enter__被触发，有返回值则赋值给as声明的变量
执行代码块.....
<__main__.Foo object at 0x0000022354BF4E48> a.txt
with中代码块执行完毕时执行
'''

__exit__()中的三个参数分别代表异常类型，异常值和追溯信息,with语句中代码块出现异常，则with后的代码都无法执行

 

class Foo:
    def __init__(self,name):
        self.name = name

    def __enter__(self):
        print("出现with语句，对象的__enter__被触发，有返回值则赋值给as声明的变量")
        return self
    def __exit__(self, exc_type, exc_val, exc_tb):
        print("with中代码块执行完毕时执行")
        print(exc_type) # <class 'AttributeError'>
        print(exc_val) # raise a error
        print(exc_tb) # <traceback object at 0x00000255B7EBA208>

with Foo('a.txt') as f:
    print("执行代码块.....")
    raise AttributeError("raise a error")
    print('*'*10) # -----不会执行
print('0'*10) # -----不会执行

如果__exit()返回值为True,那么异常会被清空，就好像啥都没发生一样，with后的语句正常执行

class Foo:
    def __init__(self,name):
        self.name = name

    def __enter__(self):
        print("出现with语句，对象的__enter__被触发，有返回值则赋值给as声明的变量")
        return self
    def __exit__(self, exc_type, exc_val, exc_tb):
        print("with中代码块执行完毕时执行")
        print(exc_type) # <class 'AttributeError'>
        print(exc_val) # raise a error
        print(exc_tb) # <traceback object at 0x00000255B7EBA208>
        return True

with Foo('a.txt') as f:
    print("执行代码块.....")
    raise AttributeError("raise a error")
    print('*'*10) # -----不会执行
print('0'*10) 
'''
出现with语句，对象的__enter__被触发，有返回值则赋值给as声明的变量
执行代码块.....
with中代码块执行完毕时执行
<class 'AttributeError'>
raise a error
<traceback object at 0x000001293C15A248>
0000000000
'''

View Code