将Python3控制台输出保存到文件的方法

这个需求也是很简单,有时候我们可以通过控制台输出一些有用的结果,例如用pptree:https://github.com/clemtoy/pptree

但是我们往往也希望把控制台的输入保存到文件供具体分析,参考了一些帖子和博客:

https://www.52pojie.cn/thread-959203-1-1.html

https://www.cnblogs.com/ranjiewen/p/9943111.html

综合以上的讨论,其实在自己的Python文件最开头加上这样一段代码就可以了:

import sys
class Logger(object):
    def __init__(self, filename='default.log', stream=sys.stdout):
        self.terminal = stream
        self.log = open(filename, 'w')

    def write(self, message):
        self.terminal.write(message)
        self.log.write(message)

    def flush(self):
        pass
   
sys.stdout = Logger('a.log', sys.stdout)
sys.stderr = Logger('a.log_file', sys.stderr)

确实很方便有木有。

你可能感兴趣的:(Python技巧)