Jak prawidłowo obsługiwać i zatrzymywać zamknięcie systemu (i SIGTERM), aby zakończyć pracę w języku Python?

Question

Podstawowe potrzeby: Mam demona Pythona, który wywołuje inny program przez os.system. Moim życzeniem jest być w stanie odpowiednio obsłużyć wyłączenie systemu lub SIGTERM, aby umożliwić wywoływany program wrócić, a następnie wyjść.

Co ja już próbowałem: Próbowałem podejście za pomocą sygnału:

import signal, time 

def handler(signum = None, frame = None): 
    print 'Signal handler called with signal', signum 
    time.sleep(3) 
    #here check if process is done 
    print 'Wait done' 

signal.signal(signal.SIGTERM , handler) 

while True: 
    time.sleep(6) 

Wykorzystanie time.sleep nie wydaje się działać, a drugi druk nigdy nazywa.

Przeczytałem kilka słów o atexit.register (handler) zamiast signal.signal (signal.SIGTERM, handler), ale nic nie jest wywoływane przy zabijaniu.

Answer 1

Twój kod prawie działa, z wyjątkiem tego, że zapomniałeś wyjść po oczyszczeniu.

Często potrzebujemy przechwytywać różne inne sygnały, takie jak INT, HUP i QUIT, ale nie tyle z demonami.

import sys, signal, time 

def handler(signum = None, frame = None): 
    print 'Signal handler called with signal', signum 
    time.sleep(1) #here check if process is done 
    print 'Wait done' 
    sys.exit(0) 

for sig in [signal.SIGTERM, signal.SIGINT, signal.SIGHUP, signal.SIGQUIT]: 
    signal.signal(sig, handler) 

while True: 
    time.sleep(6) 

W wielu systemach zwykłe procesy nie mają wiele czasu na czyszczenie podczas zamykania. Aby być bezpiecznym, możesz write an init.d script powstrzymać demona i czekać na niego.