使用 importlib 选择模块并在多处理函数中使用

我想在我的主函数中根据传递给 Python 脚本的参数选择要导入的模块。所以，我正在使用其中之一

blah = importlib.import_module("blah1")
blah = importlib.import_module("blah2")

其中“blahX”是同一接口的不同实现。

我还想使用以下方法将工作传递给不同的流程multiprocessing module.

blah = None

def f(a, b):
    print blah.f(a,b)

if __name__ == '__main__':

    # call importlib.import_module here...

    a = 1
    b = 2
    p = multiprocessing.Process(target=f, args=(a, b))
    p.start()
    p.join()

问题是传递给的函数multiprocessing.Process不知道我在 main 中导入的模块。这与我使用的不同import

import blah1 as blah
#import blah2 as blah

但随后我就失去了在运行时选择模块的能力。

我该如何修复这个设计？

你打电话时mp.Process(...)，多处理模块派生一个子进程（在 Unix 上）或启动一个新的 Python 进程并导入调用模块（在 Windows 上）。在 Unix 上，您当前的代码可以工作，因为分叉时的所有全局变量都会被新的子进程复制。

然而，在 Windows 上，调用模块是导入的。由于 blah 的定义，例如，

blah = importlib.import_module("math")

内部受到保护

if __name__ == '__main__':

的新定义blah不转移到子进程。

因此，要使其在 Windows 上工作，您可以将模块的名称传递给目标函数，然后调用importlib.import_module there:

import importlib
import multiprocessing as mp
blah = None

def f(a, b, name):
    blah = importlib.import_module(name)
    print blah.hypot(a,b)

if __name__ == '__main__':
    a = 1
    b = 2
    p = mp.Process(target=f, args=(a, b, "math"))
    p.start()
    p.join()