在Python多处理.Process中,是否必须使用`__name__==__main__`?
问题描述
我正在编写一个支持易于使用的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。所以跳过它是不安全的。测试属于模块的顶级(全局)级别,而不是在某个类中。它的目的是阻止模块在导入时自动运行任务(而不是定义类、函数等),而不是直接运行。
相关文章