如何将Selenium文件下载逻辑封装为函数并循环处理下拉选项

完全可以这么做，把重复的下拉选择、更新、下载操作封装成函数，再循环遍历所有选项值逐个执行即可，具体实现如下：

1. 封装核心操作函数 #

将你原有的代码逻辑封装成可复用的函数，接收浏览器驱动对象和目标选项值作为参数：

from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.common.exceptions import NoSuchElementException, TimeoutException

def download_file_by_option(driver, option_value):
    try:
        # 选择指定值的下拉选项
        driver.find_element(By.XPATH, f"//option[@value='{option_value}']").click()
        # 等待更新按钮可点击后点击（替换presence为clickable更稳妥，避免元素存在但不可点击）
        WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.CLASS_NAME, "UpdateButton"))).click()
        # 等待下载按钮可点击后点击
        WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.CLASS_NAME, "DownloadButton"))).click()
        print(f"已完成选项值 {option_value} 的文件下载")
    except (NoSuchElementException, TimeoutException) as e:
        print(f"处理选项值 {option_value} 时出错：{str(e)}")

2. 循环遍历所有选项值执行 #

准备好所有需要处理的15个选项值列表，通过循环逐个调用封装好的函数：

# 替换为你实际的15个选项值
target_options = ['39', '40', '41', '42', '43']  # 示例值，自行补充完整

# 遍历执行下载
for value in target_options:
    download_file_by_option(driver, value)
    # 可选：如果下载后页面需要刷新或等待文件保存完成，可在此添加额外等待逻辑

注意事项 #

• 把原代码中的presence_of_element_located换成element_to_be_clickable更稳妥，因为元素存在不代表可以点击，能有效减少点击失败的情况。
• 异常捕获逻辑会在某个选项处理失败时打印错误信息，同时继续处理下一个选项，不会中断整个循环。
• 若下载后页面有加载延迟，可在循环内添加额外的等待（比如等待某个页面元素出现），确保下一次操作前页面状态正常。

内容的提问来源于stack exchange，提问作者Stickle