selenium 高亮效果

大家在用设计器拾取网页时，会出现亮色的小框框。这里使用 selenium 驱动网页时，也可以达到同样的效果。
不说太多，上代码：

import unittest
import time
from selenium import webdriver


def highLightElement(driver, element):
    driver.execute_script(
        "document.querySelector('{0}').setAttribute('style', 'background:green; border:2px solid red;')"
        .format(element))


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

    def test_highlight(self):
        url = "http://www.baidu.com"
        self.driver.get(url)
        searchbox = "#kw"
        highLightElement(self.driver, searchbox)
        time.sleep(3)

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


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