Selenium处理下载诊断日志

from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.chrome.options import Options
import time

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

# 设置 ChromeDriver 路径
service = Service('/path/to/chromedriver')

# 启动浏览器
driver = webdriver.Chrome(service=service, options=chrome_options)

# 打开目标网页
driver.get('https://example.com')

# 找到下载按钮并点击
download_button = driver.find_element(By.ID, 'download-button')
download_button.click()

# 等待下载完成
time.sleep(10)  # 根据实际情况调整等待时间

# 处理下载的日志文件
import os

download_dir = '/path/to/download/directory'
log_files = [f for f in os.listdir(download_dir) if f.endswith('.log')]

for log_file in log_files:
    with open(os.path.join(download_dir, log_file), 'r') as file:
        log_content = file.read()
        # 处理日志内容
        print(log_content)

# 关闭浏览器
driver.quit()