Python with提前退出:坑与解决方案

news/2024/11/29 14:53:06/

Python with提前退出:坑与解决方案

问题的起源

早些时候使用with实现了一版全局进程锁,希望实现以下效果:

Python with提前退出:坑与解决方案

Python with提前退出:坑与解决方案

全局进程锁本身不用多说,大部分都依靠外部的缓存来实现的,redis上用的是setnx,有时候根据需要加上缓存击穿问题、随机延后以防止对缓存本身造成压力。

当时同样写了单元测试来测试这段代码的有效性:

Python with提前退出:坑与解决方案

Python with提前退出:坑与解决方案

看起来非常完美地通过了。

这样的一个全局进程锁是通过__enter__方法抛出异常, __exit__方法中捕获异常来实现的:

Python with提前退出:坑与解决方案

Python with提前退出:坑与解决方案

看起来还不错,毕竟单元测试都过了。

但是,这样的实现是有问题的:

原因在于__exit__ 的执行不是包在__enter__ 之外的,因此__enter__抛出的异常,不会被__exit__捕获。

上面的单元测试恰好通过,是因为其中有两个with语句,外面的with 捕获的其实是里面的__enter__ 抛出的异常

使用改进后的单元测试:

Python with提前退出:坑与解决方案

Python with提前退出:坑与解决方案

就会发现单元测试过不去了。

这个问题是我试图使用with实现另一个逻辑:AB测试 时出现的,同样是__enter__抛出异常,exit 试图捕获:

Python with提前退出:坑与解决方案

Python with提前退出:坑与解决方案

调试没有通过的单元测试的时候发现,抛出异常后根本没有执行到__enter__。

第一种解决方案

既然想明白了with的执行顺序,那么第一种解决方案就呼之欲出了:既然__exit__捕获的异常在__enter__执行完成之后,那么我们提供一个函数确认一下就可以了,把ABContext实现改成这样:

Python with提前退出:坑与解决方案

Python with提前退出:坑与解决方案

使用的时候:

Python with提前退出:坑与解决方案

Python with提前退出:坑与解决方案

但这样的解决方法并不优雅,万一使用这个ABContext的时候忘记用ensure方法了,那么就等于完全没用这个Context方法,太容易失误了,而且代码也失去了Pythonic的性质。

第二种解决方法

翻了一下contextlib的标准库文档,发现有一个已经废弃的函数:contextlib.nested

Python with提前退出:坑与解决方案

Python with提前退出:坑与解决方案

可以执行多个上下文:

Python with提前退出:坑与解决方案

Python with提前退出:坑与解决方案

这个废弃的特性在Python2.7之后,可以直接由with关键字执行,形如:

Python with提前退出:坑与解决方案

Python with提前退出:坑与解决方案

这个特性还不错,根据__enter__的执行顺序的话,那么我们可以实现一个由第一个 context的__exit__来捕获,第二个context的__enter__来抛出异常,

如同这样:

Python with提前退出:坑与解决方案

Python with提前退出:坑与解决方案

结合前面我们实现的ABContext的使用是这样的:

Python with提前退出:坑与解决方案

Python with提前退出:坑与解决方案

good,单元测试就这样过了!

能不能再给力点?

确实,在with里要写俩context有点蛋疼,并不是特别优雅,能不能还是回到最初的那种用法:我们只用写一条context,这一个context做到了两个context的事情?

要是nested那个函数还在就好了。。要的其实就是它的功能。

Python3.1之后contextlib提供了一个ExitStack的功能来提供一个模拟的功能,但试了一下发现,实际上只调用了__enter__方法,但没有做对应的异常捕获。

第三种解决方案

哈哈哈哈把自己绕到圈子里去了,想了一下,同样是一个缩进的代码块,为什么不能用if来解决呢!不就是个:

Python with提前退出:坑与解决方案

Python with提前退出:坑与解决方案

的问题。。。

TIL

总之学到了contextlib里的一些有用的函数和装饰器,也第一次发现with可以放个context。

虽然放多个context的动态构造还有待研究,with 后面的代码块也不能填一个元组或者列表。

最后

分享一份Python的学习资料,但由于篇幅有限,完整文档可以扫码免费领取!!!

1)Python所有方向的学习路线(新版)

总结的Python爬虫和数据分析等各个方向应该学习的技术栈。

在这里插入图片描述

比如说爬虫这一块,很多人以为学了xpath和PyQuery等几个解析库之后就精通的python爬虫,其实路还有很长,比如说移动端爬虫和JS逆向等等。

img

(2)Python学习视频

包含了Python入门、爬虫、数据分析和web开发的学习视频,总共100多个,虽然达不到大佬的程度,但是精通python是没有问题的,学完这些之后,你可以按照我上面的学习路线去网上找其他的知识资源进行进阶。

在这里插入图片描述

(3)100多个练手项目

我们在看视频学习的时候,不能光动眼动脑不动手,比较科学的学习方法是在理解之后运用它们,这时候练手项目就很适合了,只是里面的项目比较多,水平也是参差不齐,大家可以挑自己能做的项目去练练。

在这里插入图片描述


http://www.ppmy.cn/news/1249203.html

相关文章

什么是计算机病毒?

计算机病毒 1. 定义2. 计算机病毒的特点3. 计算机病毒的常见类型和攻击方式4. 如何防御计算机病毒 1. 定义 计算机病毒是计算机程序编制者在计算机程序中插入的破坏计算机功能或者破坏数据,影响计算机使用并且能够自我复制的一组计算机指令或程序代码。因其特点与生…

交流负载测试使用场景

交流负载测试是一种在特定环境下,对电力设备、汽车电子部件,工业自动化设备、网络设备、家电产品,航空航天设备以及医疗器械等产品进行测试的方法,该测试的目的是评估这些设备在实际运行条件下的性能和可靠性。 1电力设备测试 交…

深入了解Spring Boot中@Async注解的8大坑点

文章目录 1. 缺少EnableAsync注解2. 异步方法需独立3. 不同的异步方法间无法相互调用4. 返回值为void的异步方法无法捕获异常5. 外部无法直接调用带有Async注解的方法6. Async方法不适用于private方法7. 缺失异步线程池配置8. 异步方法与事务的兼容结语 🎉深入了解S…

【hacker送书第6期】深入理解Java核心技术

第6期图书推荐 内容简介作者简介精彩书评参与方式 内容简介 《深入理解Java核心技术:写给Java工程师的干货笔记(基础篇)》是《Java工程师成神之路》系列的第一本,主要聚焦于Java开发者必备的Java核心基础知识。全书共23章&#xf…

Vue组件开发:工具提示组件的实现方法

在Web开发当中,工具提示(Tooltip)是一种常用的用户界面组件,用于向用户提供额外的信息或说明。它通常以文本形式显示在鼠标悬停或点击某个元素时,为用户提供更详细的内容展示。在本文中,我们将探讨如何使用…

什么是价差?fpmarkets澳福用原油来解释

相信不管是新手还是老手,只要是在市场中交易的投资者,都会听说过“价差”。价差是交易所或经纪人在市场上进行交易时收取的佣金。但fpmarkets澳福认为这个词更准确的定义是“金融资产购买价格和出售价格之间的差额”。这个定义中的关键是“价格之间的差异…

set与map

set与map 一、序列式容器与关联式容器二、pair1、键值对2、作用3、构造函数4、make_pair(1)构造函数(2)作用 5、代码6、运行结果 三、set1、概念2、代码3、运行结果4、说明 四、multiset1、与set的关系2、代码3、运行结果 五、map…

Android Studio 添加so无法打包进apk问题

1.开发环境: Android Studio 2022.3.1 Patch 2 jdk 17 gradle-7.4 2.build.grade配置检查 首先查看build.gradle中是否设置sourceSets ,如果设置的话,打包的时候so是被指导libs目录下的,所有就不能把jnilibs下。 sourceSets {mai…