如何在Python退出时强制运行一段代码
这篇文章将为大家详细讲解有关如何在Python退出时强制运行一段代码,文章内容质量较高,因此小编分享给大家做个参考,希望大家阅读完这篇文章后对相关知识有一定的了解。
成都创新互联公司是一家专业提供珙县企业网站建设,专注与成都网站制作、成都网站设计、H5网站设计、小程序制作等业务。10年已为珙县众多企业、政府机构等服务。创新互联专业网站建设公司优惠进行中。
设想这样一个场景,你要给一个项目开发测试程序,程序开始运行的时候,会创建初始环境,测试完成以后,会清理环境。 |
设想这样一个场景,你要给一个项目开发测试程序,程序开始运行的时候,会创建初始环境,测试完成以后,会清理环境。
这段逻辑本身非常简单:
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退出时强制运行一段代码就分享到这里了,希望以上内容可以对大家有一定的帮助,可以学到更多知识。如果觉得文章不错,可以把它分享出去让更多的人看到。
本文标题:如何在Python退出时强制运行一段代码
标题网址:http://scyanting.com/article/ggghhg.html