selenium 解决 id定位、class定位中，属性值带空格的解决办法

kindertester2023-12-08 21:40

一、前置说明

selenium遇到下面这种元素：

复制代码

<th id="demo id" class="value1 value2 value3 ">

1、虽然id一般不会有空格，但是前端错误的这种写法(如下图)，会造成使用id定位不到元素，如：

python 复制代码

find_element('id', 'demo id')

这种写法会直接报错。

2、class属性有空格，表示有多个属性，一般的解决办法就是使用其中一个属性，如：

python 复制代码

find_element('class name', 'value1')

这种写法是可以定位到元素，但是往往测试人员在使用时，会将其错误的写成：

python 复制代码

find_element('class name', 'value1 value2 value3')

二、解决方案

为了避免上述这种常见的错误，可以写一个方法封装一下，遇到这种错误时，将其转为xpath定位即可。

python 复制代码

    def _fix_find_elements_by_multiple_values_error(self, by, value) -> List[WebElement]:
        """
        修复 `find_elements` 方法在使用 'id' 或 'class name' 定位时,
        value为复合值（比如value='value1 value2'）可能引发的 InvalidArgumentException。
        """
        try:
            return super().find_elements(by, value)
        except InvalidArgumentException as original_e:
            if by == 'class name':
                by = 'class'
            try:
                return super().find_elements('xpath', f'//*[@{by}="{value}"]')
            except:
                raise original_e

三、Demo验证

欢迎技术交流：