导入和使用使用多处理的模块,而不会在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上,多处理必须导入主脚本才能运行目标函数,这意味着将执行该文件中的顶级模块代码.只有当顶级模块代码本身试图产生一个新进程时才会出现问题.在您的示例中,顶级模块代码不使用多处理,因此没有无限的流程链.

编辑:现在我得到你所要求的.你不需要保护multi.py.你需要保护你的主脚本,无论它是什么.如果您遇到崩溃,那是因为在主脚本中您正在multi.start()使用顶级模块代码.您的脚本需要如下所示:

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

主脚本中始终需要"保护" .