Windows计算机上IPython控制台中的多处理-如果__name_要求

Question

我正在Windows计算机上使用IPython和Spyder IDE。在IDE启动时，会加载一组py文件来定义一些使我的工作更轻松的函数。一切正常。

现在，我想升级这些功能之一以使用多重处理，但是在Windows上，这需要if __name__ == "__main__":声明。因此，似乎无法直接调用该函数并从IPython控制台传递参数。

例如，其中一个py文件（我们称其为test.py）可能类似于以下代码。

import multiprocessing as mp
import random
import string

# define a example function
def rand_string(length, output):
    """ Generates a random string of numbers, lower- and uppercase chars. """
    rand_str = ''.join(random.choice(
                string.ascii_lowercase
                + string.ascii_uppercase
                + string.digits)
           for i in range(length))
    output.put(rand_str)


def myFunction():
    # Define an output queue
    output = mp.Queue()

    # Setup a list of processes that we want to run
    processes = [mp.Process(target=rand_string, args=(5, output)) for x in range(4)]

    # Run processes
    for p in processes:
        p.start()

    # Exit the completed processes
    for p in processes:
        p.join()

    # Get process results from the output queue
    results = [output.get() for p in processes]

    print(results)

在我的IPython控制台中，我想使用

myFunction()

触发所有计算。但是在Windows上最终会出现BrokenPipe错误。

当我放

if __name__ == "__main__":
     myFunction()

在py文件的末尾，然后运行完整的文件

runfile(test.py)

有用。当然。但这使传递参数到函数非常困难，因为我总是必须编辑test.py文件本身。

我的问题是：如何在不将多处理功能放入此if __name__ == "__main__":语句的情况下运行它？

Answer 1

因此，我解决了该特定问题。

1. 将的定义放在rand_string另一个名为的文件中 test2。

2. 导入test2为模块插入我的test.py脚本

import test2 as test2

1. 修改以下行以访问test2模块

   processes = [mp.Process(target=test2.rand_string, args=(5, output)) for x in range(4)]
   

2. 跑 test.py

3. 呼叫 myFunction()

4. 要开心 ：）

该解决方案基于此多处理教程，该教程建议从另一个脚本导入目标函数。此解决方案绕过if __name__-wrapper的安全自我导入，以访问目标函数。