如何实现Python程序在退出时强制执行特定代码段？

本文共计720个文字，预计阅读时间需要3分钟。

设想这样一个场景，你需要为某个项目开发一个测试程序。程序开始运行时，会创建一个初始环境。测试完成后，会清理这个环境。这段逻辑非常简单：`setup()`、`test()`、`clean()`。测试代码的核心部分如下。

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

这段逻辑本身非常简单：

setup()
test()
clean()

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

你可能想到，如果这样写会怎么样呢：

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

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

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

这个时候，我们就可以使用Python自带的atexit这个模块了。