8.python之上下文管理协议

使用上下文管理协议,有什么好处?python

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

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


以前所提到的with代码块,就是python的上下文管理操做,好比说经过python打开一个文件,就能够经过with代码块结合open去实现,经过这种方式打开的文件,执行了相应的操做后,无需咱们手动去close文件,文件就会自动关闭。ide

好比:spa

with open('a.txt') as f:对象

'代码块'ci

#上面这个例子,就是一个上下文管理协议,即with语句,为了让一个对象能够去兼容with语句,则必须在这个对象的类中,去声明__enter__和__exit__方法。资源

这种上下文的管理,就是经过类中的__enter__和__exit__这两个内置方法去实现的。it


下面是__enter__和__exit__的用法示例:class

class test:

    def __init__(self,name):

        self.name = name

    def __enter__(self):

        #print "只要with语句一出现,这个对象的__enter__方法就会被触发,__enter__这个方法的返回值会赋值给as 后面声明的变量"

        print "我是__enter__方法,with出现时就会执行我~"

    def __exit__(self, exc_type, exc_val, exc_tb):

        print "我是__exit__方法,__enter__执行完毕会执行我"

with test(123) as t1:

    print "aaaa"


咱们来看看输出结果:

我是__enter__方法,with出现时就会执行我~

aaaa

我是__exit__方法,__enter__执行完毕会执行我


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

#传入__exit__方法中的exc_type,exc_val,exc_tb,分别是(异常类,异常的值,追踪信息)这三个参数只有当__enter__方法或者with下的代码块下的代码出现异常,这三个参数才会有值,不然就是三个None。

#__exit__()中的三个参数分别表明异常类型,异常值和追溯信息,with语句中代码块出现异常,则with后的代码都没法执行,直接开始执行__exit__方法。


关于上下文管理的异常处理:

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) #------------------------------->不会执行

从上面这个例子能够看出来,当with的代码块的执行出现异常的时候,python会直接开始执行对象的__exit__方法,当__exit__内的方法也执行完毕,整个程序就终止掉了。


那么,如何处理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) #------------------------------->会执行



当__exit__方法,返回的为值True时,就好像啥都没发生同样,with后的语句正常执行,with代码块中的异常被屏蔽掉了!!


最后关于python的上下文管理,作一个总结:

  1. 在没有任何异常的状况下,整个代码块的内容运行完毕后会去触发对象的__exit__方法,它的三个参数都为None。

  2. 当有异常存在的状况下,从异常位置,直接触发__exit__。

    2.1 当手动将__exit__方法的返回值修改成True,with语句就会屏蔽掉异常。

    2.2 当__exit__的返回值不为True,在with代码块中遇到异常就会抛出。

    2.3 当__exit__这个方法一旦运行完毕,就表明了整个with语句执行完毕。