如何优雅地实现在Python退出时强制运行一段代码

 设想这样一个场景，何优你要给一个项目开发测试程序，雅地程序开始运行的实现时强时候，会创建初始环境，退出测试完成以后，制运会清理环境。行段

这段逻辑本身非常简单：

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.register def clean():     print(清理环境相关的代码) setup() test() 

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

如下图所示：

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

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

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

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