使用Python将网页转换为图像？

from selenium import webdriver
from selenium.webdriver.chrome.service import Service
from selenium.webdriver.common.by import By
from webdriver_manager.chrome import ChromeDriverManager
from PIL import Image
import time

def capture_full_page_screenshot(driver, file_name):
    # 获取整个页面的高度
    total_height = driver.execute_script("return document.body.scrollHeight")
    # 获取浏览器窗口的宽度和高度
    viewport_width = driver.execute_script("return document.documentElement.clientWidth")
    viewport_height = driver.execute_script("return window.innerHeight")
    
    # 设置浏览器窗口大小为视口大小
    driver.set_window_size(viewport_width, viewport_height)
    
    # 创建一个空白图像，大小为整个页面的高度和视口的宽度
    screenshot = Image.new('RGB', (viewport_width, total_height))
    
    # 分段截取页面内容并拼接
    scroll_y = 0
    while scroll_y < total_height:
        driver.execute_script(f"window.scrollTo(0, {scroll_y});")
        time.sleep(0.2)  # 等待页面加载
        partial_screenshot = driver.get_screenshot_as_png()
        partial_image = Image.open(partial_screenshot)
        screenshot.paste(partial_image, (0, scroll_y))
        scroll_y += viewport_height
    
    # 保存最终的图像
    screenshot.save(file_name)

# 初始化Chrome浏览器
driver = webdriver.Chrome(service=Service(ChromeDriverManager().install()))

try:
    # 打开目标网页
    driver.get('https://example.com')
    
    # 捕获并保存全页面截图
    capture_full_page_screenshot(driver, 'full_page_screenshot.png')
finally:
    # 关闭浏览器
    driver.quit()