Python 標准輸出 sys.stdout 重定向（轉）

add by zhj: 其實很少使用sys.stdout，之前django的manage.py命令的源碼中使用了sys.stdout和sys.stderr，所以專門查了一下

這兩個命令與print的區別，發現其實沒多大區別，用print就好了

 

原文：http://www.cnblogs.com/turtle-fly/p/3280519.html

本文環境：Python 2.7 

使用 print obj 而非 print(obj)

 

一些背景

sys.stdout 與 print

當我們在 Python 中打印對象調用 print obj 時候，事實上是調用了 sys.stdout.write(obj+'\n')

print 將你需要的內容打印到了控制台，然后追加了一個換行符

print 會調用 sys.stdout 的 write 方法

以下兩行在事實上等價：

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

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 <module>
    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 也都可以被重定向到多個地址，舉一反三的事情就自己動手實踐吧