python - 从python中的多个线程捕获打印输出

Question

我有一个函数可以在内部启动多个并行运行的线程。线程打印一些东西，我想从外部函数捕获这个输出。我尝试了下一个代码来捕获输出:

import sys, io
stdout = sys.stdout
sys.stdout = io.StringIO()
threads_conn(connect, devices) #- here many threads starts with many print inside
output = sys.stdout.getvalue()
sys.stdout = stdout

print(output)

此代码有效。但问题是只有当所有的广告都完成时才会打印输出。所以我卡住了。一旦所有线程都完成 - 打印整个输出。我想要的 - 一旦在内部生成就可以输出 - 换句话说是实时的。现在我立即打印了整个输出缓冲区。如何让所有线程实时输出？

Answer 1

你想要做的是编写你自己的 TextIOBase 类(或者，如果你想要二进制数据，编写你自己的 RawIOBase 然后包装一个普通的 TextIOWrapper 围绕它)，您可以在其中放置您想要的任何行为。

从the docs可以看出.您需要为 TextIOBase 实现的是detach、read、readline 和write。前三个与您正在做的事情无关。

---

那么，您的write 应该是什么样的？嗯，这取决于你想做什么。

听起来您的目标是将所有内容都发送到真正的标准输出和 StringIO。如果是这样，这是非常微不足道的。

唯一的问题是，如果其中一个目标引发异常，或者写入的字节数比另一个少，等等，你想做什么。因为 IOString 永远不会做这些中的任何一个，我们可以编写一些非常愚蠢的东西，它只是假设真正的标准输出所做的一切都是正确的。

class TeeTextIO(io.TextIOBase):
    def __init__(self, target):
        self.target = target
        self.stringio = io.StringIO()
    def write(self, s):
        writecount = self.target.write(s)
        self.stringio.write(s[:writecount])
        return writecount

现在:

stdout = sys.stdout
sys.stdout = TeeTextIO(sys.stdout)
threads_conn(connect, devices) #- here many threads starts with many print inside
output = sys.stdout.stringio.getvalue()
sys.stdout = stdout

现在，输出已经进入真正的 stdout，但它也被存储在 StringIO 中，供您以后使用。

(请注意，此类将与任何 TextIOBase 一起使用，就像您打开 的文件一样，而不仅仅是与stdout。我们没有花任何钱就可以让它通用，所以为什么不呢？)

---

如果您想做一些完全不同的事情怎么办，比如将每个 write 随机分布在 10 个不同的文件中？应该很明显:

class SpreadTextWriter(io.TextIOBase):
    def __init__(self, *files):
        self.files = files
    def write(self, s):
        return random.choice(self.files).write(s)