from selenium import webdriver
from selenium.webdriver.firefox.options import Options
from selenium.webdriver.firefox.service import Service

# 设置 Firefox 下载选项
profile = webdriver.FirefoxProfile()
profile.set_preference("browser.download.folderList", 2)  # 0=桌面, 1=默认下载位置, 2=自定义位置
profile.set_preference("browser.download.dir", "/path/to/download/directory")
profile.set_preference("browser.download.manager.showWhenStarting", False)
profile.set_preference("browser.helperApps.neverAsk.saveToDisk", "application/vnd.ms-excel,application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")

# 启动浏览器
options = Options()
driver = webdriver.Firefox(firefox_profile=profile, options=options)

# 访问 URL 并点击下载链接
driver.get("your_url_here")
download_button = driver.find_element_by_id("download_button_id")
download_button.click()

from selenium import webdriver
from selenium.webdriver.chrome.options import Options

chrome_options = Options()
prefs = {
    "download.default_directory": "/path/to/download/directory",
    "download.prompt_for_download": False,
    "download.directory_upgrade": True,
    "safebrowsing.enabled": True
}
chrome_options.add_experimental_option("prefs", prefs)

driver = webdriver.Chrome(options=chrome_options)
driver.get("your_url_here")