在Python多处理.Process中,是否必须使用`__name__==__main__`?

2022-04-10 00:00:00 python python-multiprocessing

问题描述

我正在编写一个支持易于使用的API的类,以添加不同的设置来运行给定的程序(class.add(args)),并使用多处理(class.benchmark(num_processes=5))对所有设置进行基准测试。

从多处理.Process的文档来看,似乎都使用if __name__ == '__main__'。跳过使用它是否安全?

例如,类方法benchmark(num_processes=5)启动并联接进程,而另一个python文件file.py创建一个类并简单地调用class.benchmark(num_processes=5)。它会照常工作吗?


解决方案

如标题"安全导入主模块"下的multiprocessing guidelines中所述,某些形式的多处理需要导入您的主模块,因此如果缺少__name__ == '__main__'检查,您的程序可能会在fork bomb中不受控制地运行。尤其是在Windows上,CPython不能fork。所以跳过它是不安全的。测试属于模块的顶级(全局)级别,而不是在某个类中。它的目的是阻止模块在导入时自动运行任务(而不是定义类、函数等),而不是直接运行。

相关文章