2020-06-13-print语句重定向和同时输出到屏幕

今天耗时近3个小时解决这个问题，在网上基本上是两种方法，一个使用logging模块，一个是sys的标准输出到日志。我前面已经写了很多print语句，这个时候不想更改代码，更不想重新学习新的模块，所以只想加几句代码在脚本中，不影响我代码整体。终于找到如下方法啦！不小心关掉原文链接，找不到出处，大家将就~先上代码

class Logger(object):
    def __init__(self, fileN="Default.log"):
        self.terminal = sys.stdout
        self.log = open(fileN, "w")
    def write(self, message):
        self.terminal.write(message)
        self.log.write(message)
        self.flush()
    def flush(self):
        self.log.flush()
sys.stdout = Logger("PredictionTransmembrane.log") ##只需要在这里写入输出日志名称即可边打印变输出啦。

小小的记录一下，不码太多字啦~