Python退出时强制运行一段代码的实现方法
2020-04-30 07:38
94 查看
设想这样一个场景,你要给一个项目开发测试程序,程序开始运行的时候,会创建初始环境,测试完成以后,会清理环境。
这段逻辑本身非常简单:
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退出强制运行代码内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!
您可能感兴趣的文章:
相关文章推荐
- 一段代码可以让Python直接在Linux/Mac/Unix上运行
- python运行时修改代码的方法——monkey patch
- 请写出一段 Python 代码实现分组一个 list 里面的元素,比如 [1,2,3,...100]变成 [[1,2,3],[4,5,6]....]
- Objective-C运行时编程 - 实现自动化description方法的思路及代码示例
- 用PHP写一段代码,实现不使用第3个变量,交换$a、$b的值,$a、$b的初始值自己定。(五种方法实现)
- js实现运行代码需要刷新的解决方法
- sublime text3中运行python 中input()解决方法—配置SublimeREPL实现交互式调试程序
- Python实现在线程里运行scrapy的方法
- C# 给某个方法设定执行超时时间 C#如何控制方法的执行时间,超时则强制退出方法执行 C#函数运行超时则终止执行(任意参数类型及参数个数通用版)
- python_获取当前代码行号_获取当前运行的类名和函数名的方法
- 5 种使用 Python 代码轻松实现数据可视化的方法
- Python一行代码实现快速排序的方法
- 3行Python代码实现图像照片抠图和换底色的方法
- Python代码在Windows和Unix都可运行的方法
- Python实现去除代码前行号的方法
- python实现在每个独立进程中运行一个函数的方法
- python通过面向对象程序设计方法 实现铅球运行轨迹的计算
- Linux 下 Python 实现按任意键退出的实现方法
- Java--查看某一段代码或方法运行时间的方法
- Python 1行代码实现文本分类(实战笔记),含代码详细说明及运行结果