Gestion gracieuse des interruptions de clavier dans les pools multitraitements Python

Lorsque vous travaillez avec le pool multitraitement de Python, la gestion des événements KeyboardInterrupt n'est pas toujours simple. Dans cet article, nous allons explorer comment gérer de telles interruptions et garantir que les processus se terminent correctement.

L'exemple de code fourni illustre le défi. Malgré la présence d'un bloc catch pour KeyboardInterrupt, il ne s'exécute pas lorsque vous appuyez sur control-C. Au lieu de cela, le programme se bloque jusqu'à ce qu'il soit terminé en externe.

La racine de ce problème réside dans un bug Python lié au blocage sur une condition dans threading.Condition.wait(). Dans ce contexte, KeyboardInterrupt n’est jamais envoyé. En conséquence, cela empêche la gestion de l'interruption au sein du pool.

Une solution à ce problème consiste à spécifier un délai d'attente lors de l'attente des résultats. La méthode map_async() avec un paramètre timeout peut être utilisée à la place de map(). Cette approche permet à KeyboardInterrupt d'être reconnu et traité :

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

En définissant une valeur de délai d'attente élevée , nous disons essentiellement au pool de continuer à attendre les résultats jusqu'à ce qu'il soit interrompu par l'utilisateur.

Il est important de noter qu'il existe certaines limites à cette solution de contournement. Si le délai d'attente expire avant que toutes les tâches ne soient terminées, les résultats incomplets seront ignorés. Par conséquent, il est crucial de choisir une valeur de délai d’attente appropriée qui équilibre la réactivité avec le risque de résultats incomplets.