如何让 Python 信号处理器在子进程退出前可靠执行-Python教程-PHP中文网

如何让 Python 信号处理器在子进程退出前可靠执行

本文详解如何确保 python 信号处理器（如 sigint/sigterm）能完整执行清理逻辑（如终止并等待子进程），避免因信号被子进程继承导致主程序提前退出。核心方案是隔离信号——让子进程忽略关键信号，仅由父进程响应并优雅终止。

本文详解如何确保 python 信号处理器（如 sigint/sigterm）能完整执行清理逻辑（如终止并等待子进程），避免因信号被子进程继承导致主程序提前退出。核心方案是隔离信号——让子进程忽略关键信号，仅由父进程响应并优雅终止。

在 Python 中使用 signal.signal() 注册处理器时，一个常见却易被忽视的问题是：当子进程（如 7z、ffmpeg 等外部命令）与主进程同属一个进程组时，操作系统发送的信号（如 Ctrl+C 触发的 SIGINT）会同时送达父子进程。这导致子进程可能抢先响应信号而异常退出，或父进程的信号处理器尚未执行完 terminate() + wait() 就被强制终止，从而跳过资源清理，造成文件损坏、临时文件残留等问题。

✅ 正确做法：信号隔离（Signal Isolation）

关键原则是：让子进程忽略所有需由父进程处理的信号，仅父进程响应并协调终止流程。由于 execve() 会重置已捕获的信号为默认行为，但保留 SIG_IGN（忽略），因此最稳妥的方式是在子进程启动前显式忽略目标信号。

方案一：使用 shell 包装脚本（推荐，简洁可靠）

创建 7z.sh 脚本，屏蔽信号后执行实际命令：

#!/usr/bin/env bash
# 7z.sh —— 忽略信号，安全执行 7z
trap '' SIGTERM SIGINT SIGQUIT SIGHUP
exec "$@"

赋予可执行权限：chmod +x 7z.sh

Python 主程序中提前注册信号处理器，并调用该包装脚本：

Miller CSV TSV JSON 数据处理器

Miller (mlr) 是一个命令行工具，用于查询、整形和重新格式化名称索引数据，如 CSV、TSV、JSON 和 JSON Lines。它将 awk、sed、cut、join 和 sort 的功能整合到一个专为结构化数据处理而构建的单一工具中。

下载

import sys
import subprocess
import signal

zipProc = None  # 全局引用，便于 handler 访问

def signal_handler(sig, frame):
    print(f"Handling signal {sig}...")
    if zipProc and zipProc.poll() is None:  # 确保进程仍在运行
        zipProc.terminate()
        try:
            zipProc.wait(timeout=10)  # 设置超时，防止无限阻塞
        except subprocess.TimeoutExpired:
            print("Child process did not terminate gracefully; forcing kill.")
            zipProc.kill()
            zipProc.wait()
    print("Cleanup completed. Exiting.")
    sys.exit(0)

# ⚠️ 关键：必须在启动子进程前注册处理器！
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()  # 等待压缩完成（正常流程）

方案二：纯 Python 子进程（无需额外脚本）

若不想依赖 shell 脚本，可在 Python 子进程中调用 os.execv() 前主动忽略信号：

# launcher.py（作为子进程运行）
import os
import signal
import sys

# 忽略所有需父进程处理的信号
for sig in (signal.SIGTERM, signal.SIGINT, signal.SIGQUIT, signal.SIGHUP):
    signal.signal(sig, signal.SIG_IGN)

# 执行真正的命令
os.execv('/usr/bin/7z', ['7z', 'a', '-t7z', sys.argv[1], sys.argv[2]])

主程序改为：

# 启动 launcher.py 作为中间进程
cmd = [sys.executable, 'launcher.py', out_file, test_dir]
zipProc = subprocess.Popen(cmd)

⚠️ 注意事项与最佳实践

注册时机至关重要：signal.signal() 必须在 subprocess.Popen() 之前调用，否则信号可能在注册前到达。
避免全局变量陷阱：示例中将 zipProc 设为模块级变量以便 handler 访问；生产环境建议封装为类或使用 functools.partial 绑定参数。
wait() 需加超时：防止子进程僵死导致主程序永久挂起，应配合 timeout 参数及异常处理。
communicate() 与信号冲突：Popen.communicate() 是阻塞调用，会阻止信号立即触发 handler。若需响应中断，改用轮询 poll() 或 wait(timeout=...)。
跨平台兼容性：SIGHUP 在 Windows 不可用，建议按平台条件注册（if hasattr(signal, 'SIGHUP'):）。