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

# 假設(shè) selected_folder 是用戶期望的下載目錄路徑
# 例如：selected_folder = "C:\Users\YourUser\Downloads\SeleniumDownloads"
# 或者在 Linux/macOS 上：selected_folder = "/home/youruser/Downloads/SeleniumDownloads"

# 注意：為了演示，這里使用一個(gè)示例路徑，實(shí)際應(yīng)用中應(yīng)替換為您的目標(biāo)路徑
selected_folder = os.path.join(os.getcwd(), "selenium_downloads") # 在當(dāng)前工作目錄下創(chuàng)建

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      # 升級(jí)目錄處理
}
chrome_options.add_experimental_option('prefs', prefs)
chrome_options.add_argument("--enable-logging")

# 假設(shè) Chromedriver 服務(wù)已正確配置
# 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")
# ... 執(zhí)行點(diǎn)擊下載按鈕等操作 ...

import os

# 目標(biāo)下載目錄
target_download_folder = os.path.join(os.getcwd(), "selenium_downloads")

# 檢查目錄是否存在，如果不存在則創(chuàng)建
if not os.path.exists(target_download_folder):
    try:
        os.makedirs(target_download_folder)
        print(f"成功創(chuàng)建下載目錄: {target_download_folder}")
    except OSError as e:
        print(f"創(chuàng)建目錄失敗: {target_download_folder} - {e}")
        # 處理錯(cuò)誤，例如退出程序或使用默認(rèn)下載目錄
        raise

# 驗(yàn)證目錄是否可寫 (可選但推薦)
if not os.access(target_download_folder, os.W_OK):
    print(f"警告: 目錄 {target_download_folder} 可能沒有寫入權(quán)限。")
    # 可以在此處嘗試修改權(quán)限或選擇其他目錄

import os

# 假設(shè) target_download_folder 已經(jīng)過存在性檢查和創(chuàng)建
target_download_folder = os.path.join(os.getcwd(), "selenium_downloads")
if not os.path.exists(target_download_folder):
    os.makedirs(target_download_folder)

# 規(guī)范化路徑
normalized_download_path = os.path.normpath(target_download_folder)
print(f"規(guī)范化后的下載目錄為: {normalized_download_path}")

# 將規(guī)范化后的路徑用于 Chrome 選項(xiàng)
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，并設(shè)置自定義下載目錄。
    確保目錄存在且路徑規(guī)范化。
    """
    # 1. 定義并處理下載目錄
    base_dir = os.path.dirname(os.path.abspath(__file__)) # 獲取當(dāng)前腳本所在目錄
    target_download_folder = os.path.join(base_dir, download_folder_name)

    # 檢查目錄是否存在，如果不存在則創(chuàng)建
    if not os.path.exists(target_download_folder):
        try:
            os.makedirs(target_download_folder)
            print(f"成功創(chuàng)建下載目錄: {target_download_folder}")
        except OSError as e:
            print(f"創(chuàng)建目錄失敗: {target_download_folder} - {e}")
            raise # 拋出異常，阻止程序繼續(xù)執(zhí)行，因?yàn)橄螺d目錄不可用

    # 規(guī)范化路徑，處理多余斜杠等
    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") # 啟動(dòng)時(shí)最大化窗口

    prefs = {
        'download.default_directory': normalized_download_path,
        'savefile.default_directory': normalized_download_path,
        'download.prompt_for_download': False,  # 禁用下載提示框
        'download.directory_upgrade': True      # 升級(jí)目錄處理
    }
    chrome_options.add_experimental_option('prefs', prefs)
    chrome_options.add_argument("--enable-logging") # 啟用日志記錄，方便調(diào)試

    # 3. 初始化 Chromedriver
    # 假設(shè) Chromedriver 路徑在 PATH 環(huán)境變量中，或者你可以指定 Service 對(duì)象
    # 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")

        # 導(dǎo)航到一個(gè)包含下載鏈接的頁面進(jìn)行測(cè)試
        # 例如，可以找一個(gè)提供文件下載的公共測(cè)試頁面
        # driver.get("https://www.example.com/download_test_page")

        # 假設(shè)我們點(diǎn)擊一個(gè)會(huì)觸發(fā)下載的元素
        # download_button = driver.find_element_by_id("download_file_button")
        # download_button.click()

        print("Chromedriver 已啟動(dòng)，并配置了自定義下載目錄。")
        print(f"請(qǐng)檢查文件是否下載到: {download_path}")

        # 等待一段時(shí)間，以便下載完成并觀察結(jié)果
        time.sleep(10)

    except Exception as e:
        print(f"發(fā)生錯(cuò)誤: {e}")
    finally:
        if 'driver' in locals() and driver:
            driver.quit()
            print("Chromedriver 已關(guān)閉。")