Python:在个人标准输出上编辑文本 - 双文本

Question

我需要将控制台中显示的日志保存到文件中，因此我编辑了 sys.stdout。 (使用来自 Redirect stdout to a file in Python? 的代码)

但是当我试图通过在它之前添加一些东西来编辑写入函数中的文本时，问题就出现了。结果这个“[stack]”在文本变量前后添加。

import sys

class Logger(object):
    def __init__(self):
        self.terminal = sys.stdout
        self.file = open("log.txt", "a")
    def flush(self):
        self.terminal.flush()
        self.file.flush()
    def write(self, text):
        self.terminal.write("[stack]" + text)
        self.file.write(text)
        self.flush();

sys.stdout = Logger()

print "Test log"
print "Another test log"

结果:

[stack]Test log[stack]
[stack]Another test log[stack]

Answer 1

我一直在为这个问题绞尽脑汁，直到我认为我应该通过调试器运行它。发生这种情况的原因是，当您使用 print 时，它会尝试同时写入文本正文和配置的 end，默认情况下是换行符.

因此，每个 print 语句都会导致对 Logger.write 的两次单独调用，一次是(例如)Test Log，第二次是是 \n。这导致输出 [stack] Test Log[stack]\n。

这里是一个更正的实现:

import sys


class Logger(object):
    def __init__(self, stream, default_sep=' ', default_end='\n'):
        self.terminal = stream
        self.default_sep = default_sep
        self.default_end = default_end
        self.continuing_same_print = False
        self.file = open("log.txt", "a")

    def flush(self):
        self.terminal.flush()
        self.file.flush()

    def write(self, text):
        if text is self.default_end:
            self.continuing_same_print = False
        elif text is self.default_sep:
            self.continuing_same_print = True

        new_text = text
        if text in {self.default_sep, self.default_end}:
            pass
        elif self.continuing_same_print:
            pass
        else:
            new_text = '[stack]' + new_text

        self.terminal.write(new_text)
        self.file.write(text)
        self.flush()


sys.stdout = Logger(sys.stdout)

print("Test", "log")
print("Another test log")
print()

输出

[stack]Test log
[stack]Another test log

编辑

更新了实现以支持在打印语句中打印多个对象。