专栏首页北京马哥教育Python with提前退出:坑与解决方案

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

问题的起源

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

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

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

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

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

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

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

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

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

使用改进后的单元测试:

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

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

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

第一种解决方案

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

使用的时候:

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

第二种解决方法

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

可以执行多个上下文:

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

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

如同这样:

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

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

能不能再给力点?

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

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

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

第三种解决方案

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

的问题。。。

TIL

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

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

文章转载于 马哥教育官网!

原文链接:https://www.magedu.com/84634.html

我来说两句

0 条评论
登录 后参与评论

相关文章

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

    ? 问题的起源 早些时候使用with实现了一版全局进程锁,希望实现以下效果: ? 全局进程锁本身不用多说,大部分都依靠外部的缓存来实现的,redis上用的是s...

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

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

    小小科
  • Python 中的装饰器

    一, 引用 [书] 流畅的Python [书] Effective Python 二, 基本概念 问题1:装饰器是什么? 解答: 严格来说,装饰器只是语法糖,...

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

    ? 问题的起源 早些时候使用with实现了一版全局进程锁,希望实现以下效果: ? 全局进程锁本身不用多说,大部分都依靠外部的缓存来实现的,redis上用的是s...

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

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

    小小科
  • 没有event loop的PHP

    javascript是单线程脚本语言,所以有了event loop机制,但是 php真的有多进程,多线程吗? 一,php利用socket来实现多线程 在服务器端...

    前朝楚水
  • Canvas基础积累

    这代码显示的结果,其实不是我们想的那样,其实还是存在一些问题,因fill()上方的路径状态还是存在有效的,所以为了解决这个问题,引入了beginPath()和c...

    迹_Jason
  • [HTML5] Canvas绘制简单形状

    使用canvas来进行绘画,它像很多其他dom对象一样,有很多属性和方法,操作这些方法,实现绘画

    陶士涵
  • 你还在傻傻的写驱动吗?

    因为工作项目中8位单片机经常用microchip,而32位常用NXP的,而两家都在积极推广自己的芯片配置,代码生成工具,microchip有MCC,而NXP有M...

    用户1605515
  • 腾讯 Qzone 系统架构设计选型与变迁

    今天跟大家分享InfoQ主持人,走进腾讯系列采访视频之《腾讯 Qzone 系统架构设计选型与变迁》相关对话视频。 本期嘉宾:孙超,西安交通大学硕士 2006年...

    腾讯大讲堂

扫码关注云+社区

领取腾讯云代金券