如何防范Python multiprocessing炸弹？模块内多进程安全使用方案

这个坑我踩过好几次了——Windows系统的multiprocessing机制和Linux/macOS不一样（它用spawn而不是fork），如果模块顶级代码里直接启动进程，又没加if __name__ == "__main__"防护，就会无限递归创建子进程，也就是所谓的"炸弹"。下面给你两种场景的解决方案：

一、在模块内检测并终止"炸弹" #

当然可以！核心思路是判断当前进程是不是主进程，或者当前模块是不是被作为主脚本运行，一旦检测到是子进程被重新导入的场景，直接退出阻止后续逻辑执行。

方案1：通过__name__和系统平台判断 #

在模块的最开头（任何启动进程的代码之前）加入这段代码：

import sys

if sys.platform == "win32" and __name__ != "__main__":
    # 这是Windows下子进程被spawn时的导入场景，直接退出
    sys.exit(0)

这个判断精准针对Windows系统，只有当模块被作为主脚本运行时（__name__ == "__main__"）才会继续执行后续代码，子进程导入时会直接退出，从根源上阻止炸弹触发。

方案2：通过进程名称判断 #

multiprocessing的主进程默认名称是MainProcess，子进程是Process-N，可以用这个特征判断：

import multiprocessing
import sys

if multiprocessing.current_process().name != "MainProcess":
    sys.exit(0)

这个方法跨平台生效，但如果你希望只针对Windows，结合sys.platform判断即可。

二、在模块内安全启动multiprocessing.Process #

可以做到，但必须把进程启动逻辑放在受保护的执行路径里，避免模块被导入时触发进程创建。

正确的模块写法示例 #

# my_module.py
import multiprocessing
import sys

def worker_task():
    """子进程要执行的任务"""
    print("子进程运行中...")

def start_worker():
    """对外暴露的启动函数"""
    # 先做防护判断
    if sys.platform == "win32" and __name__ != "__main__":
        print("警告：请在主脚本的if __name__ == '__main__'块中调用此函数")
        return
    
    p = multiprocessing.Process(target=worker_task)
    p.start()
    p.join()

# 只有当模块被直接运行时（比如python my_module.py）才启动进程
if __name__ == "__main__":
    start_worker()

用户使用时的正确姿势 #

用户在自己的主脚本里需要这样调用：

# main.py
import my_module

if __name__ == "__main__":
    my_module.start_worker()

额外小技巧：使用freeze_support() #

如果你的模块需要支持打包成exe（比如用PyInstaller），在模块开头加入这个函数也能间接防止炸弹：

from multiprocessing import freeze_support

# 必须放在所有multiprocessing相关代码之前
freeze_support()

它的内部逻辑也是检测是否为主进程，子进程导入时会直接退出，既能支持打包，也能解决普通场景的炸弹问题。

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