Почему сигнал.SIGTERM не обрабатывается должным образом в моей основной теме?

Question

У меня есть код python, который работает непрерывно (сбор данных датчика). Он должен быть запущен при загрузке с использованием start-stop-daemon. Тем не менее, я хотел бы умело убить процесс, поэтому я начал с совета в сообщении How to process SIGTERM signal gracefully? и поместил мой основной цикл в отдельный поток. Я бы хотел, чтобы он был изящно закрыт его, когда он работает как демон (start-stop-daemon отправит сигнал об уничтожении), и когда я запускаю его ненадолго для тестирования в терминале самостоятельно (я нажимаю ctrl-c).

Однако обработчик сигнала, похоже, не вызывается, если я убью процесс (даже без использования потока, «done (killed)» никогда не попадает в файл, к которому я перенаправлен). И когда я нажимаю ctrl-c, сбор только продолжается и продолжает печатать данные в терминале (или в файл, к которому я перенаправляюсь).

Что я делаю неправильно в следующем коде?

from threading import Thread 
import time, sys, signal 

shutdown_flag = False #used for gracefull shutdown 

def main_loop(): 
    while not shutdown_flag: 
     collect_data() # contains some print "data" statements 
     time.sleep(5) 
    print "done (killed)" 

def sighandler(signum, frame): 
    print 'signal handler called with signal: %s ' % signum 
    global shutdown_flag 
    shutdown_flag = True 

def main(argv=None): 
    signal.signal(signal.SIGTERM, sighandler) # so we can handle kill gracefully 
    signal.signal(signal.SIGINT, sighandler) # so we can handle ctrl-c 
    try: 
     Thread(target=main_loop, args=()).start() 
    except Exception, reason: 
     print reason 

if __name__ == '__main__': 
    sys.exit(main(sys.argv)) 

Answer 1

Вы прекращаете свой основной поток с этим утверждением:

if __name__ == '__main__': 
    sys.exit(main(sys.argv)) 

Таким образом, ваш обработчик сигнала никогда не будет работать. Обработчик сигнала является частью основного потока, а не созданной вами нитью main_loop. Таким образом, как только основной поток выходит, функция вызова сигнала больше не вызывается.

Вам нужно что-то вроде этого:

def sighandler(signum, frame): 
    print 'signal handler called with signal: %s ' % signum 
    global shutdown_flag 
    shutdown_flag = True 
    sys.exit() # make sure you add this so the main thread exits as well. 

if __name__ == '__main__': 
    main(sys.argv) 
    while 1: # this will force your main thread to live until you terminate it. 
     time.sleep(1) 

Простой тест, чтобы увидеть, сколько потоков выполняются в программе заключается в следующем:

def main_loop(): 
    while not shutdown_flag: 
     collect_data() # contains some print "data" statements 
     time.sleep(5) 
     import threading 
     print threading.enumerate() 
    print "done (killed)"