时间:2022-10-02 12:52:15 | 栏目:JavaScript代码 | 点击:次
在Selenium中可以使用drvier.execute_script()来执行Javascript脚本,支持多行语句。
使用Javascript可以实现以下功能:
使用Javascript参数
在使用Javascript语句时,还可以动态传入参数或元素对象,Javascript语句中使用占位符“argument[n]”来表示取第几个参数,如:
js = "arguments[0].setAttribute('style', arguments[1]);"
这里埋设了两个参数,一个是元素对象,另一个是样式字符串。
element = driver.find_element_by_id("kw")
style = "background: red; border: 2px solid yellow;"
执行脚本时,按顺序携带参数即可:
driver.execute_script(js, element, style)
获取Javascript返回值
在Javascript语句中加上return字样,如获取页面高度:
js = 'return document.documentElement.scrollHeight;'
执行该后便可以拿到其返回值:
page_height = driver.execute_script(js)
完整示例
import time from selenium import webdriver driver = webdriver.Chrome() driver.get('https://www.baidu.com/') js = 'arguments[0].setAttribute("style", arguments[1]);' element = driver.find_element('id', 'kw') style = 'background: red; border: 2px solid yellow;' driver.execute_script(js, element, style) page_height = driver.execute_script('return document.documentElement.scrollHeight;') print(page_height) time.sleep(3)
运行结果如图:
打印结果:
890