Python上下文管理协议:__enter__和__exit__

上下文管理器(context manager)是Python2.5开始支持的一种语法，用于规定某个对象的使用范围。一旦进入或者离开该使用范围，会有特殊操作被调用 (比如为对象分配或者释放内存)。它的语法形式是with...as...

所谓上下文管理协议,就是咱们打开文件时常用的一种方法:with

__enter__(self):当with开始运行的时候触发此方法的运行

__exit__(self, exc_type, exc_val, exc_tb):当with运行结束之后触发此方法的运行

exc_type如果抛出异常,这里获取异常的类型 

exc_val如果抛出异常,这里显示异常内容

exc_tb如果抛出异常,这里显示所在位置

代码示例:

class Sample:
    def __enter__(self): 
        print "In __enter__()" 
        return "Foo"

     def __exit__(self, type, value, trace):
         print "In __exit__()"

    def get_sample(): 
        return Sample() 

with get_sample() as sample: 
    print "sample:", sample        

输出如下：

In __enter__()
sample: Foo
In __exit__()

__enter__()方法被执行
__enter__()方法返回的值 - 这个例子中是”Foo”，赋值给变量’sample’
执行代码块，打印变量”sample”的值为 “Foo”
__exit__()方法被调用

改一下代码，看看具体如何工作的:

class Sample:
    def __enter__(self):
        return self
    def __exit__(self, type, value, trace):
        print "type:", type
        print "value:", value
        print "trace:", trace

    def do_something(self):
        bar = 1/0
        return bar + 10

with Sample() as sample:
    sample.do_something()

代码执行后：

type: <type 'exceptions.ZeroDivisionError'>
value: integer division or modulo by zero
trace: <traceback object at 0x1004a8128>
Traceback (most recent call last):
     File "./with_example02.py", line 19, in <module>
        sample.do_somet hing()
    File "./with_example02.py", line 15, in do_something
        bar = 1/0
ZeroDivisionError: integer division or modulo by zero

开发库时，清理资源，关闭文件等等操作，都可以放在__exit__方法当中。因此，Python的with语句是提供一个有效的机制，让代码更简练，同时在异常产生时，清理工作更简单。