如何使用 Selenium WebDriver 将 HTML 代码导入 Python

Selenium WebDriver 是一个流行的 Python 测试框架，可以用来模拟浏览器行为，测试 Web 应用程序的功能及性能。它不仅可以用来自动化 Web 应用程序测试，还可以用来爬取网页数据、自动化操作网页等。本文将介绍如何使用 Selenium WebDriver 将 HTML 代码导入 Python。

安装 Selenium WebDriver

在使用 Selenium WebDriver 之前，需先安装 Selenium WebDriver 库。使用 pip 命令即可完成安装。

pip install selenium

导入 HTML 代码

使用 Selenium WebDriver 导入 HTML 代码有两种方法：使用 get 方法打开网页链接和使用 execute_script 方法执行 JavaScript 代码。下面将分别介绍这两种方法。

使用 get 方法打开网页链接

使用 get 方法可以打开指定的网页链接并获取源代码。代码如下所示：

from selenium import webdriver

# 创建 WebDriver 实例
driver = webdriver.Chrome()

# 打开网页链接并获取源代码
driver.get('https://www.baidu.com')
html = driver.page_source

# 输出 HTML 代码
print(html)

# 关闭 WebDriver 浏览器窗口
driver.quit()

以上代码使用 Chrome 浏览器打开百度首页，并获取其源代码。通过 driver.page_source 方法可以获取当前网页的 HTML 代码。接着可通过 print 函数输出获取到的 HTML 代码。最后，使用 driver.quit 方法关闭浏览器窗口。

使用 execute_script 方法执行 JavaScript 代码

使用 execute_script 方法可以执行 JavaScript 代码，并将执行结果返回。代码如下所示：

from selenium import webdriver

# 创建 WebDriver 实例
driver = webdriver.Chrome()

# 执行 JavaScript 代码并获取结果
script = 'return document.documentElement.outerHTML;'
html = driver.execute_script(script)

# 输出 HTML 代码
print(html)

# 关闭 WebDriver 浏览器窗口
driver.quit()

以上代码执行了一个简单的 JavaScript 代码，即返回文档对象的 outerHTML 属性。这个属性包含了当前网页的 HTML 代码。可通过 driver.execute_script 方法执行 JavaScript 代码，并将结果保存到变量中。最后，利用 print 函数输出获取到的 HTML 代码。同样，使用 driver.quit 方法关闭浏览器窗口。

总结

使用 Selenium WebDriver 可以方便地将 HTML 代码导入 Python，通过 get 方法和 execute_script 方法可以分别打开网页链接和执行 JavaScript 代码，并获取其中的 HTML 代码。同时，使用 Selenium WebDriver 也可以进行 Web 应用程序的自动化测试和网页数据的爬取，具有广泛的应用。