python爬虫之selenium--高亮显示正在操作的元素-selenium-何三笔记

python爬虫之selenium--高亮显示正在操作的元素

前言

在进行web自动化的时候，如果我们想知道正在操作的元素，我们可以通过js的方式来实现

实战

from selenium import webdriver
import unittest, time


def highLightElement(driver, element):
    '''
    封装好的高亮显示页面元素的方法
    使用JS代码将传入的页面元素对象的背景颜色和边框
　　 颜色分别设置为绿色和红色
    '''
    driver.execute_script("arguments[0].setAttribute('style',arguments[1]);", element,
                          "background:green;border:2px solid red;")


class TestDemo(unittest.TestCase):
    def setUp(self):
        self.driver = webdriver.Chrome()

    def test_HighLightWebElement(self):
        url = 'http://www.sogou.com'
        self.driver.get(url)
        searchBox = self.driver.find_element_by_id('query')
        # 调用高亮显示元素的封装函数，将搜索框高亮显示
        highLightElement(self.driver, searchBox)
        time.sleep(3)
        searchBox.send_keys('测试开发')
        sumitbutton = self.driver.find_element_by_id('stb')
        # 调用高亮显示元素的封装函数，将搜索按钮高亮显示
        highLightElement(self.driver, sumitbutton)
        time.sleep(3)
        sumitbutton.click()
        time.sleep(3)

    def tearDown(self):
        self.driver.quit()


if __name__ == '__main__':
    unittest.main()