如何将 SIGTERM 传播给通过子进程创建的子进程

Question

给定以下 Python 脚本：

a.py:

#!/usr/bin/env python3
# a.py
import signal
import subprocess
import os

def main():
    print('Starting process {}'.format(os.getpid()))
    subprocess.check_call('./b.py')

if __name__ == '__main__':
    main()

b.py：

#!/usr/bin/env python3
# b.py
import signal
import time
import os

def cleanup(signum, frame):
    print('Cleaning up...')
    raise RuntimeError("Error")

def main():
    print('Starting process {}'.format(os.getpid()))
    signal.signal(signal.SIGINT, cleanup)
    signal.signal(signal.SIGTERM, cleanup)

    while True:
        print('Hello')
        time.sleep(1)

if __name__ == '__main__':
    main()

如果我执行a.py，然后我通过kill -15 <pid_of_a_py> 杀死它，它会杀死a.py，但b.py 继续运行：

$ ./a.py 
Starting process 119429
Starting process 119430
Hello
Hello
Hello
Hello
Hello
Terminated   // On a separate terminal, I ran "kill -15 119429"
$ Hello
Hello
Hello
Hello

这是为什么呢？如何确保SIGTERM 从a.py 传播到b.py？还要考虑一个更深的链 a.py -> b.py -> c.py -> d.py ... 我只想为最里面的脚本显式设置错误处理和清理。

Answer 1

一种解决方案是从a.py 显式抛出SystemExit

#!/usr/bin/env python3
# a.py
import signal
import subprocess
import os


def cleanup(signum, frame):
    raise SystemExit(signum)

def main():
    signal.signal(signal.SIGINT, cleanup)
    signal.signal(signal.SIGTERM, cleanup)
    print('Starting process {}'.format(os.getpid()))
    subprocess.check_call('./b.py')

if __name__ == '__main__':
    main()

或者，您可以使用Popen 启动进程，并在父进程退出时调用Popen.send_signal 到子进程。

编辑：

我已对该主题进行了一些阅读，这是一种预期的行为。 kill -15 <pid> 将信号发送到指定的进程，只有这个，信号不应该被传播。但是，您可以向进程组发送信号，该信号也会杀死所有子进程。语法是kill -15 -<pgid>（注意额外的破折号）。进程组 ID 通常与领导进程 ID 相同。