Python 如何在父进程结束时终止子进程

在本文中，我们将介绍如何在Python中实现在父进程结束时终止子进程的方法。

1. 使用`atexit`模块

atexit模块提供了在程序退出时执行特定操作的功能。我们可以利用这个模块来终止子进程。下面是一个示例：

import atexit
import os
import signal

def cleanup():
    # 终止子进程
    os.kill(child_pid, signal.SIGTERM)

# 注册cleanup函数，在程序退出时执行
atexit.register(cleanup)

# 在父进程中启动子进程
child_pid = os.fork()
if child_pid == 0:
    # 在子进程中执行任务
    # ...

# 在父进程中继续执行其他操作
# ...

在上面的示例中，我们定义了一个cleanup函数，该函数通过os.kill()和signal.SIGTERM信号终止子进程。然后，我们使用atexit.register()注册了这个函数，在程序退出时执行。

2. 使用`subprocess`模块

Python的subprocess模块提供了更高级的接口来创建子进程，并可以方便地管理子进程的生命周期。我们可以使用subprocess.Popen类来创建子进程，并通过Popen对象的terminate()方法来终止子进程。下面是一个示例：

import subprocess
import os
import signal

# 在父进程中启动子进程
child_proc = subprocess.Popen(["python", "child.py"])

# 在父进程退出时终止子进程
def cleanup():
    child_proc.terminate()

# 注册cleanup函数，在程序退出时执行
atexit.register(cleanup)

# 在父进程中继续执行其他操作
# ...

在上面的示例中，我们使用subprocess.Popen创建了一个子进程，并将子进程的句柄存储在child_proc变量中。然后，我们定义了一个cleanup函数，使用terminate()方法终止子进程。最后，我们使用atexit.register()注册了这个函数，在程序退出时执行。

3. 使用`psutil`模块

psutil是一个第三方库，提供了对系统进程和系统资源的访问功能。我们可以使用psutil来获取和操作进程信息，包括终止子进程。下面是一个示例：

import psutil

# 在父进程中启动子进程
child_proc = psutil.Popen(["python", "child.py"])

# 在父进程退出时终止子进程
def cleanup():
    child_proc.terminate()

# 注册cleanup函数，在程序退出时执行
atexit.register(cleanup)

# 在父进程中继续执行其他操作
# ...

在上面的示例中，我们使用psutil.Popen创建了一个子进程，并将子进程的句柄存储在child_proc变量中。然后，我们定义了一个cleanup函数，使用terminate()方法终止子进程。最后，我们使用atexit.register()注册了这个函数，在程序退出时执行。

总结

本文介绍了几种方法来实现在父进程结束时终止子进程的功能。我们可以使用atexit模块注册一个清理函数，在程序退出时调用该函数终止子进程；也可以使用subprocess模块的Popen对象的terminate()方法来终止子进程；还可以使用psutil模块获取和操作进程信息来终止子进程。根据实际需求，我们可以选择合适的方法来管理子进程的生命周期，确保在父进程结束时终止子进程。