Python subprocess.Popen 输出如何捕获并存储？

在 Python 中捕获和存储 subprocess.Popen 输出

简介

Python 的 subprocess.Popen 函数是一个强大的工具，用于执行外部命令并捕获它们的输出。要将输出存储在字符串中，需要使用 communicate 方法，它将阻塞进程并返回一个元组，其中包含标准输出和标准错误。本文将分步指南，以及一些技巧和最佳实践，以帮助你有效地捕获和存储 subprocess.Popen 调用结果。

步骤

1. 创建子进程

import subprocess

p = subprocess.Popen("命令", stdout=subprocess.PIPE)

• stdout=subprocess.PIPE 将子进程的标准输出重定向到一个管道，以便稍后读取。

2. 捕获输出

output, err = p.communicate()

• communicate 方法将等待子进程完成，然后返回一个元组，其中包含子进程的标准输出 (output) 和标准错误 (err)。output 是一个字节串，包含子进程的标准输出。

3. 解码字节串

output = output.decode("utf-8")

• 由于 output 是一个字节串，因此需要将其解码为字符串。你可以根据需要选择适当的编码。

技巧和最佳实践

• 使用非阻塞调用： 如果你需要非阻塞地捕获输出，可以使用 subprocess.Popen.poll 方法。
• 处理错误： 检查 p.returncode 以确定子进程是否成功执行。
• 使用上下文管理器： 可以使用 with 语句作为上下文管理器，这将在进程完成后自动关闭管道。
• 使用通用新行： 在 Python 3.7 及更高版本中，可以使用 universal_newlines=True 参数来将 communicate 返回的字节串解码为字符串。

结论

通过遵循这些步骤，你可以有效地将 subprocess.Popen 调用结果存储在字符串中。这对于自动执行任务、解析输出或记录日志信息至关重要。

常见问题解答

1. 什么是 subprocess.PIPE？
   它将子进程的标准输出重定向到一个管道，以便你可以使用 communicate 方法读取输出。
2. communicate 方法做什么？
   它等待子进程完成并返回标准输出和标准错误。
3. 我应该如何解码字节串？
   使用 output.decode("utf-8") 或 universal_newlines=True 参数将字节串解码为字符串。
4. 如何检查子进程是否成功执行？
   检查 p.returncode 的值，如果为 0，则表示成功。
5. 如何非阻塞地捕获输出？
   使用 subprocess.Popen.poll 方法。