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

How to use more arguments with signal-handler?

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)