在程序退出之前做一些事情

Question

如何拥有一个在程序退出之前执行的函数或其他东西？我有一个脚本将在后台不断运行，我需要它在退出之前将一些数据保存到文件中。有这样做的标准方法吗？

Answer 1

查看 atexit 模块：

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

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

import atexit

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

atexit.register(exit_handler)

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

Answer 2

如果您希望某些内容始终运行，即使出现错误，请像这样使用 try:finally: -

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

if __name__=='__main__':
    main()

如果您还想处理异常，您可以在 之前插入 except:代码>最后：

Answer 3

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

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

我提到这一点只是为了让您了解这一点；执行此操作的“正确”方法是上面提到的 atexit 模块。

Answer 4

这是根据其他答案改编的版本。
它应该可以正常退出、终止和 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)

Answer 5

这可以处理正常退出以及使用 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: ")

Answer 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

Answer 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