上下文管理协议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后的语句正常执行
![](https://images.cnblogs.com/OutliningIndicators/ContractedBlock.gif)
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 '''
用途或者说好处:
1.使用with语句的目的就是把代码块放入with中执行,with结束后,自动完成清理工作,无须手动干预
2.在需要管理一些资源比如文件,网络连接和锁的编程环境中,可以在__exit__中定制自动释放资源的机制,你无须再去关系这个问题,这将大有用处