手动终止Python脚本时如何触发未保存数据的保存动作？

问题描述 #

我找了很久都没找到简单的解决办法。我有一个无限运行的Python脚本，只有在满足条件（收集到足够数据）时才会保存文件。请问能不能在终止脚本执行时触发一个函数，保存已经收集但还没存储的数据？每次我需要操作（比如关闭电脑）时，都得在PyCharm里手动终止脚本，结果导致部分未保存的数据丢失。

解决方案 #

感谢@Alexander提供的思路，问题已经解决。以下是示例代码，清晰展示实现方式：

import atexit
import time

@atexit.register
def on_close():
    print('success') # 在这里编写保存数据的逻辑

while True:
    print('a') 
    time.sleep(2)

现在点击PyCharm的停止按钮时，on_close函数会自动执行，就能把未保存的数据存储下来了。

内容的提问来源于stack exchange，提问作者Matija Cerne