十四 __enter__和__exit__
我们知道在操作文件对象的时候可以这么写
with open('a.txt') as f:
'代码块'
上述叫做上下文管理协议,即with语句,为了让一个对象兼容with语句,必须在这个对象的类中声明__enter__和__exit__方法
__exit__()中的三个参数分别代表异常类型,异常值和追溯信息,with语句中代码块出现异常,则with后的代码都无法执行
如果__exit()返回值为True,那么异常会被清空,就好像啥都没发生一样,with后的语句正常执行
# coding=utf-8
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中代码块执行完毕时执行我啊')
print(exc_type)
print(exc_val)
print(exc_tb)
#return True
with Open('aa.txt') as f:
print(f)
print(f.name)
print(sdsd) # 触发 __exit__ ,之后的代码就不会运行
print('-------')
print('End....')
'''
运行结果:
出现with语句,对象的__enter__被触发,有返回值则赋值给as声明的变量
<__main__.Open object at 0x0000016D7FA95BE0>
aa.txt
with中代码块执行完毕时执行我啊
<class 'NameError'>
name 'sdsd' is not defined
<traceback object at 0x0000016D7FA925C8>
End....
'''
'''
总结:
with obj as f:
代码块
1. with obj ----> 触发obj.__enter__(),拿到返回值
2. as f --> f = 返回值
3. with obj as f 等同于 f = obj.__enter__()
4. 执行代码块
1> 没有异常,整个代码块运行完毕后去触发__exit__(),它的三个参数都返回None
2> 有异常的情况下,从异常出现的位置直接触发__exit))
a:如果__exit__返回值为True,代表吞掉了异常
b:如果__exit__返回值不为True,代表吐出了异常
c:__exit__运行完毕,就代表整个with语句运行完毕
'''
用途或者说好处:
1.使用with语句的目的就是把代码块放入with中执行,with结束后,自动完成清理工作,无须手动干预
2.在需要管理一些资源比如文件,网络连接和锁的编程环境中,可以在__exit__中定制自动释放资源的机制,你无须再去关系这个问题,这将大有用处