捕捉从 systemd 发送到 python 守护进程/服务的 SIGINT (Ctrl+C) 信号

Question

编辑：从原始版本缩小问题范围，最初假设所有 SIGINT 覆盖都被忽略，但它实际上只是子流程之一，编辑以反映这一点。

我希望在从 systemd 接收到 SIGINT (Ctrl+C) 时安全关闭 python。 但是，命令sudo systemctl kill --signal=SIGINT myapp忽略了我的 subprocess.Popen subprocess.Popen(args, stdout=PIPE, stderr=PIPE, preexec_fn = os.setsid)行，这阻止了 SIGINT 进入被调用进程（当不使用 systemd），并且无论如何都会使我的程序崩溃。

这是我的设置（类似于： How can I make a python daemon handle systemd signals? ）：

shutdown = False

def shutdown_handler(signal, frame):

    global shutdown

    is_thread = frame.f_code.co_name == "my_thread_func"
    if shutdown:
        logging.info("Force shutdown for process {0}".format(os.getpid()))
        raise KeyboardInterrupt
    else:
        shutdown = True
        if not is_thread:
            logging.info("Shutdown signal received. Waiting for sweeps to finish.")
            logging.info("Press Ctrl-C again to force shutdown.")
        return

signal.signal(signal.SIGINT, shutdown_handler)

在其他地方： subprocess.Popen(args, stdout=PIPE, stderr=PIPE, preexec_fn = os.setsid)

当不使用 systemd 运行时（如python daemon.py ），Popen 子进程继续按需要运行。 但是当使用sudo systemctl kill --signal=SIGINT myapp时，它会将信号发送到父进程、子进程和 Popen（命令行）进程。

systemd[1]: fi_iot.service: Sent signal SIGINT to main process 512562 (python3) on client request.
systemd[1]: fi_iot.service: Sending signal SIGINT to process 512978 (python3) on client request.
systemd[1]: fi_iot.service: Sending signal SIGINT to process 513023 (my-cli-tool) on client request.

有谁知道为什么会这样？

我也愿意接受有关实现此方法的替代方法的建议（例如，将 ExecStop= arg 添加到我的系统配置中，或使用自定义信号而不是 SIGINT），尽管我宁愿尽可能少地覆盖默认行为，我想要sudo systemctl stop myapp做它应该做的事情，而我的自定义代码不会把事情搞砸或让其他人感到困惑。

编辑：这个问题似乎特定于 Popen function 的调用方式，我可能会尝试将其设置为 SIGIGN 并查看它是否有效，这篇文章的早期版本表明这是一个比看起来更广泛的问题。

Answer 1

在 python3 中， ctl-c被引发为错误KeyboardInterrupt

抓住它使用try ， except KeyboardInterrupt:

抓住它的最好办法是创建一个 main 方法，然后将 try 放在你调用它的地方。

更新：

Popen 有一个名为send_signal的方法，因此您可以将 systemd 信号转发给它

相关 python 文档https://docs.python.org/3/library/subprocess.html#subprocess.Popen.send_signal

Answer 2

解决方案：使用不同的方法来防止 subprocess.Popen 覆盖有效的信号：

def preexec_function():
    # used by Popen to tell driver to ignore SIGINT
    signal.signal(signal.SIGINT, signal.SIG_IGN)

proc = Popen(args, stdout=PIPE, stderr=PIPE, preexec_fn = preexec_function)

现在子进程忽略了信号，不像副驾驶的preexec_fn=os.setsid ，它不做我想从 systemd 做的事情，这是我使用 GPT-3 生成的我不理解的代码得到的。

我可能会考虑使用 Showierdata9978 关于使用 send_signal 的建议，这可以让我在按下第二个 Ctrl+C 时发送中断，允许它安全地关闭，尽管忽略。