Teclado Interrompe com o Pool de multiprocessamento do python
Como posso lidar com eventos KeyboardInterrupt com pools de multiprocessamento do python? Aqui está um exemplo simples:
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()
Ao executar o código acima, oKeyboardInterrupt
fica levantado quando eu pressiono^C
, mas o processo simplesmente pára nesse ponto e eu tenho que matá-lo externamente.
Eu quero poder apertar^C
a qualquer momento e fazer com que todos os processos saiam de forma elegante.