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)