python - 动态重定向函数的标准输出

Question

假设我有一个 python 函数 foo。我想在后台运行它(使用 threading 或 multiprocessing)，并使其所有 stdout 和 stderr 实时可用，因为它是由 foo< 生成的。如果 foo 是一个 shell 命令，我可以通过使用 subprocess.Popen 和生成器来实现这一点，如 here 中所述。 ，但我需要使用 python 函数来做到这一点。 multiprocessing 模块非常适合生成新进程，但似乎很难实时获取子进程的标准输出。

P.S:我需要使用服务器发送事件协议(protocol)将函数的标准输出重定向到生成的网页

Answer 1

就其值(value)而言，我认为这是一个坏主意，理想情况下您应该重写 foo 函数以以其他方式返回其输出。也就是说...

如果foo只是一个标准的纯Python函数，仅使用标准Python函数来生成其输出(print和file.write) ，那么您可以将其标准输出重定向到 StringIO 或类似的东西。

Python 2.x 示例:

import sys, StringIO

def foo():
    print "Haha! You can't capture this output!"
    print>>sys.stderr, "... Or this!"

output = StringIO.StringIO()

saved, sys.stdout = sys.stdout, output
sys.stderr = output
foo()
sys.stdout = sys.stderr = saved

print output.getvalue()