解决 Selenium Chromedriver 下载目录设置失败问题

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
import os

# 假设 selected_folder 是用户期望的下载目录路径
# 例如：selected_folder = "C:\Users\YourUser\Downloads\SeleniumDownloads"
# 或者在 Linux/macOS 上：selected_folder = "/home/youruser/Downloads/SeleniumDownloads"

# 注意：为了演示，这里使用一个示例路径，实际应用中应替换为您的目标路径
selected_folder = os.path.join(os.getcwd(), "selenium_downloads") # 在当前工作目录下创建

chrome_options = webdriver.ChromeOptions()
chrome_options.add_argument("--disable-notifications")
chrome_options.add_argument("--start-maximized")

prefs = {
    'download.default_directory': selected_folder,
    'savefile.default_directory': selected_folder,
    'download.prompt_for_download': False,  # 禁用下载提示框
    'download.directory_upgrade': True      # 升级目录处理
}
chrome_options.add_experimental_option('prefs', prefs)
chrome_options.add_argument("--enable-logging")

# 假设 Chromedriver 服务已正确配置
# service = Service(executable_path="path/to/chromedriver")
# driver = webdriver.Chrome(service=service, options=chrome_options)

print(f"尝试配置的下载目录为: {selected_folder}")
# driver.get("your_download_page_url")
# ... 执行点击下载按钮等操作 ...

import os

# 目标下载目录
target_download_folder = os.path.join(os.getcwd(), "selenium_downloads")

# 检查目录是否存在，如果不存在则创建
if not os.path.exists(target_download_folder):
    try:
        os.makedirs(target_download_folder)
        print(f"成功创建下载目录: {target_download_folder}")
    except OSError as e:
        print(f"创建目录失败: {target_download_folder} - {e}")
        # 处理错误，例如退出程序或使用默认下载目录
        raise

# 验证目录是否可写 (可选但推荐)
if not os.access(target_download_folder, os.W_OK):
    print(f"警告: 目录 {target_download_folder} 可能没有写入权限。")
    # 可以在此处尝试修改权限或选择其他目录

import os

# 假设 target_download_folder 已经过存在性检查和创建
target_download_folder = os.path.join(os.getcwd(), "selenium_downloads")
if not os.path.exists(target_download_folder):
    os.makedirs(target_download_folder)

# 规范化路径
normalized_download_path = os.path.normpath(target_download_folder)
print(f"规范化后的下载目录为: {normalized_download_path}")

# 将规范化后的路径用于 Chrome 选项
prefs = {
    'download.default_directory': normalized_download_path,
    'savefile.default_directory': normalized_download_path,
    'download.prompt_for_download': False,
    'download.directory_upgrade': True
}
# ... 其余 ChromeOptions 和 WebDriver 初始化代码 ...

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
import os
import time # 用于演示等待

def setup_chromedriver_with_custom_download_path(download_folder_name="selenium_downloads"):
    """
    配置 Chromedriver，并设置自定义下载目录。
    确保目录存在且路径规范化。
    """
    # 1. 定义并处理下载目录
    base_dir = os.path.dirname(os.path.abspath(__file__)) # 获取当前脚本所在目录
    target_download_folder = os.path.join(base_dir, download_folder_name)

    # 检查目录是否存在，如果不存在则创建
    if not os.path.exists(target_download_folder):
        try:
            os.makedirs(target_download_folder)
            print(f"成功创建下载目录: {target_download_folder}")
        except OSError as e:
            print(f"创建目录失败: {target_download_folder} - {e}")
            raise # 抛出异常，阻止程序继续执行，因为下载目录不可用

    # 规范化路径，处理多余斜杠等
    normalized_download_path = os.path.normpath(target_download_folder)
    print(f"最终配置的下载目录为: {normalized_download_path}")

    # 2. 配置 ChromeOptions
    chrome_options = webdriver.ChromeOptions()
    chrome_options.add_argument("--disable-notifications")
    chrome_options.add_argument("--start-maximized") # 启动时最大化窗口

    prefs = {
        'download.default_directory': normalized_download_path,
        'savefile.default_directory': normalized_download_path,
        'download.prompt_for_download': False,  # 禁用下载提示框
        'download.directory_upgrade': True      # 升级目录处理
    }
    chrome_options.add_experimental_option('prefs', prefs)
    chrome_options.add_argument("--enable-logging") # 启用日志记录，方便调试

    # 3. 初始化 Chromedriver
    # 假设 Chromedriver 路径在 PATH 环境变量中，或者你可以指定 Service 对象
    # service = Service(executable_path="/path/to/chromedriver")
    # driver = webdriver.Chrome(service=service, options=chrome_options)
    driver = webdriver.Chrome(options=chrome_options) # 如果 chromedriver 在 PATH 中

    return driver, normalized_download_path

# 示例使用
if __name__ == "__main__":
    try:
        driver, download_path = setup_chromedriver_with_custom_download_path("my_selenium_downloads")

        # 导航到一个包含下载链接的页面进行测试
        # 例如，可以找一个提供文件下载的公共测试页面
        # driver.get("https://www.example.com/download_test_page")

        # 假设我们点击一个会触发下载的元素
        # download_button = driver.find_element_by_id("download_file_button")
        # download_button.click()

        print("Chromedriver 已启动，并配置了自定义下载目录。")
        print(f"请检查文件是否下载到: {download_path}")

        # 等待一段时间，以便下载完成并观察结果
        time.sleep(10)

    except Exception as e:
        print(f"发生错误: {e}")
    finally:
        if 'driver' in locals() and driver:
            driver.quit()
            print("Chromedriver 已关闭。")