from selenium import webdriver
from selenium.webdriver.chrome.options import Options
import os
import time
from urllib.parse import urlparse

def capture_screenshot(url):
    # 配置Chrome浏览器选项
    chrome_options = Options()
    chrome_options.add_argument("--headless")  # 无头模式，不显示浏览器界面
    chrome_options.add_argument("--log-level=3")  # 关闭日志记录

    # 创建Chrome浏览器实例
    driver = webdriver.Chrome(options=chrome_options)

    # 打开指定的URL链接
    driver.get(url)

    # 等待页面加载完成
    driver.implicitly_wait(10)

    # 获取整个网页的尺寸
    page_width = driver.execute_script("return document.body.scrollWidth")
    page_height = driver.execute_script("return document.body.scrollHeight")

    # 设置浏览器窗口大小以适应整个页面
    driver.set_window_size(page_width, page_height)

    # 截取整个页面的截图
    screenshot = driver.get_screenshot_as_png()

    # 关闭浏览器实例
    driver.quit()

    # 获取桌面路径
    desktop_path = os.path.join(os.path.expanduser("~"), "Desktop")

    # 解析URL获取域名部分
    parsed_url = urlparse(url)
    domain = parsed_url.netloc

    # 将截图保存到桌面，以域名命名
    screenshot_path = os.path.join(desktop_path, f"{domain}.png")
    with open(screenshot_path, "wb") as file:
        file.write(screenshot)

# 示例调用
urls = ["https://www.example1.com", "https://www.example2.com", "https://www.example3.com"]

for url in urls:
    capture_screenshot(url)
    time.sleep(30)