在python中捕获多线程的打印输出
我有一个函数，里面有多个并行运行的线程。线程打印一些东西，我想从外部函数中捕获这个输出。我尝试了下面的代码来捕获输出：
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)这段代码可以工作。但问题是，只有当所有theads都完成时，才会打印输出。所以我把它冻住了。一旦所有线程都完成了-整个输出都会被打印出来。我想要的-有输出，一旦它在内部生成-换句话说，在实时。现在，我一次打印出了整个输出缓冲区。如何让所有线程实时输出？
回答 1
Stack Overflow用户
发布于 2018-07-26 05:28:38
您想要做的是编写您自己的TextIOBase类(或者，如果您想要二进制数据，可以编写自己的RawIOBase，然后在它周围包装一个常用的TextIOWrapper )，您可以在其中放置您想要的任何行为。
正如您在the docs中看到的那样。您需要为TextIOBase实现的所有内容是detach、read、readline和write。前三个与你正在做的事情无关。
那么，你的write应该是什么样子的呢？嗯，这取决于你想做什么。
听起来你的目标是把所有东西都放到真正的stdout和StringIO上。如果是这样的话，这是相当微不足道的。
唯一的问题是，如果其中一个目标引发异常，或者写入的字节数比另一个目标少，等等。由于IOString永远不会做这些事情，我们可以编写一些非常愚蠢的东西，假设无论真正的stdout做了什么都是正确的事情。
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一起工作，比如您open的文件，而不仅仅是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)https://stackoverflow.com/questions/51527750
复制相似问题