如何使用selenium在小视口上进行高分辨率截图？

from selenium import webdriver

# 创建Chrome浏览器的WebDriver实例
driver = webdriver.Chrome()

# 设置浏览器窗口大小为所需的小视口大小
driver.set_window_size(800, 600)

driver.get("https://www.baidu.com")
driver.implicitly_wait(10)  # 等待页面加载完成，最多等待10秒

from PIL import Image

# 获取页面的总高度
total_height = driver.execute_script("return document.body.scrollHeight")

# 设置截图的宽度和高度
screenshot_width = driver.get_window_size().get("width")
screenshot_height = driver.get_window_size().get("height")

# 创建一个空白图片，用于拼接截图
screenshot = Image.new("RGB", (screenshot_width, total_height))

# 计算需要滚动的次数
scroll_count = total_height // screenshot_height

# 滚动页面并截图
for i in range(scroll_count + 1):
    # 滚动到当前位置
    driver.execute_script(f"window.scrollTo(0, {i * screenshot_height})")
    # 等待页面滚动完成
    driver.implicitly_wait(1)
    # 截取当前视口的截图
    screenshot_part = driver.get_screenshot_as_png()
    screenshot_part = Image.open(BytesIO(screenshot_part))
    # 将截图拼接到完整截图上
    screenshot.paste(screenshot_part, (0, i * screenshot_height))

# 保存完整截图
screenshot.save("screenshot.png")