如何在信号处理程序中使用更多参数

Question

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)

Answer 1

处理程序的签名并没有真正受到限制，只是通过这两个参数调用了。您可以自由地制作一个具有更多参数的信号处理程序，并在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)