python with用法

python中with可以明显改进代码友好度,比如:

with open('a.txt') as f:  
    print f.readlines()  

  


为了我们自己的类也可以使用with, 只要给这个类增加两个函数__enter__, __exit__即可:

>>> class A:  
    def __enter__(self):  
        print 'in enter'  
    def __exit__(self, e_t, e_v, t_b):  
        print 'in exit'  
  
>>> with A() as a:  
    print 'in with'  
  
in enter  
in with  
in exit 

  


另外python库中还有一个模块contextlib,使你不用构造含有__enter__, __exit__的类就可以使用with:

>>> from contextlib import contextmanager  
>>> from __future__ import with_statement  
>>> @contextmanager  
... def context():  
...     print 'entering the zone'  
...     try:  
...         yield  
...     except Exception, e:  
...         print 'with an error %s'%e  
...         raise e  
...     else:  
...         print 'with no error'  
...  
>>> with context():  
...     print '----in context call------'  
...  
entering the zone  
----in context call------  
with no error  

  


使用的最多的就是这个contextmanager, 另外还有一个closing 用处不大

from contextlib import closing  
import urllib  
  
with closing(urllib.urlopen('http://www.python.org')) as page:  
    for line in page:  
        print line  

  

posted @ 2011-10-27 13:28  酱油哥  阅读(283)  评论(0编辑  收藏  举报