Python subprocess调用外部程序偶发卡死 如何检测进程状态并终止

基于Python subprocess模块循环调用外部exe程序的常规实现代码如下：

import subprocess

arg_list = ['a', 'b', 'c', 'd']
for arg in arg_list:
    proc = subprocess.Popen(['some_app.exe', arg])
    proc.communicate()

该实现存在明确卡点：

• 多数场景下子进程可正常退出，脚本按顺序执行无异常
• 偶发场景下被调用的exe会弹出带「OK」按钮的模态弹窗，阻塞进程退出逻辑，导致脚本永久卡住
• 不同入参对应的子进程正常运行时长差异极大，无法通过设置固定超时阈值规避卡死问题
• 核心需求：不依赖超时规则，精准识别子进程是否处于「等待用户点击弹窗按钮」的挂起状态，识别后主动终止进程即可跳过异常卡点

方案1：Windows平台下枚举子进程窗口属性识别弹窗 #

该方案适配Windows平台运行GUI exe的场景，识别准确率最高，无需猜测运行时长：

• 核心逻辑：正常执行业务逻辑的进程不会创建顶层可见的模态对话框，仅在弹出提示框等待用户交互时，才会生成带交互按钮的窗口类
• 实现逻辑：基于pywin32库调用Win32 API，枚举目标进程关联的所有顶层窗口，判断是否存在属于该进程的、可见的、包含标准OK按钮控件的对话框窗口
• 参考实现代码：

import subprocess
import time
import win32gui
import win32process

def has_blocking_dialog(pid):
    """检测指定pid的进程是否存在阻塞式OK弹窗"""
    dialog_exists = False
    def enum_callback(hwnd, _):
        nonlocal dialog_exists
        # 过滤不可见窗口
        if not win32gui.IsWindowVisible(hwnd):
            return
        # 匹配窗口所属进程
        _, win_pid = win32process.GetWindowThreadProcessId(hwnd)
        if win_pid != pid:
            return
        # 标准对话框OK按钮的控件ID为2，检测是否存在可见的OK按钮
        ok_btn_handle = win32gui.GetDlgItem(hwnd, 2)
        if ok_btn_handle and win32gui.IsWindowVisible(ok_btn_handle):
            dialog_exists = True
    win32gui.EnumWindows(enum_callback, None)
    return dialog_exists

arg_list = ['a', 'b', 'c', 'd']
for arg in arg_list:
    proc = subprocess.Popen(['some_app.exe', arg])
    # 轮询检测，间隔可根据实际场景调整，避免过高占用资源
    while proc.poll() is None:
        if has_blocking_dialog(proc.pid):
            # 识别到弹窗直接终止进程
            proc.terminate()
            time.sleep(0.5)
            # 强杀兜底，防止terminate调用不生效
            if proc.poll() is None:
                proc.kill()
            break
        time.sleep(1)
    proc.communicate()

• 适配说明：如果目标程序使用自定义皮肤的非标准对话框，只需提取对应弹窗的窗口类名、特征控件属性，调整判断规则即可，识别可靠性远高于超时方案。

方案2：基于进程CPU占用特征辅助判断 #

如果不想调用Win32层API，可以使用该轻量跨平台方案：

• 核心逻辑：程序正常执行业务计算、IO操作时，CPU占用会维持在非0的波动水平；弹出模态弹窗等待用户输入时，进程会进入消息等待挂起状态，CPU占用会连续较长时间（如连续10秒以上）维持在0%附近
• 实现逻辑：基于psutil库定期采集子进程的CPU使用率，连续多次采集到CPU使用率低于阈值（如<0.1%）时，即可判定为处于等待交互的阻塞状态
• 方案优缺点：无需适配GUI窗口逻辑，跨平台兼容性好；缺点是存在极低概率误判（如进程刚好等待耗时极长的IO操作），可满足大部分非严格场景需求。

方案3：拦截弹窗自动点击，无需终止进程 #

如果不需要终止异常进程，可以在识别到OK弹窗时，直接向按钮发送点击消息，让进程继续走完流程自动退出，逻辑比强制杀进程更温和。实现逻辑和方案1一致，仅需将终止进程的代码替换为向OK按钮句柄发送点击消息即可。

注意：不要尝试通过重定向子进程标准输入输出流规避弹窗，这类模态弹窗属于Win32 GUI层控件，和控制台IO流完全独立，重定向流不会影响弹窗弹出逻辑。

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