首页 > 代码库 > Python:contextlib模块——上下文管理器工具
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()
Python:contextlib模块——上下文管理器工具