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

# 假設(shè) selected_folder 是你希望設(shè)置的下載目錄路徑
selected_folder = "/path/to/your/download/directory" # 示例路徑

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
}
chrome_options.add_experimental_option('prefs', prefs)
chrome_options.add_argument("--enable-logging") # 開(kāi)啟日志有助于調(diào)試

# 假設(shè) service 已經(jīng)定義，例如：service = Service('/path/to/chromedriver')
# driver = webdriver.Chrome(service=service, options=chrome_options)
# ... 執(zhí)行下載操作 ...

import os

# 推薦使用絕對(duì)路徑
# 獲取當(dāng)前腳本所在目錄作為基礎(chǔ)路徑
base_dir = os.path.dirname(os.path.abspath(__file__))
# 或者使用當(dāng)前工作目錄
# base_dir = os.getcwd()

# 構(gòu)建下載目錄路徑，例如在當(dāng)前腳本目錄下創(chuàng)建一個(gè) 'downloads' 文件夾
download_folder_name = "downloads"
selected_folder = os.path.join(base_dir, download_folder_name)

# 確保路徑是絕對(duì)路徑（os.path.join 結(jié)合 os.path.abspath 更安全）
selected_folder = os.path.abspath(selected_folder)

print(f"Resolved download path: {selected_folder}")

if not os.path.exists(selected_folder):
    try:
        os.makedirs(selected_folder)
        print(f"Created download directory: {selected_folder}")
    except OSError as e:
        print(f"Error creating directory {selected_folder}: {e}")
        # 處理目錄創(chuàng)建失敗的情況，例如權(quán)限不足
        raise

# 進(jìn)一步驗(yàn)證：確保它是一個(gè)目錄而不是文件
if not os.path.isdir(selected_folder):
    raise ValueError(f"Specified path {selected_folder} is not a valid directory.")

import os
from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By # 導(dǎo)入By用于元素定位

# --- 配置下載目錄 ---
# 1. 定義期望的下載目錄（推薦使用絕對(duì)路徑）
# 這里以在當(dāng)前腳本所在目錄創(chuàng)建一個(gè) 'downloads' 文件夾為例
current_script_dir = os.path.dirname(os.path.abspath(__file__))
target_download_dir = os.path.join(current_script_dir, "downloads")

# 2. 確保下載目錄存在，如果不存在則創(chuàng)建
if not os.path.exists(target_download_dir):
    try:
        os.makedirs(target_download_dir)
        print(f"Download directory created: {target_download_dir}")
    except OSError as e:
        print(f"Error creating download directory {target_download_dir}: {e}")
        raise # 目錄創(chuàng)建失敗是嚴(yán)重問(wèn)題，應(yīng)停止程序

# 3. 驗(yàn)證路徑是否為有效目錄
if not os.path.isdir(target_download_dir):
    raise ValueError(f"Resolved download path is not a valid directory: {target_download_dir}")

print(f"Using download directory: {target_download_dir}")

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

prefs = {
    'download.default_directory': target_download_dir,
    'savefile.default_directory': target_download_dir,
    'download.prompt_for_download': False, # 禁用下載提示框，實(shí)現(xiàn)靜默下載
    'download.directory_upgrade': True,
    'plugins.always_open_pdf_externally': True # 如果有PDF下載，避免在瀏覽器內(nèi)打開(kāi)
}
chrome_options.add_experimental_option('prefs', prefs)
chrome_options.add_argument("--enable-logging") # 開(kāi)啟Chromedriver日志，有助于調(diào)試

# --- 啟動(dòng) WebDriver ---
# 請(qǐng)將 'path/to/your/chromedriver' 替換為你的 chromedriver 實(shí)際路徑
try:
    service = Service("path/to/your/chromedriver")
    driver = webdriver.Chrome(service=service, options=chrome_options)
    print("WebDriver launched successfully.")

    # --- 執(zhí)行下載操作示例 ---
    # driver.get("http://example.com/some_page_with_download_button")
    # download_button = driver.find_element(By.ID, "download_button_id")
    # download_button.click()
    # print("Download button clicked. Check the specified directory for the file.")

    # 簡(jiǎn)單等待一段時(shí)間，讓下載完成
    # import time
    # time.sleep(10)

except Exception as e:
    print(f"An error occurred: {e}")
finally:
    if 'driver' in locals() and driver:
        # driver.quit() # 根據(jù)實(shí)際情況決定是否關(guān)閉瀏覽器
        pass