一、 __enter__和__exit__
我们知道在操作文件对象的时候可以这么写:
with为工厂函数,open得到的就是一个类,赋值为f。
1 with open('a.txt') as f: 2 '代码块'
上述叫做上下文管理协议,即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: # 并不是把Open('a.txt')结果返回给f,而是触发Open中的enter方法,将enter函数的返回值给f print('=====>执行代码块')
print(f) #<__main__.Foo object at 0x0000000001199198> #说明f为一个对象 print(f.name) #可以利用f来调用它当中的变量name #等待with对应代码块执行结束,触发exit函数
__exit__()中的三个参数分别代表异常类型,异常值和追溯信息,with语句中代码块出现异常,则with后的代码都无法执行
View Code
如果__exit()返回值为True,那么异常会被清空,就好像啥都没发生一样,with后的语句正常执行
View Code
练习:模拟Open
用途或者说好处:
1.使用with语句的目的就是把代码块放入with中执行,with结束后,自动完成清理工作,无须手动干预
2.在需要管理一些资源比如文件,网络连接和锁的编程环境中,可以在__exit__中定制自动释放资源的机制,你无须再去关系这个问题,这将大有用处