如何在 Python 中捕获脚本 stdout 输出？

从 Python 脚本中捕获 stdout 输出

使用外部脚本时，通常需要捕获其 stdout 输出以进行进一步处理。一种常见的方法是使用 write() 函数打印输出，然后将其存储在变量中。然而，这种方法并不总是有效，如以下脚本所示：

<code class="python"># module writer.py
import sys

def write():
    sys.stdout.write("foobar")</code>

为了捕获 write() 函数的输出，提出了涉及 cStringIO 模块的解决方案：

<code class="python">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</code>

但是，还有更高效、Pythonic 的方法来实现此目标：

使用 contextlib.redirect_stdout (Python 3.4 )

此方法利用 contextlib .redirect_stdout 上下文管理器临时将 stdout 输出重定向到 StringIO 对象：

<code class="python">from contextlib import redirect_stdout
import io

f = io.StringIO()
with redirect_stdout(f):
    help(pow)
s = f.getvalue()</code>

直接使用 io.StringIO

另一种方法是直接创建 StringIO 对象并手动将 stdout 输出路由到它：

<code class="python">import io

output = io.StringIO()
sys.stdout = output
write()
sys.stdout.seek(0)
captured_output = output.read()</code>

通过实现这些方法，开发人员可以有效地捕获和处理脚本输出，而无需诉诸复杂的解决方法。

以上是如何在 Python 中捕获脚本 stdout 输出？的详细内容。更多信息请关注PHP中文网其他相关文章！