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

小编这次要给大家分享的是如何实现Python退出时强制运行一段代码,文章内容丰富,感兴趣的小伙伴可以来了解一下,希望大家阅读完这篇文章之后能够有所收获。

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

设置() 　　试一试: 　　text () 　　除了例外e: 　　打印(& # 39;运行异常:& # 39;,e) 　　清洁()

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

有什么办法,既能让程序报错,又能在报错已经还能运行清洁()呢,# 63;

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

进口atexit
　　
　　@atexit.register
　　def清洁():
　　打印(& # 39;清理环境相关的代码& # 39;)
　　
　　设置()
　　测试()

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

如下图所示:

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

你可以注册多个退出函数,他们会按照注册时间从晚到早以此执行。例如:

进口atexit
　　
　　@atexit.register
　　def clean_1 ():
　　…
　　
　　@atexit.register
　　def clean_2 ():
　　…

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

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