JavaScript简称JS,有的测试场景需要JS脚本辅助完成Selenium无法做到的测试工作。webdriver提供了execute_script()方法调用JS代码。execute_script()可以在当前窗口/框架中执行JS脚本,并返回结果。可以使用它操作DOM元素、获取元素属性、执行异步操作等。
from selenium import webdriver
from selenium.webdriver.common.by import By
import time
# 初始化 WebDriver
driver = webdriver.Chrome()
driver.get("https://www.baidu.com")
获取页面标题
title = driver.execute_script("return document.title;")
print(f"Page title is:{title}")
查找搜索输入框并输入关键字
search_input = driver.find_element(By.ID, "kw")
driver.execute_script("arguments[0].value = 'Selenium';", search_input)
查找搜索按钮并点击
search_button = driver.find_element(By.ID, "su")
driver.execute_script("arguments[0].click();", search_button)
滚动到页面底部
driver.execute_script("window.scrollTo(0, document.body.scrollHeight);")
改变搜索框背景颜色
driver.execute_script("arguments[0].style.backgroundColor = 'blue';", search_input)
获取第一个搜索结果的链接文本
first_result_text = driver.execute_script("return document.querySelector('.t a').innerText;")
获取第一个搜索结果的链接
first_result_url = driver.execute_script("return document.querySelector('.t a').href;")
获取网页源代码
page_source = driver.execute_script("return document.documentElement.outerHTML;")
print(f"Page source length:{len(page_source)}")
刷新页面
driver.execute_script("location.reload();")