python 等待所有子进程结束

原创

mob64ca12db7156 2024-09-13 04:35:35 ©著作权

文章标签 子进程 Python python 文章分类 Python 后端开发

©著作权归作者所有：来自51CTO博客作者mob64ca12db7156的原创作品，请联系作者获取转载授权，否则将追究法律责任

如何实现 Python 等待所有子进程结束

在 Python 编程中，处理并发执行的子进程是一项常见的任务。我们经常需要启动多个子进程，并在所有子进程完成后继续执行主程序。在这篇文章中，我们将详细讨论如何使用 Python 实现“等待所有子进程结束”的目标。

完整的流程概述

在实现这个功能之前，让我们先梳理一下整个过程。以下是实现“等待所有子进程结束”的步骤：

步骤	描述	代码示例
1	导入需要的模块	`import subprocess`
2	启动子进程并将其保存在列表中	`process = subprocess.Popen(...)`
3	等待所有子进程结束	`for proc in process_list:`
4	处理子进程的返回结果	`proc.wait()`
5	总结任务或清理	`print("All subprocesses finished.")`

实现步骤详解

1. 导入需要的模块

在开始之前，我们需要导入 Python 的 subprocess 模块。这个模块提供了许多用于创建和管理子进程的功能。

import subprocess  # 导入subprocess模块，以便我们可以使用它来启动和管理子进程

2. 启动子进程并将其保存在列表中

我们可以使用 subprocess.Popen() 来启动子进程，并将所有子进程存储到一个列表中，以便后续处理。

# 创建一个空列表来保存所有创建的子进程
process_list = []

# 启动子进程
for i in range(5):  # 假设我们启动5个子进程
    process = subprocess.Popen(["python", "-c", f"print('Subprocess {i}')"])  
    # 使用Popen启动一个新的子进程，并执行一个简单的打印任务
    process_list.append(process)  # 将子进程添加到列表中

3. 等待所有子进程结束

一旦所有子进程被启动，我们需要等待它们完成。在 Python 中，我们可以遍历我们的进程列表并调用 wait() 方法。

# 等待所有子进程结束
for proc in process_list:
    proc.wait()  # 等待每个子进程结束

4. 处理子进程的返回结果

在所有子进程完成后，你可能想处理每个子进程的返回值（如果有）。Popen 对象有一个 returncode 属性，可以帮助我们获得返回结果。

# 处理每个子进程的返回结果
for proc in process_list:
    print(f"Subprocess finished with return code: {proc.returncode}")  
    # 输出每个子进程结束时的返回码

5. 总结任务或清理

最后，你可以添加清理代码或打印一条消息以确认所有任务已完成。

print("All subprocesses finished.")  # 打印所有子进程完成的消息

甘特图表示

下面是一个甘特图，展示了启动和等待子进程的过程：

gantt
    title Python 等待所有子进程结束的流程
    dateFormat  YYYY-MM-DD
    section 启动子进程
    启动子进程          :a1, 2023-10-20, 1d
    section 等待子进程
    等待子进程结束      :after a1  , 2023-10-21, 1d