假设有一个脚本执行以下操作:
# module writer.py
import sys
def write():
sys.stdout.write("foobar")
现在假设我要捕获write
函数的输出并将其存储在变量中以进行进一步处理。天真的解决方案是:
# module mymodule.py
from writer import write
out = write()
print out.upper()
但这是行不通的。我想出了另一个解决方案,它可以工作,但是请告诉我是否有解决问题的更好方法。谢谢
import sys
from cStringIO import StringIO
# setup the environment
backup = sys.stdout
# ####
sys.stdout = StringIO() # capture output
write()
out = sys.stdout.getvalue() # release output
# ####
sys.stdout.close() # close the stream
sys.stdout = backup # restore original stdout
print out.upper() # post processing