Python:contextlib模块——上下文管理器工具

上篇博文简单的讲了下with语句的用法以及上下文管理器对象的概念，想必对上下文管理器对象有一定的了解。一个对象如果实现了上下文管理协议，也就是在类中定义了__enter__()和__exit__()方法两个方法的对象，都可以称为上下文管理器对象。
但是，Python中有个contextlib模块，是个比with优美的东西，提供上下文机制的，它是通过Generator装饰器实现的，不再是采用__enter__和__exit__。
contextlib模块对外有三个接口，分别是contextlib.contextmanager,contextlib.nested,
contextlib.closing。
（1）contextlib.contextmanager
contextmanager装饰器，装饰的函数必须是一个生成器。然后返回一个函数，在函数调用的时候返回一个上下文管理器，是一种针对函数级别的上下文管理机制。常用方式如下：

#coding=utf-8
from contextlib import contextmanager

@contextmanager
def make_context() :
print 'enter'
try :
yield {}
except RuntimeError, err :
print 'error' , err
finally :
print 'exit'

with make_context() as value :
print value

输出结果：
enter
{}
exit
目前也没找到好的使用例子，以后有的话，会再贴上。
（2）contextlib.nested
nested为了一次调用多个上下文管理器的时候用，将所有上下文管理器里的__enter__放到vars里，一次yield返回，然后unpack。就产生多个对象了。__exit__都放到exits列表里，在finally里统一调用，这些__exit__在调用上下文管理器产生异常时，如何处理。如果返回false,将会把异常继续抛出，返回true则不会抛出异常。但是nested已经过时了，因为with已经可以通过多个上下文的直接嵌套了。用法如下例：

#coding=utf-8
from contextlib import contextmanager,nested

@contextmanager
def make_context(name) :
print 'enter', name
yield name
print 'exit', name

with nested(make_context('A'), make_context('B')) as (a, b) :
print a
print b

with make_context('A') as a, make_context('B') as b :
print a
print b

输出结果：
enter A
enter B
A
B
exit B
exit A
enter A
enter B
A
B
exit B
exit A

（3）contextlib.closing
文件类支持上下文管理器, 但是有一些对象不支持。还有一些类使用close()方法但是不支持上下文管理器，使用closing()来为他创建一个上下文管理器。就是把有close方法却没有__exit__方法的对象变成一个上下文管理器。
看下面代码：

from contextlib import closing
class Door(object) :
def open(self) :
print 'Door is opened'
def close(self) :
print 'Door is closed'

with closing(Door()) as door :
door.open()