python - 导入和使用使用多处理的模块，而不会在 Windows 上导致无限循环

Question

我有一个名为 multi.py 的模块。如果我只是想将 multi.py 作为脚本执行，那么避免在 Windows 上崩溃(产生无限数量的进程)的解决方法是将多处理代码放在:

if __name__ == '__main__':

但是，我试图将它作为模块从另一个脚本导入并调用 multi.start()。如何实现？

# multi.py
import multiprocessing

def test(x):
    x**=2

def start():
    pool = multiprocessing.Pool(processes=multiprocessing.cpu_count()-2)
    pool.map(test, (i for i in range(1000*1000)))
    pool.terminate()
    print('done.')

if __name__ == '__main__':
    print('runs as a script,',__name__)
else:
    print('runs as imported module,',__name__)

这是我的 test.py 我运行:

# test.py
import multi
multi.start()

Answer 1

我不太明白你在问什么。你不需要做任何事情来防止它产生无限多的进程。我只是在 Windows XP 上运行它 --- 导入文件并运行 multi.start() --- 它在几秒钟内完成正常。

你必须执行 if __name__=="__main__" 保护的原因是，在 Windows 上，多处理必须导入主脚本才能运行目标函数，这意味着 top-该文件中的级别模块代码将被执行。仅当顶级模块代码本身尝试生成新进程时才会出现问题。在您的示例中，顶层模块代码不使用多处理，因此不存在无限进程链。

编辑:现在我明白你的意思了。您不需要保护 multi.py。您需要保护您的主脚本，无论它是什么。如果您遇到崩溃，那是因为在您的主脚本中，您在顶层模块代码中执行了 multi.start()。您的脚本需要如下所示:

import multi
if __name__=="__main__":
    multi.start()

main 脚本始终需要“保护”。