如何在不中断文件输出的情况下手动终止 Python 脚本？

Question

我对 Python 有点陌生，所以我想这个问题有一个简单的答案。但我似乎无法在任何地方找到解决方案。

我有一个 Python 脚本，它不断接受来自流式 API 的输入并将数据保存到文件中。

我需要停止脚本来修改代码时的问题。如果我使用 ctrl-f2，有时我会在脚本写入输出文件的过程中捕获它，并且文件最终会损坏。

有没有一种简单的方法可以手动停止 Python，让它完成当前代码行的执行？

Answer 1

您可以捕获SIGTERM 或SIGINT 信号并设置一个全局变量，您的脚本会定期检查它是否应该退出。这可能意味着您需要将操作分成更小的块，以便更频繁地检查退出变量

import signal

EXIT = False

def handler(signum, frame):
    global EXIT
    EXIT = True

signal.signal(signal.SIGINT, handler)


def long_running_operation():
    for i in range(1000000):
        if EXIT:
            # do cleanup or raise exception so that cleanup
            # can be done higher up.
            return
        # Normal operation.