如何在Python中用Selenium控制窗口下载dialga-box的打开/保存/取消

from selenium import webdriver
from selenium.webdriver.chrome.options import Options
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.action_chains import ActionChains

chrome_options = Options()
chrome_options.add_argument("--disable-extensions")
chrome_options.add_argument("--disable-gpu")
chrome_options.add_argument("--headless")  # 无界面模式
chrome_options.add_experimental_option("prefs", {
    "download.default_directory": "your_download_path",
    "download.prompt_for_download": False,
    "download.directory_upgrade": True,
    "safebrowsing.enabled": True
})

driver = webdriver.Chrome(options=chrome_options)

driver.get("dialga-box_download_link")

# 打开下载对话框
driver.find_element_by_link_text("Download").click()

# 模拟按下键盘的Tab键，将焦点切换到下载对话框上
ActionChains(driver).send_keys(Keys.TAB).perform()

# 模拟按下键盘的Enter键，打开下载文件
ActionChains(driver).send_keys(Keys.ENTER).perform()

# 模拟按下键盘的Tab键，将焦点切换到保存按钮上
ActionChains(driver).send_keys(Keys.TAB).perform()

# 模拟按下键盘的Enter键，保存文件
ActionChains(driver).send_keys(Keys.ENTER).perform()

# 如果需要取消下载，可以模拟按下键盘的Tab键，将焦点切换到取消按钮上
# 然后模拟按下键盘的Enter键，取消下载