python - 写入多个流的包装器

Question

在 python 中，是否有一种简单的方法来设置一个类似文件的对象来写入，该对象实际上由多个输出流支持？例如，我想要这样的东西:

file1 = open("file1.txt", "w")
file2 = open("file2.txt", "w")
ostream = OStreamWrapper(file1, file2, sys.stdout)

#Write to both files and stdout at once:
ostream.write("ECHO!")

所以我正在寻找的是OStreamWrapper。我知道自己编写会很容易，但如果有现成的，我宁愿使用它，而不必担心查找和覆盖边缘情况。

Answer 1

class OStreamWrapper(object):

    def __init__(self, *streams):
        self.streams = list(streams)

    def write(self, string):
        for stream in self.streams:
            stream.write(string)

    def writelines(self, lines):
        # If you want to use stream.writelines(), you have
        # to convert lines into a list/tuple as it could be
        # a generator.
        for line in lines:
            for stream in self.streams:
                stream.write(line)

    def flush(self):
        for stream in self.streams:
            stream.flush()