作者: kingname

设想这样一个场景,你要给一个项目开发测试程序,程序开始运行的时候,会创建初始环境,测试完成以后,会清理环境。

这段逻辑本身非常简单:

setup()test()clean()

但由于测试的代码比较复杂,你总是在调试的时候程序异常,导致每次 clean() 函数还没有来得及运行,程序就崩溃了。

你可能想到,如果这样写会怎么样呢:

setup()try:    text()except Exception as e:    print('运行异常:', e)clean()

似乎看起来,程序一定会运行到 clean() 函数,但是,如果你代码写的多,你就应该知道,滥用 try...except... 会让你非常痛苦。例如它突然给你打印一个 运行异常: 1 。你根本不知道是哪里出了问题,也不知道具体出了什么问题。为了找到问题,你必须让程序把错误爆出来。但这样一来, clean() 又不能正常运行了。

有什么办法,既能让程序报错,又能在报错已经还能运行 clean() 呢?

这个时候,我们就可以使用Python自带的 atexit 这个模块了。它的使用方法非常简单:

import atexit@atexit.registerdef clean():    print('清理环境相关的代码')setup()test()

这样一来,我们不需要显式调用 clean 函数了。无论程序正常结束,还是程序异常报错, clean 函数里面的内容总会执行。

如下图所示:


python 运行完后没退出进程 python运行完程序直接退出_Code


atexit 使用中有下面几个注意事项:

  • 你可以注册多个退出函数,他们会按照注册时间从晚到早以此执行。例如:
import atexit@atexit.registerdef clean_1():    ...@atexit.registerdef clean_2():    ...

会先运行 clean_2() 后运行 clean_1()

  • 如果 clean() 函数有参数,那么你可以不用装饰器,而是直接调用 atexit.register(clean_1, 参数1, 参数2, 参数3='xxx') 。
  • 如果程序是被你没有处理过的系统信号杀死的,那么注册的函数无法正常执行。
  • 如果发生了严重的Python内部错误,你注册的函数无法正常执行。
  • 如果你手动调用了 os._exit() ,你注册的函数无法正常执行。