Comment capturer le signal de terminaison Ctrl+C en Python3

Contexte technique

Une attention particulière est requise pour les programmes Python de longue durée, tels que les backends de serveur ou les programmes de calcul scientifique. Lors de l'exécution de certaines opérations, telles que l'utilisation de Ctrl+C pour terminer un programme en cours d'exécution, nous devrons peut-être le terminer plus tôt. Habituellement, il existe deux possibilités pour cette situation : la première est que le programme comporte une erreur et que le programme doit être arrêté pour ajustement. L'autre est que le programme lui-même est correct, mais qu'il s'exécute trop lentement ou qu'il souhaite se terminer plus tôt. Dans ce scénario, nous espérons souvent conserver les résultats de calcul correspondants. Cependant, si nous utilisons un format de stockage de données tiers pour stocker les données, il se peut qu'il ne prenne pas en charge le stockage continu. Il est très courant de sauvegarder les résultats une fois l'exécution du programme terminée. Si le programme est terminé à mi-chemin, des moyens spéciaux sont nécessaires pour sauvegarder ses résultats.

Cas de base

Regardons d'abord un cas relativement simple : un programme ordinaire qui imprime des nombres, imprime un nombre toutes les 1 seconde, nous pouvons utiliser le signal.signal de python pour capturer ce signal de terminaison.

# signal_exit.py
import signal
import sys
 
def signal_handler(signal, frame):
    print ('\nSignal Catched! You have just type Ctrl+C!')
    sys.exit(0)
 
if __name__ == '__main__':
    import time
 
    signal.signal(signal.SIGINT, signal_handler)
    for x in range(100):
        time.sleep(1)
        print (x)

Lorsque nous sommes à mi-chemin de l'exécution de ce programme et que nous appuyons sur Ctrl+C en même temps, nous obtiendrons les résultats suivants :

$ python3 signal_exit.py
0
1
2
^C
Signal Catched You ! il suffit de taper Ctrl+C !

Ce résultat montre que nous avons capturé l'opération externe de Ctrl+C pendant l'exécution du programme et traité l'opération en conséquence avant de terminer l'exécution du programme. Il convient de noter que si l'opération de terminaison de sys.exit(0) n'est pas ajoutée à ce moment, le programme ne sera pas arrêté et continuera à s'exécuter, ce qui équivaut à capter le signal de terminaison anormal mais à n'effectuer aucun traitement.

Passer des paramètres externes au signal de fin

Dans le cas ci-dessus, nous n'avons capturé que le signal externe de « terminaison de l'exécution », mais si nous allons plus loin, nous voulons capturer le dernier numéro de sortie, ceci Comment faire fonctionner ? La fonction signal.signal elle-même ne prend pas en charge le passage de nombreux paramètres. À l'heure actuelle, il est recommandé de créer une classe par vous-même et d'encapsuler la fonction signal_handler en tant que fonction membre de la classe, afin que nous puissions obtenir les paramètres internes correspondants. , comme dans le cas suivant Comme indiqué :

# signal_exit.py
import signal
import sys
import time
 
class Printer:
    def __init__(self):
        self.x = 0
        signal.signal(signal.SIGINT, self.signal_handler)
 
    def signal_handler(self, signal, frame):
        print ('\nSignal Catched! You have just type Ctrl+C! The last number is: {}'.format(self.x))
        sys.exit(0)
 
    def run(self, counter=10):
        while self.x < counter:
            print (self.x)
            time.sleep(1)
            self.x += 1
 
if __name__ == &#39;__main__&#39;:
    printer = Printer()
    printer.run(counter=100)

À ce moment, si vous appuyez sur Ctrl+C en même temps pendant l'exécution du programme, le résultat sera le suivant :

$ python3 signal_exit.py
0
1
2
3
^C
Signal Catched ! Vous venez de taper Ctrl+C Le dernier chiffre est : 3

Comme vous pouvez le voir, nous avons réussi à capturer le dernier paramètre de sortie.

Ce qui précède est le contenu détaillé de. pour plus d'informations, suivez d'autres articles connexes sur le site Web de PHP en chinois!