当我们使用print(obj)在console上打印对象的时候,实质上调用的是sys.stdout.write(obj+'\n'),print在打印时会自动加个换行符,以下两行等价:
sys.stdout.write('hello'+'\n')
print('hello')
从控制台重定向到文件
原始的 sys.stdout 指向控制台,如果把文件的对象的引用赋给 sys.stdout,那么 print 调用的就是文件对象的 write 方法,将对象写入文件中:
f_handler=open('out.log','w')
sys.stdout=f_handler
print('hello')
如果你还想在控制台打印一些东西的话,最好先将原始的控制台对象引用保存下来,向文件中打印之后再恢复 sys.stdout
__console__=sys.stdout
# redirection start
# ...
# redirection end
sys.stdout=__console__
同时重定向到控制台和文件
如果我们希望打印的内容一方面输出到控制台,另一方面输出到文件作为日志保存,那么该怎么办?将打印的内容保留在内存中,而不是一打印就将 buffer 释放刷新,我们可以通过自定义一个类来实现这个功能:
import os,sys
class redirection():
def __init__(self):
self.buf = ''
self.__console__ = sys.stdout
self.path = 'C:\\Users\\***\\Desktop' # 存放日志的路径
def write(self,output_stream):
self.buf += output_stream
def to_log(self,filename):
file = open(self.path + '\\' + filename,'a')
sys.stdout = file
print(self.buf)
file.close
def to_console(self):
sys.stdout = self.__console__
print(self.buf)
def flush(self):
self.buf = ''
def main():
obj = redirection()
input_stream = input("输入:")
while input_stream: # 实现循环输入
obj.write(input_stream + '\n')
input_stream = input("输入:")
obj.to_log('log.txt') # 日志名称为log.txt
obj.to_console()
obj.flush()
if __name__ == "__main__":
main()
运行结果:
=================== RESTART: C:/Users/***/Desktop/tets.py ===================
输入:abcd
输入:12345
输入:hello
输入:
abcd
12345
hello
此时打开log.txt文件,会发现输出到console的三行文字也已经被写入。
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。