将print输出的内容保存到txt文件中

# 将print输出的内容保存到txt文件中

class Logger(object):
    def __init__(self, filename=os.path.join(base_path,'log/Default.log')):
        self.terminal = sys.stdout
        self.log = open(filename, "a", encoding='utf-8')

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

    def flush(self):
        pass

sys.stdout = Logger(os.path.join(base_path,'log/main_log.txt'))