python的sys.stdout重定向

本文环境：Python 2.7
使用 print obj 而非 print(obj)

背景

（1）sys.stdout 与 print

当我们在 Python 中打印对象调用 print obj 时候，事实上是调用了 sys.stdout.write(obj+'\n')  
print 将你需要的内容打印到了控制台，然后追加了一个换行符  
print 会调用 sys.stdout 的 write 方法

以下两行在事实上等价：

sys.stdout.write('hello'+'\n')   
print 'hello'  

（2）sys.stdin 与 raw_input

当我们用 raw_input('Input promption: ') 时，事实上是先把提示信息输出，然后捕获输入 

以下两组在事实上等价：  

hi=raw_input('hello? ')   

print 'hello? ',                 #comma to stay in the same line   
hi=sys.stdin.readline()[:-1]     # -1 to discard the '\n' in input stream  

从控制台重定向到文件

原始的 sys.stdout 指向控制台  
如果把文件的对象的引用赋给 sys.stdout，那么 print 调用的就是文件对象的 write 方法  

f_handler=open('out.log', 'w')   
sys.stdout=f_handler   
print 'hello'  
# this hello can't be viewed on concole   
# this hello is in file out.log  

记住，如果你还想在控制台打印一些东西的话，最好先将原始的控制台对象引用保存下来，向文件中打印之后再恢复 sys.stdout  
__console__=sys.stdout   
# redirection start #   
...   
# redirection end   
sys.stdout=__console__  

同时重定向到控制台和文件

如果我们希望打印的内容一方面输出到控制台，另一方面输出到文件作为日志保存，那么该怎么办？  
将打印的内容保留在内存中，而不是一打印就将 buffer 释放刷新，那么放到一个字符串区域中会怎样？

a=''   
sys.stdout=a   
print 'hello'  
OK，上述代码是无法正常运行的  
Traceback (most recent call last): File   
".\hello.py", line xx, in print 'hello'   
AttributeError: 'str'   
object has no attribute 'write'  

错误很明显，就是上面强调过的，在尝试调用 sys.stdout.write() 的时候，发现没有 write 方法  
另外，这里之所以提示 attribute error 而不是找不到函数等等，我猜想是因为 python 将对象/类的函数指针记录作为对象/类的一个属性来对待，只是保留了函数的入口地址  
既然这样，那么我们必须给重定向到的对象实现一个 write 方法： 

import sys   
class __redirection__:   
  def __init__(self):   
    self.buff=''   
    self.__console__=sys.stdout   

  def write(self, output_stream):   
    self.buff+=output_stream   

  def to_console(self):   
    sys.stdout=self.__console__   
    print self.buff   

  def to_file(self, file_path):  
    f=open(file_path,'w')   
    sys.stdout=f   
    print self.buff   
    f.close()   

  def flush(self):   
    self.buff=''   

  def reset(self):   
    sys.stdout=self.__console__   

if __name__=="__main__":   
  # redirection   
  r_obj=__redirection__()   
  sys.stdout=r_obj  

  # get output stream   
  print 'hello'   
  print 'there'   

  # redirect to console   
  r_obj.to_console()   

  # redirect to file   
  r_obj.to_file('out.log')   

  # flush buffer  
  r_obj.flush()   

  # reset   
  r_obj.reset()  
同样的，sys.stderr, sys.stdin 也都可以被重定向到多个地址，举一反三的事情就自己动手实践吧