以playwright脚本为例,详解Python with as处理异常的原理

大家在做playwright自动化测试时,一定会遇到下面的写法

with sync_playwright() as p:
         自动化代码

很多同学可能只是按照这种写法来编写项目的自动化测试代码,对于具体细节可能并不了解,今天我来结合playwright讲解一下 Python中的 with ... as 用法。建议大家拷贝文档中的脚本实际运行一下,学习的效果会更好!

with ... as 概述

在Python中,with语句用于异常处理,使代码更简洁、可读性更强。它简化了文件流等公共资源的管理。基础表达式如下:

with 表达式a [as target]:
        代码块

在上面的表达式中, [as target] 中的部分可以省略。其中,target 参数用于指定一个变量,该语句会将表达式a指定的结果保存到该变量中,我们可以通过with ... as 操作上下文对象。具体实现方法是为一个类定义__enter__和__exit__两个函数。

with 表达式a [as target] 的执行过程是首先执行__enter__ 函数,它的返回值会赋给as后面的target,如果不写as target,返回值会被忽略;然后开始执行代码块中的语句;最后不论执行成功或者失败都会执行__exit__函数,为了更好的理解其运行原理,请参考下面的详细代码解释:

with obj as f:
      f.method(...)
# obj 表示一个对象(或是一个表达式, 结果为一个对象)
# 调用 obj 对象的 __enter__ 方法, 返回值赋值给 as 右边的变量 f,即: f = obj.__enter__()
# 执行 with 代码块中的代码 f.method(...)
# 执行完 with 代码块中的代码后, 无论是否发生异常, 调用 obj 的 __exit__ 方法,即: obj.__exit__(...)

上面的过程其实等价于

obj = ...
f = obj.__enter__()
try:
    # f.method(...)
finally:
    obj.__exit__(...)

注意这里是 try finally 而不是 try except!!!

with ... as 实例

下面举一个例子,让大家了解一下,with真正强大之处——它可以处理异常。

class WithSample:

    def __enter__(self):
        print("__enter__()")
        return self

    def __exit__(self, type, value, trace):
        print(  "__exit__()")
        print( "type:", type) #输出type值
        print( "value:", value) #输出value值
        print("trace:", trace) #输出trace值

    def do_something(self):
        num = 1 / 0 #模拟异常出现
        return num + 1

with WithSample() as sample:
    sample.do_something()

print("---- end ----") #这句话很有用!

输出:

Traceback (most recent call last):

File "C:/Users/Administrator/PycharmProjects/playwright/demo2.py", line 19, in <module>

sample.do_something()

File "C:/Users/Administrator/PycharmProjects/playwright/demo2.py", line 13, in do_something

num = 1 / 0

ZeroDivisionError: division by zero

__enter__()

__exit__()

type: <class 'ZeroDivisionError'>

value: division by zero

trace: <traceback object at 0x000002C7FE590288>

从上面代码我们得知,在with后面的代码块会首先执行__enter__方法的值,然后在抛出任何异常时,_exit__方法都会被执行,并输出type、value和trace的值。

一个容易被忽略的问题

这里需要注意一下,代码后面的语句 print("---- end ----") 并没有执行,这就说明程序发生了异常,执行了__exit__ 以后,程序就退出了!为什么呢?还记得前面讲过的知识点吗?

这里划重点!with obj as f 等价于try ... finally,而不是 try ... except... finally

最后补充一句!如果想捕获 with as 的异常,我们仍然需要使用try ... except... finally

try:
    with api  as f:
          #business as usual
except Exception as e:
          #handle exception

我的每一篇文章都希望帮助读者解决实际工作中遇到的问题!如果文章帮到了您,劳烦点赞、收藏、转发!您的鼓励是我不断更新文章最大的动力!

猜你喜欢

转载自blog.csdn.net/liwenxiang629/article/details/130620609