python with statement 进行文件操作指南

由于之前有一个项目老是要打开文件，然后用pickle.load(file)，再处理。。。最后要关闭文件，所以觉得有点繁琐，代码也不简洁。所以向python with statement寻求解决方法。

在网上看到一篇文章：http://effbot.org/zone/python-with-statement.htm是介绍with 的，参考着例子进行了理解。

如果经常有这么一些代码段的话，可以用一下几种方法改进：

代码段：

    set thing up
    try：
      do something
    except :
      handle exception
    finally:
      tear thing down

案例1：

假如现在要实现这么一个功能，就是打开文件，从文件里面读取数据，然后打印到终端，之后关闭文件。

那么从逻辑上来说，可以抽取"打印到终端"为数据处理部分，应该可以独立开来作为一个函数。其他像打开、关闭文件应该是一起的。

文件名为：for_test.txt

方法1：

用函数，把公共的部分抽取出来。

    #!/usr/bin/env python 
    from __future__ import with_statement  
    filename = 'for_test.txt' 
    def output(content): 
      print content 
    #functio solution 
    def controlled_execution(func): 
      #prepare thing 
      f = None 
      try: 
        #set thing up 
        f = open(filename, 'r') 
        content = f.read() 
        if not callable(func): 
          return 
        #deal with thing  
        func(content) 
      except IOError, e: 
        print 'Error %s' % str(e) 
      finally: 
        if f:  
          #tear thing down 
          f.close() 
    def test(): 
      controlled_execution(output) 
    test() 

方法2：

用yield实现一个只产生一项的generator。通过for - in 来循环。

代码片段如下：

    #yield solution 
    def controlled_execution(): 
      f = None 
      try: 
        f = open(filename, 'r') 
        thing = f.read() 
        #for thing in f: 
        yield thing 
      except IOError,e: 
        print 'Error %s' % str(e) 
      finally: 
        if f:  
          f.close() 
    def test2(): 
      for content in controlled_execution(): 
        output(content) 

方法3：

用类的方式加上with实现。

代码片段如下：

    #class solution 
    class controlled_execution(object): 
      def __init__(self): 
        self.f = None 
      def __enter__(self): 
        try: 
          f = open(filename, 'r') 
          content = f.read() 
          return content 
        except IOError ,e: 
          print 'Error %s' % str(e) 
          #return None 
      def __exit__(self, type, value, traceback): 
        if self.f: 
          print 'type:%s, value:%s, traceback:%s' % \ 
              (str(type), str(value), str(traceback)) 
          self.f.close() 
    def test3(): 
      with controlled_execution() as thing: 
        if thing: 
          output(thing) 

方法4：

用with实现。不过没有exception handle 的功能。

    def test4(): 
      with open(filename, 'r') as f: 
        output(f.read()) 

      print f.read() 

最后一句print是用来测试f是否已经被关闭了。

最后总结一下，写这篇文章的目的主要是受了一句话的刺激："使用语言的好特性，不要使用那些糟糕的特性"！python真是有很多很优雅的好特性，路漫漫其修远兮，吾将上下而求索。。。