Python Selenium：如何判断下载完成

在本文中，我们将介绍如何使用Python的Selenium库来判断一个下载任务是否已经完成。当我们使用Selenium自动化测试网站时，经常会遇到需要下载文件的情况。但是，由于下载时间的不确定性，我们需要一种方法来判断下载任务是否已经完成，以便我们可以继续进行后续的测试。

阅读更多：Python 教程

1. 设置ChromeOptions

为了使用Selenium自动下载文件，我们需要配置Chrome浏览器的选项。我们可以使用ChromeOptions类来设置一些浏览器的偏好选项，例如指定文件的下载路径。下面是一个示例代码：

from selenium import webdriver

options = webdriver.ChromeOptions()
options.add_experimental_option('prefs', {
    'download.default_directory': 'your_download_directory'  # 设置下载路径
})

driver = webdriver.Chrome(options=options)

通过上述代码，我们可以将下载文件保存到指定的路径，这样方便我们后续的判断和处理。

2. 监听下载事件

要判断下载是否完成，我们需要监听浏览器的下载事件。在Selenium中，我们可以通过重载webdriver.Chrome类的execute_cdp_cmd方法来实现。下面是一个示例代码：

from selenium import webdriver

def enable_download(driver, download_dir):
    driver.execute_cdp_cmd("Page.setDownloadBehavior", {
        "behavior": "allow",
        "downloadPath": download_dir
    })

options = webdriver.ChromeOptions()
options.add_experimental_option('prefs', {
    'download.default_directory': 'your_download_directory'  # 设置下载路径
})

driver = webdriver.Chrome(options=options)
enable_download(driver, 'your_download_directory')

上述代码中，我们定义了一个enable_download函数，用于启用下载功能。通过调用driver.execute_cdp_cmd方法，并传递Page.setDownloadBehavior命令，我们可以设置浏览器的下载行为。

3. 等待下载完成

一旦我们启用了Chrome浏览器的下载功能，我们还需要等待下载任务完成。为了实现这一点，我们可以使用WebDriverWait类来等待下载文件的出现。下面是一个示例代码：

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC

def enable_download(driver, download_dir):
    driver.execute_cdp_cmd("Page.setDownloadBehavior", {
        "behavior": "allow",
        "downloadPath": download_dir
    })

options = webdriver.ChromeOptions()
options.add_experimental_option('prefs', {
    'download.default_directory': 'your_download_directory'  # 设置下载路径
})

driver = webdriver.Chrome(options=options)
enable_download(driver, 'your_download_directory')

# 下载文件的链接或按钮
download_button = driver.find_element(By.XPATH, "//a[@class='download-button']")
download_button.click()

# 等待下载完成
wait = WebDriverWait(driver, 10)
wait.until(EC.presence_of_element_located((By.XPATH, "//div[@class='download-complete']")))

在上述示例代码中，我们首先找到下载文件的链接或按钮，并点击它以触发下载。然后，我们使用WebDriverWait类和expected_conditions模块中的presence_of_element_located方法来等待下载完成。通过指定下载完成后元素的Locator，我们可以在下载完成后继续进行后续的操作。

4. 判断下载是否成功

有时候，下载并不是始终成功的，可能出现网络问题或其他异常情况。为了判断下载是否成功，我们可以使用os模块来检查下载文件的存在。下面是一个示例代码：

import os
from selenium import webdriver

def enable_download(driver, download_dir):
    driver.execute_cdp_cmd("Page.setDownloadBehavior", {
        "behavior": "allow",
        "downloadPath": download_dir
    })

options = webdriver.ChromeOptions()
options.add_experimental_option('prefs', {
    'download.default_directory': 'your_download_directory'  # 设置下载路径
})

driver = webdriver.Chrome(options=options)
enable_download(driver, 'your_download_directory')

# 下载文件的链接或按钮
download_button = driver.find_element(By.XPATH, "//a[@class='download-button']")
download_button.click()

# 等待下载完成
wait = WebDriverWait(driver, 10)
wait.until(EC.presence_of_element_located((By.XPATH, "//div[@class='download-complete']")))

# 判断下载是否成功
file_path = os.path.join('your_download_directory', 'file_name.zip')
if os.path.exists(file_path):
    print("下载成功！")
else:
    print("下载失败！")

通过上述示例代码，我们可以检查下载文件的路径是否存在，从而判断下载是否成功。