__enter__和__exit__

一、 __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__中定制自动释放资源的机制,你无须再去关系这个问题,这将大有用处

猜你喜欢

转载自www.cnblogs.com/Josie-chen/p/8903492.html
今日推荐