一日一技:让Python在退出时强制运行一段代码

off999 2024-11-21 19:21 28 浏览 0 评论

文章来源：未闻Code

作者： kingname

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

当然小编这里也准备一份适合你的学习资料爬虫，web开发的学习资料视频教程，私信小编“01”都可以免费获取！

这段逻辑本身非常简单：

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() ，你注册的函数无法正常执行。

退出python