如何在终止时执行代码

Question

我希望我的部分代码仅在我手动停止程序时执行（例如按下pycharm 中的停止按钮）。我认为 finally 声明可以为我做这件事。像这样：

try:
   do_sth()
finally:
   print("you stopped the program")

但它不起作用。我最后和除了都试过了，但都没有奏效。我想当我们停止程序运行时，发生了keyboard_intrrupt 错误，所以最终肯定可以工作。

有什么想法吗？

Answer 1

您可以使用atexit 库。

来自documentation：

atexit 模块定义了注册和注销清理函数的函数。这样注册的函数会在解释器正常终止时自动执行。

这可以这样实现：

#Rest of program...

def before_termination():
    #Do something...

import atexit    
atexit.register(before_termination)

但是，这仅在正常程序终止期间调用，而不是在发生键盘中断时调用。