Python atexit - 退出前执行操作

有时我们需要在Python脚本退出前执行某些操作，或者在测试时需要模拟程序延时退出的影响，我们都可以使用atexit来帮我们实现。

atexit介绍

python atexit模块定义了一个register函数，用于在python解释器中注册一个退出函数，这个函数在解释器正常终止时自动执行，一般用来做一些资源清理的操作（如关闭数据库连接等）。

如果脚本是非正常crash，或者通过os._exit()退出，都不会调用退出函数。

正常退出前执行

import time
import atexit

@atexit.register
def exit_handle():
    time.sleep(10)
    print('exit')

def echo():
    time.sleep(5)
    print('abc')
   
echo()

可以看到在正常退出情况下，例子脚本在等待5秒后打印’abc’，再等待10秒打印完’exit’后，脚本进程才真正结束。
如果在执行time.sleep(5)时CRTL+C，可以看到exit_handle函数是不会执行的。

✨ 被kill时退出前执行

退出函数在脚本被kill时不会产生作用，为了在kill时能正常生效，需要使用signal捕获关闭信号。

import time
import atexit
from signal import signal, SIGTERM

@atexit.register
def exit_handle():
    time.sleep(10)
    print('exit')

# 使用signal捕获关闭信号，保证被kill时退出前执行
signal(SIGTERM, lambda signum, stack_frame: exit(1))

while True:
    pass