手动停止Python脚本时触发动作

Question

我搜索了很长时间，但找不到简单的解决方案。 我有一个 python 脚本，它会无限期运行并在满足条件（收集到足够的数据）时保存一些文件。 有没有办法终止脚本的执行并触发 function 来保存收集的（但尚未保存的）数据？ 每次我必须做某事（比如关闭计算机）时，我必须手动停止（终止）脚本（在 Pycharm 中）并且我丢失了一部分尚未保存的数据。

编辑：感谢@Alexander，我能够解决这个问题。 一个简单的代码可能会更好地概述解决方案：

import atexit
import time

@atexit.register
def on_close():
    print('success') #save my data

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

现在，当单击“停止”按钮时，将执行“on_close”function，我可以保存我的数据...

Answer 1

使用atexit模块。 它是python 标准库的一部分。

import atexit

@atexit.register
def on_close():
    ... do comething

atexit.register(func, *args, **kwargs)

将 func 注册为 function 以在终止时执行。 任何要传递给 func 的可选 arguments 必须作为 arguments 传递给 register()。 可以多次注册相同的 function 和 arguments。

这个 function 返回 func，这使得它可以用作装饰器。