Python 标准输出 sys.stdout 重定向

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Python 标准输出 sys.stdout 重定向相关的知识,希望对你有一定的参考价值。

使用 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 也都可以被重定向到多个地址,举一反三的事情就自己动手实践吧

以上是关于Python 标准输出 sys.stdout 重定向的主要内容,如果未能解决你的问题,请参考以下文章

Python 处理标准输出

python -u

python print · sys.stdout · sys.stderr

python有关改变标准输出的默认编码的问题

sys.stdout.flush() 方法的使用

Python 3 TypeError: must be str, not bytes with sys.stdout.write()