程序完成时终止Python非守护后台进程

我完全懂你的困扰——当你有个跑着无限任务、还绑定了非守护线程池的子进程时，主进程退出它却赖着不走，这事儿确实比预想的棘手。下面给你几个靠谱的解决方案，从优雅到强硬依次排列：

方案1：给子进程加信号处理，优雅终止线程池 #

核心思路是让子进程能响应主进程的终止信号，主动停掉线程池并退出，避免硬杀带来的资源泄漏。

先在子进程代码里注册信号处理逻辑，收到终止信号时清理线程池：

import signal
from concurrent.futures import ThreadPoolExecutor
import time

def worker():
    while True:
        # 模拟无限循环的工作任务
        time.sleep(1)

def handle_exit(signum, frame):
    print("子进程收到终止信号，开始清理线程池...")
    executor.shutdown(wait=True)  # 等待所有正在执行的任务完成
    exit(0)

if __name__ == "__main__":
    global executor
    executor = ThreadPoolExecutor(max_workers=4)
    executor.submit(worker)
    
    # 注册终止信号的处理函数
    signal.signal(signal.SIGTERM, handle_exit)
    signal.signal(signal.SIGINT, handle_exit)
    
    # 让子进程保持运行（依赖线程池的非守护线程）
    while True:
        time.sleep(1)

然后在主进程里，用atexit注册清理函数，退出前给子进程发信号并等待：

import subprocess
import atexit
import time

proc = subprocess.Popen(["python", "child_process.py"])

def cleanup_child():
    print("主进程即将退出，终止子进程...")
    proc.terminate()
    # 最多等5秒让子进程优雅退出，超时就硬杀
    try:
        proc.wait(timeout=5)
    except subprocess.TimeoutExpired:
        print("子进程超时未退出，强制杀死...")
        proc.kill()

atexit.register(cleanup_child)

# 模拟主进程的业务逻辑
time.sleep(10)
print("主进程业务完成，准备退出")

方案2：把子进程的线程池改为守护线程 #

如果你的任务允许被强制终止（不需要优雅收尾），可以直接把线程池的线程设为守护线程。这样子进程的主线程一旦退出，所有守护线程会被自动终止，子进程也就跟着结束了。

修改子进程里的线程池创建代码：

# 创建带守护线程的线程池
executor = ThreadPoolExecutor(max_workers=4, daemon=True)

要是你用threading.Thread手动创建线程，记得给线程对象加daemon=True参数。

这种方式最简单，但缺点是任务会被直接打断，可能留下未关闭的文件、连接等资源，只适合无状态的轻量任务。

方案3：用multiprocessing替代subprocess（更可控） #

如果子进程本身是Python代码，用multiprocessing替代subprocess会更灵活——你可以通过队列/管道给子进程发送终止指令，完全不依赖系统信号，跨平台兼容性更好。

主进程代码示例：

from multiprocessing import Process, Queue
import time
import atexit

def child_process(stop_queue):
    from concurrent.futures import ThreadPoolExecutor
    
    def worker():
        while True:
            # 定期检查终止信号
            if not stop_queue.empty():
                break
            time.sleep(1)
    
    executor = ThreadPoolExecutor(max_workers=4)
    executor.submit(worker)
    
    # 等待主进程的终止指令
    stop_queue.get()
    executor.shutdown(wait=True)
    print("子进程优雅退出")

stop_queue = Queue()
proc = Process(target=child_process, args=(stop_queue,), daemon=False)
proc.start()

def cleanup():
    print("主进程退出，通知子进程终止...")
    stop_queue.put("stop")
    proc.join(timeout=5)
    if proc.is_alive():
        proc.kill()

atexit.register(cleanup)

# 主进程业务逻辑
time.sleep(10)
print("主进程结束")

额外注意事项 #

• atexit只在主进程正常退出时触发，如果主进程被SIGKILL（比如Linux下kill -9）强制终止，atexit不会执行。要覆盖这种极端情况，可以给主进程也注册SIGTERM/SIGINT的信号处理函数，确保清理逻辑执行。
• Windows系统上，subprocess.Popen.terminate()默认是强制终止，子进程的信号处理函数不会生效。如果要在Windows上实现优雅终止，需要用CTRL_C_EVENT，但要求子进程是控制台程序且能处理控制台信号。

内容的提问来源于stack exchange，提问作者Fabian Isensee