Обработка прерываний клавиатуры в многопроцессорном пуле Python

В многопроцессорном модуле Python класс Pool предоставляет удобный способ распределения задач между несколькими процессами. Однако обработка событий KeyboardInterrupt в пулах может быть сложной задачей, о чем свидетельствует фрагмент кода:

from multiprocessing import Pool
from time import sleep
from sys import exit

def slowly_square(i):
    sleep(1)
    return i*i

def go():
    pool = Pool(8)
    try:
        results = pool.map(slowly_square, range(40))
    except KeyboardInterrupt:
        # **** THIS PART NEVER EXECUTES. ****
        pool.terminate()
        print "You cancelled the program!"
        sys.exit(1)
    print "\nFinally, here are the results: ", results

if __name__ == "__main__":
    go()

При запуске этого кода нажатие Ctrl C не запускает процесс очистки, а подпроцессы выполняются бесконечно. Чтобы решить эту проблему, рассмотрите следующее обходное решение:

Поведение, наблюдаемое в коде, является следствием ошибки Python. KeyboardInterrupt не отправляется при ожидании условия в threading.Condition.wait(). Поскольку Pool.map() использует внутреннее ожидание условия, прерывание никогда не будет получено.

Решением является использование Pool.map_async(), который позволяет указать тайм-аут. Установив достаточно длинный таймаут (например, 9999999), мы можем гарантировать, что прерывание будет вызвано в течение разумного времени.

Поэтому замените:

    results = pool.map(slowly_square, range(40))

с:

    results = pool.map_async(slowly_square, range(40)).get(9999999)

Этот обходной путь обеспечивает способ корректной обработки событий KeyboardInterrupt в многопроцессорных пулах, позволяя завершать все подпроцессы, когда пользователь отменяет программу.