__enter__,__exit__上下文管理协议

上下文管理协议__enter__,__exit__

用途或者说好处:

1.使用with语句的目的就是把代码块放入with中执行,with结束后,自动完成清理工作,无须手动干预

2.在需要管理一些资源比如文件,网络连接和锁的编程环境中,可以在__exit__中定制自动释放资源的机制,你无须再去关系这个问题,这将大有用处

我们知道在操作文件对象的时候可以这么写

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

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

class Open:
    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 Open('a.txt') as f:
    print('=====>执行代码块')
    # print(f,f.name)

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

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

    def __enter__(self):
        print('出现with语句,对象的__enter__被触发,有返回值则赋值给as声明的变量')

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('with中代码块执行完毕时执行我啊')
        print(exc_type)
        print(exc_val)
        print(exc_tb)



with Open('a.txt') as f:
    print('=====>执行代码块')
    raise AttributeError('***着火啦,救火啊***')
print('0'*100) #------------------------------->不会执行

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

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

    def __enter__(self):
        print('出现with语句,对象的__enter__被触发,有返回值则赋值给as声明的变量')

    def __exit__(self, exc_type, exc_val, exc_tb):
        print('with中代码块执行完毕时执行我啊')
        print(exc_type)
        print(exc_val)
        print(exc_tb)
        return True



with Open('a.txt') as f:
    print('=====>执行代码块')
    raise AttributeError('***着火啦,救火啊***')
print('0'*100) #------------------------------->会执行

其实看这个就行,讲的所有的点

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

    def __enter__(self):
        print('执行了__enter__方法')
        return self                 #返回实例化对象给f
    def __exit__(self, exc_type, exc_val, exc_tb):
        print('执行了__exit__方法')
        #如果爆出异常后,exc_type, exc_val, exc_tb会捕捉到错误信息,如果没有错误,三个参数值为None
        print(exc_type)     #<class 'NameError'>  错误类型,也就是错误类
        print(exc_val)      #name 'aaaaa' is not defined    #错误位置,也就是异常值
        print(exc_tb)       #<traceback object at 0x00000011F51B9988>   #没啥用,也就是追逐信息
        return True         #在有异常后,如果__exit__执行结果返回的是True,不会报错,返回其它,会报错
with Foo('a.txt') as f:     #with 实例化的一个对象,会触发对象属性中的__enter__方法
    print(f)    #__enter__方法返回值给f,f可拿到Foo实例化后的对象
    print(f.name)      #可拿到实例化的信息
    print(aaaaa)    #会抛出异常,抛出异常后会触发__exit__的运行,
                    # 1、如果__exit__的返回值为True,代表吞掉了异常信息(就是不会报错)
                    # 2、如果__exit__的返回值不为True,代表吐出了异常信息(就是会报错)
    print('不会执行')
    print('不会执行')
    print('不会执行')
print('------------------------')

 

原文地址:https://www.cnblogs.com/shangpolu/p/6233263.html