如何检测Python脚本是否通过subprocess子进程启动

核心问题定位 #

之前自定义环境变量检测方案失效，基本都是subprocess.Popen调用时env参数配置错误导致的：直接传入仅包含自定义变量的字典会覆盖子进程的全部系统环境，不仅会让自定义变量检测逻辑异常，还可能触发Python运行时的路径、编码类问题。

推荐实现：环境变量标记法（可控性最高，无额外依赖） #

第一步：修正GUI侧的subprocess调用逻辑 #

在main_gui.py中调用resize.py时，基于当前系统环境副本追加自定义标记变量，不要直接新建空环境字典：

# main_gui.py 子进程调用段
import subprocess
import os

# 复制当前进程的系统环境变量，保证子进程基础运行环境正常
run_env = os.environ.copy()
# 追加双方约定的标记变量，值可自定义，两边保持一致即可
run_env["RESIZE_RUN_MODE"] = "gui"

process = subprocess.Popen(
    ["python", "resize.py", 输入文件夹路径, 输出文件夹路径],
    stdout=subprocess.PIPE,
    stderr=subprocess.STDOUT,
    text=True,  # 直接返回字符串格式输出，省去字节解码步骤
    env=run_env  # 传入拼接完成的环境变量
)

第二步：resize.py侧加模式判断 #

脚本启动时直接读取环境变量做分支判断即可：

# resize.py 头部初始化段
import os

# 判定当前运行模式：GUI调用模式/终端直接启动模式
IS_GUI_MODE = os.getenv("RESIZE_RUN_MODE") == "gui"

第三步：分模式隔离输出逻辑 #

• 若IS_GUI_MODE为True：不初始化终端进度条组件，每完成一个处理任务就打印带约定特殊前缀的进度信息，必须加flush=True强制刷新输出缓冲区，避免GUI侧读不到实时输出：

  # 示例：GUI模式下的进度输出
  print(f"__TOTAL_TASKS__:{total_img_count}", flush=True)
  for idx, img in enumerate(img_list):
      # 执行图片缩放逻辑
      process_img(img)
      print(f"__CURRENT_PROGRESS__:{idx+1}", flush=True)
  

• 若IS_GUI_MODE为False：正常初始化progress.bar终端进度条，不打印任何带特殊标识的进度语句，避免终端出现多余乱码内容。

兜底方案：父进程检测法（无需修改GUI侧代码） #

如果暂时无法调整GUI侧的传参逻辑，可以通过检测父进程命令行判断启动来源，需要额外安装psutil依赖：

# resize.py 模式判断逻辑，需先执行 pip install psutil
import os
import psutil

def check_run_mode():
    try:
        parent = psutil.Process(os.getppid())
        parent_cmd = " ".join(parent.cmdline())
        # 匹配父进程命令中是否包含GUI脚本名
        return "main_gui.py" in parent_cmd
    except:
        # 检测异常时默认走终端直接启动模式，避免影响正常使用
        return False

IS_GUI_MODE = check_run_mode()

注意：该方案鲁棒性较差，若GUI脚本重命名、被打包成exe、或通过其他脚本嵌套调用就会检测失效，仅作为临时兜底方案使用。

常见踩坑说明 #

• 不要在两种模式下混用输出逻辑：终端进度条依赖特殊控制字符实现原地刷新，和普通print语句同时使用会导致终端进度条错位、重复输出多行
• GUI侧读取子进程输出时，要按行读取解析，匹配到约定的特殊前缀再更新进度条组件，不要随意拼接输出片段导致解析失败
• 不要给子进程设置shell=True，否则环境变量传递逻辑可能出现跨平台兼容问题。

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