Python 標準輸出 sys.stdout 重新導向

標籤：obj   等價   most   init   last   tle   檔案   sel   input   

使用 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 linehi=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_handlerprint ‘hello‘ # this hello can‘t be viewed on concole# this hello is in file out.log

記住，如果你還想在控制台列印一些東西的話，最好先將原始的控制台對象引用儲存下來，向檔案中列印之後再恢複 sys.stdout

__console__=sys.stdout# redirection start# ...# redirection endsys.stdout=__console__

同時重新導向到控制台和檔案

如果我們希望列印的內容一方面輸出到控制台，另一方面輸出到檔案作為日誌儲存，那麼該怎麼辦？

將列印的內容保留在記憶體中，而不是一列印就將 buffer 釋放重新整理，那麼放到一個字串地區中會怎樣？

a=‘‘sys.stdout=aprint ‘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 sysclass __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 也都可以被重新導向到多個地址，舉一反三的事情就自己動手實踐吧

Python 標準輸出 sys.stdout 重新導向