如何使用Subprocess执行Python脚本、传递输入并捕获输出？

利用子进程通过输入和捕获输出执行 Python 脚本

当任务是使用 subprocess 模块从另一个脚本调用 Python 脚本时，您可能会面临传递输入并在变量中获取所需输出的障碍。本文将深入探讨这些挑战的解决方案，使您能够有效地利用子流程的强大功能。

为了做好准备，请考虑一个场景，其中您有两个 Python 脚本：a.py 和 b.py。脚本 a.py 提示用户进行某些查询并生成 JSON 格式的输出。 B.py 的目标是调用 a.py，提供必要的输入，并捕获输出。

实现这一点的关键在于利用 subprocess 模块的 check_output 方法。此方法将包含 Python 可执行文件和 a.py 路径的命令列表作为输入。它将指定的输入（在本例中为一系列查询）提供给 a.py 并将其标准输出捕获为字符串。

以下是演示此方法的示例代码片段：

<code class="python">import os
import sys
from subprocess import check_output

script_path = os.path.join(get_script_dir(), 'a.py')
output = check_output([sys.executable, script_path],
                      input='\n'.join(['query 1', 'query 2']),
                      universal_newlines=True)</code>

或者，您可以选择更灵活的方法，从 b.py 导入模块 a 并调用 a.py 中的函数。然而，确保 a.py 使用 if __name__=="__main__" 保护来防止在导入时执行不需要的代码是至关重要的。

如果性能优化是您的主要关注点，请考虑利用多处理来分发您的跨多个进程的查询。当查询处理是 CPU 密集型时，此方法特别有用，因为它可以潜在地缩短执行时间。

以下代码片段说明了如何实现此目的：

<code class="python">from multiprocessing import freeze_support, Pool
import a

if __name__ == "__main__":
   freeze_support()
   pool = Pool() # use all available CPUs
   result = pool.map(a.search, ['query 1', 'query 2'])</code>

通过掌握这些技术，您将能够通过输入无缝执行 Python 脚本、捕获其输出，并利用子进程的强大功能来满足您的脚本需求。

以上是如何使用Subprocess执行Python脚本、传递输入并捕获输出？的详细内容。更多信息请关注PHP中文网其他相关文章！