在 Selenium 中下载文件

介绍

在使用 Selenium 进行自动化测试时，有时需要下载文件。但是，Selenium 并不直接支持下载文件的操作。因此我们需要通过一些技巧来实现在 Selenium 中下载文件的功能。

本文将介绍如何在 Selenium 中下载文件的方法，并提供代码片段帮助程序员更好地理解。

方法

在 Selenium 中，下载文件的常用方法有两种：使用浏览器自带的下载功能和使用第三方库进行模拟下载。

使用浏览器自带的下载功能

大多数浏览器都有自己的下载功能，可以通过模拟用户操作来触发下载。具体来说，可以通过以下步骤来实现在 Selenium 中下载文件的功能：

1. 打开需要下载文件的页面。

2. 找到下载链接，并模拟单击操作。

3. 在弹出的下载对话框中选择要保存的位置（可以通过调用系统对话框实现）。

以下是 Python 实现代码片段：

from selenium import webdriver
import os

download_dir = os.path.abspath("C:/Downloads")  # 设置下载路径
options = webdriver.ChromeOptions()
options.add_experimental_option("prefs", {
  "download.default_directory": download_dir,  # 设置下载目录
  "download.prompt_for_download": False,
  "download.directory_upgrade": True,
  "safebrowsing.enabled": True
})

browser = webdriver.Chrome(options=options)
browser.get(url)
browser.find_element_by_css_selector("#downloadLink").click()  # 点击下载链接

在这个代码片段中，我们通过 ChromeOptions 设置了下载目录，并通过 find_element_by_css_selector 找到了下载链接元素，然后模拟单击操作触发了下载。

使用第三方库进行模拟下载

有些情况下，页面上的下载链接并不是真正的下载链接，而是通过 JavaScript 做了一些处理，或是需要先进行登录等操作才能下载。这时候使用浏览器自带的下载功能就行不通了。这种情况下，我们需要使用一些第三方库来模拟下载操作。

常用的第三方库有 Requests、Selenium Wire 等。下面以 Requests 为例演示如何在 Selenium 中使用第三方库进行模拟下载：

import requests
from selenium import webdriver

# 配置浏览器
options = webdriver.ChromeOptions()
options.add_argument("--headless")  # 无头模式，不打开浏览器窗口

browser = webdriver.Chrome(options=options)
browser.get(url)

# 获取下载链接
download_link = browser.find_element_by_css_selector("#downloadLink").get_attribute("href")

# 发送下载请求
response = requests.get(download_link)

# 将文件保存到本地
with open("downloaded_file.zip", "wb") as f:
    f.write(response.content)

在这个代码片段中，我们使用了 Requests 库发送了一个 GET 请求，将文件下载到本地。

总结

本文介绍了两种在 Selenium 中下载文件的方法。首先，当下载链接直接可用时，可以借助浏览器自带的下载功能实现。其次，当下载链接需要 JavaScript 处理或登录等操作时，可以使用第三方库模拟下载。无论哪种方法，我们都可以通过模拟用户操作来触发下载。