[Web自动化] Selenium执行JavaScript语句

10.3 Selenium执行JavaScript语句

在 Selenium 中，你可以使用 execute_script() 或 execute_async_script() 方法来执行 JavaScript 语句。

以下是一些实用的例子：

10.3.1 执行简单的 JavaScript 语句

python 复制代码

from selenium import webdriver
driver = webdriver.Chrome()
driver.get("http://example.com")
# 执行简单的 JavaScript 语句，比如打印一条消息到控制台
driver.execute_script("console.log('Hello, World!');")

10.3.2 处理需要JavaScript渲染的页面元素

有时，页面元素是通过JavaScript动态生成的，直接使用Selenium的find_element方法可能找不到。这时，可以使用上述的execute_script方法来访问这些元素。

python 复制代码

# 执行JavaScript代码
driver.execute_script("alert('Hello from JavaScript!');")
# 访问需要JavaScript渲染的页面元素
element = driver.execute_script("return document.getElementById('dynamic_element');")

10.3.3 通过 JavaScript 点击元素

python 复制代码

# 假设页面上有一个按钮
button = driver.find_element_by_id("myButton")
# 使用 JavaScript 来点击这个按钮
driver.execute_script("arguments[0].click();", button)

10.3.4 获取元素属性值

python 复制代码

element = driver.find_element_by_id("myElement")
# 使用 JavaScript 获取元素的 'class' 属性值
class_name = driver.execute_script("return arguments[0].className;", element)
print(class_name)

10.3.5 设置元素属性值

python 复制代码

element = driver.find_element_by_id("myElement")
# 使用 JavaScript 设置元素的 'style.display' 属性
driver.execute_script("arguments[0].style.display = 'none';", element)

10.3.6 滚动到页面底部

python 复制代码

# 使用 JavaScript 滚动到页面底部
driver.execute_script("window.scrollTo(0, document.body.scrollHeight);")

10.3.7 等待页面元素加载完成

python 复制代码

# 等待某个元素出现在页面上
element = driver.find_element_by_id("myElement")
script = """
return new Promise((resolve, reject) => {
    const observer = new MutationObserver((mutations, observer) => {
        if (document.contains(arguments[0])) {
            resolve();
        }
    });
    observer.observe(document.body, { childList: true, subtree: true });
});
"""
driver.execute_script(script, element)

10.3.8 执行异步 JavaScript 代码

python 复制代码

# 假设你需要执行异步的 JavaScript 代码，比如等待一个 AJAX 请求完成
script = """
return fetch('https://api.example.com/data')
    .then(response => response.json())
    .then(data => {
        return data; // 或者你想要操作的数据
    });
"""
result = driver.execute_async_script(script)
print(result)

10.3.9 通过 JavaScript 获取元素的内部文本

python 复制代码

element = driver.find_element_by_id("myElement")
# 使用 JavaScript 获取元素的内部文本
text = driver.execute_script("return arguments[0].textContent;", element)
print(text)

10.3.10 通过 JavaScript 操作浏览器窗口

python 复制代码

# 使用 JavaScript 关闭当前浏览器窗口
driver.execute_script("window.close();")

这些例子展示了如何使用 execute_script() 方法执行各种 JavaScript 操作。在某些情况下，你可能需要将找到的 WebElement 作为参数传递给 JavaScript 代码，arguments[0] 是一个特殊的变量，它在 JavaScript 代码中代表了传递给 execute_script() 的第一个参数。