Блог

Python ограничивает функции обработчика сигналов (обработка SIGINT, SIGTERM и т. Д.) Следующей подписью без возможности передачи дополнительных аргументов.

 def signal_handler(sig, frame):
 

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

Проблема заключается в том, что при попытке передать события и процессы завершения работы обработчику сигналов единственный способ, которым мне удалось это сделать, — использовать глобальные переменные.

Мой вопрос: в этом случае, как можно избежать использования глобальных переменных?

 # shutdown events for graceful termination
taskhandler_shutdown = Event()
logger_shutdown = Event()

# start the processes
p_taskhandler = Process(target=taskhandler.capture, args=[taskhandler_shutdown])
p_taskhandler.start()
p_eventlogger = Process(target=eventlogger.capture, args=[logger_shutdown])
p_eventlogger.start()


def termination_signal_handler(sig, frame):

    # staggered shutdown, first terminate taskhandler
    taskhandler_shutdown.set()
    p_taskhandler.join()

    # now terminate logger process
    logger_shutdown.set()
    p_eventlogger.join()

    sys.exit(0)

 

Подпись для обработчика на самом деле не ограничена, просто она вызывается с этими двумя переданными аргументами. Вы можете создать обработчик сигналов с большим количеством параметров и предварительно настроить их с помощью partial() .

 def sigterm_handler(signum, frame, myobj):
    ...

def register_handler(myobj):
    global sigterm_handler
    sigterm_handler = partial(sigterm_handler, myobj=myobj)
    signal.signal(signal.SIGTERM, sigterm_handler)