selenium 源码通读·2 | common/exceptions.py 异常类

2023-04-10
陕西
本文字数：1326 字
阅读完需：约 4 分钟

1 异常类基类

common/exceptions.py目录定义了各种异常类；
其中WebDriverException为基类；
源码如下：

class WebDriverException(Exception):    """    Base webdriver exception.    """
    def __init__(self, msg=None, screen=None, stacktrace=None):        self.msg = msg        self.screen = screen        self.stacktrace = stacktrace
    def __str__(self):        exception_msg = "Message: %s\n" % self.msg        if self.screen is not None:            exception_msg += "Screenshot: available via screen\n"        if self.stacktrace is not None:            stacktrace = "\n".join(self.stacktrace)            exception_msg += "Stacktrace:\n%s" % stacktrace        return exception_msg

复制代码

2 基类分析

从后边的异常类可以看出，都是继承了以上的基类；
而这个基类中有三个参数：

    def __init__(self, msg=None, screen=None, stacktrace=None):        self.msg = msg        self.screen = screen        self.stacktrace = stacktrace
# msg：发生异常的信息# screen：发生异常进行截屏# stacktrace：异常堆栈信息

复制代码

此外还定义了“魔法”方法__str__，用来返回一个对象的描述信息

 def __str__(self):        exception_msg = "Message: %s\n" % self.msg        if self.screen is not None:            exception_msg += "Screenshot: available via screen\n"        if self.stacktrace is not None:            stacktrace = "\n".join(self.stacktrace)            exception_msg += "Stacktrace:\n%s" % stacktrace        return exception_msg

复制代码

从__str__中可以看出返回的是异常信息exception_msg，而其初始值为msg，当screen 和stacktrace 都不为空的时候，exception_msg为其它两个参数的拼接。

3 常见异常类 NoSuchElementException 分析

我们在做自动化的时候，经常会遇到的异常是，元素没有找到，那么它的异常类就是NoSuchElementException；
NoSuchElementException源码如下：

class NoSuchElementException(WebDriverException):    """    Thrown when element could not be found.
    If you encounter this exception, you may want to check the following:        * Check your selector used in your find_by...        * Element may not yet be on the screen at the time of the find operation,          (webpage is still loading) see selenium.webdriver.support.wait.WebDriverWait()          for how to write a wait wrapper to wait for an element to appear.    """    pass

复制代码