如何让 Python 信号处理器在进程退出前完成清理工作-Python教程-PHP中文网

如何让 Python 信号处理器在进程退出前完成清理工作

本文详解为何 Python 中的 signal handler 在子进程运行时无法可靠执行，以及如何通过隔离信号、重定向子进程信号处理机制（如使用 bash 脚本屏蔽信号）来确保终止逻辑（如 terminate() + wait()）完整执行。

本文详解为何 python 中的 signal handler 在子进程运行时无法可靠执行，以及如何通过隔离信号、重定向子进程信号处理机制（如使用 bash 脚本屏蔽信号）来确保终止逻辑（如 terminate() + wait()）完整执行。

在 Python 中注册信号处理器（如 signal.signal(SIGINT, handler)）后，仍可能遇到“按下 Ctrl+C 后 handler 未执行完、主程序就已退出”的问题。根本原因在于：默认情况下，终端发送的信号（如 SIGINT）会广播给整个进程组（process group），即父进程（Python）和子进程（如 7z）同时收到信号。若子进程未忽略这些信号，它可能抢先响应并异常退出，导致父进程的信号处理器来不及执行资源清理（如终止子进程、等待其结束），甚至因 zipProc 已失效而抛出异常。

✅ 正确做法：隔离信号作用域

核心思路是：让子进程忽略关键信号，仅由父进程（Python）负责捕获与响应。由于 execve() 会重置被处理（handled）的信号为默认行为，但保留被忽略（ignored）的信号状态，因此我们需在子进程启动前显式将其对 SIGTERM/SIGINT/SIGQUIT/SIGHUP 的处置设为 SIG_IGN。

最简洁可靠的实现方式是引入一个轻量级 shell 包装脚本：

提示词大师-python版

图片提示词生成器？不止如此。马甲系统 —— 把脑海中的画面，翻译成AI能理解的专业表达。用得越多，它越懂你：首次需要多问几句确认方向，用久了几乎一说就懂。用得越多，它越快：缓存机制让后续对话越来越省。 RAG进化：成功案例持续入库，越跑越聪明。输入「新手指南」查看完整功能介绍

下载

#!/usr/bin/env bash
# 文件名：7z.sh，赋予可执行权限：chmod +x 7z.sh
trap '' SIGTERM SIGINT SIGQUIT SIGHUP  # 子进程忽略所有目标信号
exec "$@"  # 安全执行原始命令（避免 shell 注入）

然后在 Python 主程序中：

提前注册信号处理器（必须在 Popen 之前！否则信号可能在注册前到达）；
调用该包装脚本启动子进程；
在 handler 中安全执行终止与同步逻辑。

import sys
import subprocess
import signal

def signal_handler(sig, frame):
    print(f"Handling signal {sig}...")
    if 'zipProc' in globals() and zipProc.poll() is None:  # 确保进程仍在运行
        zipProc.terminate()
        try:
            zipProc.wait(timeout=10)  # 避免无限等待，加超时更健壮
        except subprocess.TimeoutExpired:
            zipProc.kill()
            zipProc.wait()
    print("Cleanup completed.")
    sys.exit(0)

# ✅ 关键：信号处理器必须在 fork 之前注册！
for sig in [signal.SIGTERM, signal.SIGINT, signal.SIGQUIT, signal.SIGHUP]:
    signal.signal(sig, signal_handler)

test_dir = '/mnt/Z/testdir'
out_file = '/mnt/Z/testdir.7z'
cmd = ['./7z.sh', '7z', 'a', '-t7z', out_file, test_dir]  # 使用包装脚本
zipProc = subprocess.Popen(cmd)
zipProc.communicate()  # 等待压缩完成

⚠️ 注意事项与最佳实践

注册时机至关重要：signal.signal() 必须在 subprocess.Popen() 之前调用，否则子进程可能在 handler 安装前就收到信号。
避免信号竞争：不要依赖 zipProc 全局变量（推荐封装为类属性或使用 atexit + threading.Event 增强鲁棒性）；
超时保护：wait() 应设 timeout 参数，防止子进程僵死导致主程序挂起；
替代方案：若不愿引入外部脚本，可在子进程中用 Python 启动（python -c "import signal; signal.signal(...); execv(...)"），但需手动设置 signal.SIG_IGN 并 os.execv，复杂度更高；
调试技巧：handler 中添加 print() 日志，并用 strace -f python script.py 观察信号实际投递对象。