python中sys.stdout、sys.stdin意思

今天突然见到两个特殊的sys的用法，一般用到这个是比较少的吧。

1 sys.stdout.write()与 print()

当我们在 Python 中打印对象调用 print(obj) 时候，事实上是调用了 sys.stdout.write(obj+'\n')

print 将你需要的内容打印到了控制台，然后追加了一个换行符

print 会调用 sys.stdout 的 write 方法

以下两行在事实上等价：

import sys 

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

print(hello)
           

2 sys.stdin.readline()与input

sys.stdin.readline( )会将标准输入全部获取，包括末尾的'\n'，因此用len计算长度时是把换行符'\n'算进去了的，但是input( )获取输入时返回的结果是不包含末尾的换行符'\n'的。

这个一般用不到：如果在平时使用sys.stdin.readline( )获取输入的话，不要忘了去掉末尾的换行符，可以用Python的字符串处理strip( )函数（sys.stdin.readline( ).strip('\n')）或sys.stdin.readline( )[:-1]这两种方法去掉

3 从控制台重定向到文件

原始的 sys.stdout 指向控制台

如果把写文件的对象（with open 的write）的引用赋给 sys.stdout，那么 print 调用的就是文件对象的 write 方法

import sys 
with open('sys_out.log', 'w') as h:

    sys.stdout=h

    print(hello)

           

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

# redirection start 
__console__=sys.stdout 

# 重定向到控制太输出
sys.stdout=__console__
           

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

如果我们希望打印的内容一方面输出到控制台，另一方面输出到文件作为日志保存，那么该怎么办？

将打印的内容保留在内存中，而不是一打印就将 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 也都可以被重定向到多个地址，举一反三的事情就自己动手实践吧