今天突然见到两个特殊的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 也都可以被重定向到多个地址,举一反三的事情就自己动手实践吧