Selenium執行Javascript腳本參數及返回值過程的示例分析

小編給大家分享一下Selenium執行Javascript腳本參數及返回值過程的示例分析，希望大家閱讀完這篇文章之后都有所收獲，下面讓我們一起去探討吧！

在Selenium中可以使用drvier.execute_script()來執行Javascript腳本，支持多行語句。

使用Javascript可以實現以下功能：

• 移除元素隱藏、禁用、只讀等限制屬性

• 為元素添加id或高亮樣式

• 頁面滾動

• 富文本框輸入（HTML注入)

• 獲取頁面信息
  

使用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

看完了這篇文章，相信你對“Selenium執行Javascript腳本參數及返回值過程的示例分析”有了一定的了解，如果想了解更多相關知識，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！