selenium 操作过程中，元素标红高亮的两种实现方式

时间：2023-02-04 21:46:22

在使用selenium时，动作元素标红高亮，在定位问题时相当好用，有以下二种方法可以实现

一、使用js将元素属性修改

这也是网上大部分的实现方式，但有时候会有点小问题，代码如下：

只写其实某一段函数

......

def apply_style(self, element):

        self.driver.execute_script("arguments[0].setAttribute('style', arguments[1]);", element)

def set_xxx(self,element):

    original_style = element.get_attribute('style')

    self.apply_style(element, "background:green; border:2px solid red;")

    screen_shot = self.driver.get_screenshot_as_png()  #比如这里截图动作

    self.apply_style(element, original_style)  # 截图完了还原元素属性

二、比较复杂了，但相对稳定点

基本思路是：先截图，然后再拿到元素的位置，再使用pillow模块对图进行处理

拿图的坐标，之前我写过一个文章中有说明：

先将图存下，然后读取

img = Image.open(BytesIO(png_data))

brush = ImageDraw.Draw(img)

imgelement = driver.find_element_by_xpath('//img[@src="rand!loginRand.action"]')

location = imgelement.location  #获取元素x,y轴坐标

size=imgelement.size  #获取元素的长宽

blank_height = 150

rect = (0, blank_height, location.width, size.height)

canvas = Image.new('RGB', (width, height), (255, 255, 255))

out_data = BytesIO()

canvas.save(out_data, format('PNG'))

相关文章