Python多进程中能否跳过__main__模块的初始化？

问题场景 #

在Python多进程开发里，我们常看到if __name__ == "__main__"语句。有人会想：如果确定子进程不需要__main__模块里的任何内容，是不是可以把这个语句删掉？

比如下面的代码结构：
test_child.py：

from multiprocessing import Process, get_context

def f(x):
    print(f"{x}")

def start():
    ctx = get_context("spawn")
    p = ctx.Process(target=f, args=("hello",))
    p.start()
    p.join()

test_parent.py：

from test_child import start
start()

运行python test_parent.py时，理论上子进程不需要__main__模块（也就是test_parent.py）的内容，按说可以跳过相关初始化，不用加if __name__ == "__main__"，但实际运行会报错。

为什么会报错？ #

这和你用的spawn启动方式直接相关：

• 当用spawn创建子进程时，系统会启动一个全新的Python解释器进程，它会重新导入主模块（也就是你运行的test_parent.py），而不是直接继承父进程的内存空间。
• 这时候test_parent.py里的start()语句会被再次执行，导致递归创建子进程，最终触发报错。

哪怕子进程真的不需要主模块的内容，spawn的机制决定了它会重新导入主模块，所以不加if __name__ == "__main__"就会重复执行顶层代码。

可行解决方案 #

1. 给主模块加上if __name__ == "__main__"（最标准方案）

修改test_parent.py：

from test_child import start

if __name__ == "__main__":
    start()

这样当子进程重新导入test_parent.py时，__name__不再是"__main__"，start()就不会被重复执行，避免递归创建进程。

2. 改用fork启动方式（仅适用于Unix/Linux/macOS）

如果你的运行环境是类Unix系统，可以改用fork上下文：
修改test_child.py里的start()函数：

def start():
    ctx = get_context("fork")  # 替换成fork
    p = ctx.Process(target=f, args=("hello",))
    p.start()
    p.join()

fork会直接复制父进程的内存空间，不会重新导入主模块，所以不需要if __name__ == "__main__"也能正常运行。但注意fork在Windows系统下不可用。

3. 用subprocess替代multiprocessing（你提到的方案）

如果不想依赖multiprocessing的机制，可以用subprocess直接启动子进程运行脚本，手动处理参数的序列化和反序列化：
比如修改test_child.py，让它可以接收命令行参数：

import sys
import json

def f(x):
    print(f"{x}")

if __name__ == "__main__":
    if len(sys.argv) > 1:
        args = json.loads(sys.argv[1])
        f(args["x"])

然后修改test_parent.py：

import subprocess
import json

args = json.dumps({"x": "hello"})
subprocess.run(["python", "test_child.py", args])

这种方式相当于完全独立的两个进程，不存在主模块重复导入的问题，需要传递数据时用json、pickle等方式手动序列化即可。

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