fix_multiprocessing.py

# Fix keyboard interrupts when using multiprocessing.pool.imap().

# Usage: 
# import fix_multiprocessing.py

from multiprocessing.pool import IMapIterator

def wrapper(func):
  def wrap(self, timeout=None):
    # Note: the timeout of 1 googol seconds introduces a rather subtle 
    # bug for Python scripts intended to run many times the age of the universe.
    return func(self, timeout=timeout if timeout is not None else 1e100)
  return wrap
IMapIterator.next = wrapper(IMapIterator.next)

Thanks!

FYI, I have been using this code for a while in https://github.com/nexB/scancode-toolkit/blob/4ec7a92ac21bf050a872b5dc07ce22baefcfb41d/src/scancode/pool.py
Thank you ++

Hah, that's awesome! Thanks for the heads up.