Python：修改内置打印功能

Question

我想在 Python 中自定义 print 语句以获取更多文本。但是按照我的方法，似乎 Enter 键正在输入中缓冲。

我使用的程序是：

class rename_print:
    def __init__(self, stdout):
        self.stdout = stdout     
    def write(self, text):
        self.stdout.write('###' + text)     
        self.stdout.flush()
    def close(self):
        self.stdout.close()

import sys

prints = rename_print(sys.stdout)
sys.stdout = prints
print 'abc'

我得到的输出是

###abc###

我期望的输出是

###abc

这可能是什么原因？我怀疑输入流是否被 Enter 键缓冲。我该如何解决这个问题？

Answer 1

print 在默认情况下（或每个参数之间的空格）将换行符写入输出流。所以你会接到两个电话给write，一个是"abc"，一个是"\n"。

因此，如果您不想要这种行为，则必须单独处理该调用。

Answer 2

我认为正在发生的事情是 print 隐式添加了换行符。这个额外的打印也调用了你的重定向写入函数，所以你得到另一个“###\n”

这有点hacky，但试试这个：

...

def write(self, text):
    if text!="\n":
       self.stdout.write('###' + text)

...