python - 'if __name__ == "__main_ _":' 后面的任何东西都没有执行

Question

所以，这是我的情况。

我在 Eclipse 中使用 PyDev，在 Windows 中使用 Python 解释器版本 2.7.2。

我正在使用内置的多处理库来尝试 fork 出一堆进程来并行化一个计算密集型循环。我看过的教程说要用，

if __name__ == "__main__":

从根本上防止它产生近乎无限的进程并使我的系统崩溃。

问题是，我是从一个模块中调用它的，而不是我的主脚本；因此，在它被执行之后什么也没有。根本没有并行的机会。当然，如果我删除它，我会得到无限进程垃圾邮件，它会杀死执行代码的机器。

为了引用，这里是相关代码:

from tribe import DataCache
from tribe import WorldThread
from tribe import Actor
from time import sleep
import multiprocessing

class World:
def __init__(self,numThreads,numActors,tickRate):
    print "Initalizing world..."
    self.cache = DataCache.DataCache()
    self.numThreads = numThreads
    self.numActors = numActors
    self.tickRate = tickRate
    self.actors = []
    self.processes = []
    for i in range(numActors):
        self.actors.append(Actor.Actor("test.xml",self.cache))
    print "Actors loaded."
def start_world(self):
    print "Starting world"
    run_world = True;
    while run_world:
        self.world_tick()
        sleep(2)

def world_tick(self):
        if __name__ == '__main__':
            print "World tick"
            actor_chunk = len(self.actors)/self.numThreads
            if len(self.processes)==0:
                for _ in range(self.numThreads):
                    new_process = multiprocessing.Process(WorldThread.WorldProcess.work, args=(_, self.actors[_*actor_chunk,(_+1)*actor_chunk]))

它正在调用的类:

class WorldProcess():
def __init__(self):
    print "World process initilized."
    ''' Really, I'm not sure what kind of setup we'll be doing here yet. '''
def work(self, process_number, actors):
    print "World process" + str(process_number) + " running."
    for actor in actors:
        actor.tick()
    print "World process" + str(process_number) + " completed."

我的评估是否正确，即整个 if name == "main": check 只有在可执行脚本本身中才有效？如果是这样，您如何安全地从模块内 fork 进程？如果不是，为什么它不能在这里工作？

Answer 1

要控制进程数量，请使用 multiprocessing 中的 Pool 类:

from multiprocessing import Pool
p = Pool(5)
def f(x):
     return x*x
p.map(f, [1,2,3])

(编辑:根据评论，这只是 Pool 类的howto。see 更多)

不需要使用 __name__，因为您显式地将 Process 传递给要运行的实际 python 函数。

这个:

def world_tick(self):
    if __name__ == '__main__':
        print "World tick"
        actor_chunk = len(self.actors)/self.numThreads
        if len(self.processes)==0:
            for _ in range(self.numThreads):
                new_process = multiprocessing.Process(WorldThread.WorldProcess.work, args=(_, self.actors[_*actor_chunk,(_+1)*actor_chunk]))

非常糟糕。简化它。

更好的模式是:

class WorkArgs(object):
    ... many attributes follow ...

def proc_work(world_thread, work_args):
    world_thread.WorldProcess.work(work_args.a, work_args.b, ... etc)

p = Pool(5)
p.map(proc_work, [(world_thread, args0), (world_thread, args1), ...])

希望这对您有所帮助!

作为旁注，pickling 您的参数并将它们传递给其他进程将导致导入您的模块。因此，最好确保您的模块不会执行任何 fork /魔术/工作，除非被告知(例如，只有函数/类定义或 __name__ 魔术，而不是实际的代码块) .