在Python 3中使用SIGTERM时，允许进程完成而不是被中断

Question

我正在开发一些代码，当从UNIX中的命令行发送sigterm信号时，需要正常关闭。 我发现这个示例https://stackoverflow.com/a/31464349/7019148效果很好，但是存在一个问题。

码：

import signal
import time


class GracefulKiller:

    def __init__(self):
        signal.signal(signal.SIGTERM, self.exit_gracefully)
        self.kill_now = False

    def exit_gracefully(self, signum, frame):
        self.kill_now = True

    def run_something(self):
        print("starting")
        time.sleep(5)
        print("ending")


if __name__ == '__main__':
    killer = GracefulKiller()
    print(os.getpid())
    while True:
        killer.run_something()
        if killer.kill_now:
            break

    print("End of the program. I was killed gracefully :)")

当您通过kill命令kill -15 <pid> ，run_something方法被中断，进程被正常终止。 但是，是否有一种方法可以使run_something方法在终止进程之前完成？ 即防止打扰？

所需的输出：

>>> starting
*kill executed during the middle sleep*
>>> ending
>>> End of the program. I was killed gracefully :)

我的用例是，它将变成一个下载脚本，如果我想终止该过程，我希望该过程在终止之前完成下载...

Answer 1

即使捕获到退出信号， thread.join()等到线程完成。

import threading
import Queue
import time

def download_for(seconds=5):
    for i in range(seconds):
        print("downloading...")
        time.sleep(1)

    print("finished download")


download_thread = threading.Thread(target=download_for, args=(3,))

download_thread.start()
# this waits till the thread finishes even if an exit signal was received
download_thread.join()

# this would just stop the download midway
# download_for(seconds=5)

Answer 2

答案在原始问题中。 我只是把这个留给以后的Google搜索者。

我从来没有遇到过任何问题，我的终端只是在执行kill命令后打印“结束”时遇到了问题。