在程序退出前做一些事情

问题描述：

如何让某个函数或某个东西在程序退出前执行？我有一个脚本会在后台持续运行，我需要它在退出前将一些数据保存到文件中。有没有标准的方法来做到这一点？

解决方案 1：

检查atexit模块：

http://docs.python.org/library/atexit.html

例如，如果我想在应用程序终止时打印一条消息：

import atexit

def exit_handler():
    print 'My application is ending!'

atexit.register(exit_handler)

请注意，这对于脚本的正常终止非常有效，但并非在所有情况下都会被调用（例如致命的内部错误）。

解决方案 2：

如果您希望某些东西始终运行，即使出现错误，也可以try: finally:这样使用 -

def main():
    try:
        execute_app()
    finally:
        handle_cleanup()

if __name__=='__main__':
    main()

如果你还想处理异常，你可以except:在finally:

解决方案 3：

KeyboardInterrupt如果您通过引发异常（例如按 Ctrl-C）来停止脚本，则可以将其作为标准异常进行捕获。您也可以SystemExit用同样的方式进行捕获。

try:
    ...
except KeyboardInterrupt:
    # clean up
    raise

我提到这一点只是为了让你了解它；做到这一点的“正确”方法是atexit上面提到的模块。

解决方案 4：

这可以处理正常退出以及使用kill或终止进程Ctrl+C：

import sys
import atexit
import signal

def exit_handler():
    print("Cleaning up")

def kill_handler(*args):
    sys.exit(0)

atexit.register(exit_handler)
signal.signal(signal.SIGINT, kill_handler)
signal.signal(signal.SIGTERM, kill_handler)

# MAIN PROGRAM
# for example just reading from the input:
input("Press enter: ")

解决方案 5：

这是根据其他答案改编的版本。它应该可以正常退出、终止和 PyCharm 停止按钮（我可以确认的最后一个）工作（未经全面测试）。

import signal
import atexit


def handle_exit(*args):
    try:
        ... do computation ...
    except BaseException as exception:
        ... handle the exception ...


atexit.register(handle_exit)
signal.signal(signal.SIGTERM, handle_exit)
signal.signal(signal.SIGINT, handle_exit)

解决方案 6：

如果您拥有在程序的整个生命周期内都存在的类对象，那么您也可以使用该__del__(self)方法从类中执行命令：

class x:
def __init__(self):
    while True:
        print ("running")
        sleep(1)

def __del__(self):
    print("destructuring")


a = x()

如果执行中止，这在正常程序端也有效，肯定会有一些例外：

running
running
running
running
running
Traceback (most recent call last):
  File "x.py", line 14, in <module>
    a = x()
  File "x.py", line 8, in __init__
    sleep(1)
KeyboardInterrupt
destructuring

解决方案 7：

您还可以使用上下文管理器来做类似于@Brian C. Lane 的答案的事情：

from contextlib import closing


class RunThat:
    def __init__(self) -> None:
        pass

    def execute_app(self) -> None:
        print("computation")

    def close(self) -> None:
        print("handle_cleanup")


with closing(RunThat()) as rt:
    rt.execute_app()

# --- which results in
# computation
# handle_cleanup