使用Python和WebDriver實現網頁截圖並儲存為PDF文件

使用Python和WebDriver實作網頁截圖並儲存為PDF檔案

摘要：
在Web開發和測試過程中，經常需要對網頁進行截圖以便進行分析、記錄和報告。本文將介紹如何使用Python和WebDriver來實現網頁截圖，並將截圖儲存為PDF文件，以方便分享和存檔。

一、安裝與設定Selenium WebDriver：

1. 安裝Python：造訪Python官網（https://www.python.org/downloads/ ）下載並安裝適合自己作業系統的Python版本。
2. 安裝Selenium Library：開啟終端機或命令列，執行以下命令安裝Selenium庫：pip install selenium
3. 安裝WebDriver：根據自己使用的瀏覽器選擇合適的WebDriver。如使用Chrome瀏覽器，可前往ChromeDriver官網（https://sites.google.com/chromium.org/driver/）下載對應版本的WebDriver，並將其加入環境變數中。

二、編寫Python程式碼：
下面是一個完整的Python程式碼範例，使用WebDriver截取指定網址的網頁並將其儲存為PDF檔案。

from selenium import webdriver
from selenium.webdriver import ActionChains
from selenium.webdriver.chrome.options import Options
from selenium.common.exceptions import WebDriverException

# 设置Chrome浏览器选项
chrome_options = Options()
chrome_options.add_argument("--headless") # 无界面模式
chrome_options.add_argument("--disable-gpu") 

def save_webpage_as_pdf(url, save_path):
    try:
        # 创建浏览器实例
        driver = webdriver.Chrome(options=chrome_options)
        
        # 调整窗口大小以适应网页
        driver.set_window_size(1280, 800)
        
        # 访问网页
        driver.get(url)
        
        # 等待页面加载完成
        driver.implicitly_wait(5)
        
        # 获取网页高度
        total_height = driver.execute_script("return document.body.scrollHeight")
        
        # 将页面切成多个视口，每个视口高度为800
        viewports = int(total_height / 800) + 1
        
        # 初始化PDF打印选项
        driver.execute_script("document.body.style.webkitPrintColorAdjust='exact'")
        driver.execute_script("document.body.style.background='white'")
        driver.execute_script("window.scrollTo(0, 0)")
        
        # 逐个截取每个视口并保存为PDF
        for i in range(viewports):
            # 设置视口位置，每次向下滚动800像素
            driver.execute_script(f"window.scrollTo(0, {i * 800})")
            
            # 截图并保存为图片文件
            driver.save_screenshot(f"{save_path}_{i}.png")
        
        # 关闭浏览器
        driver.quit()
        
        # 使用Python库将截图合并为PDF
        from PIL import Image
        image_files = [f"{save_path}_{i}.png" for i in range(viewports)]
        images = [Image.open(img) for img in image_files]
        images[0].save(f"{save_path}.pdf", "PDF", save_all=True, append_images=images[1:])
        
        # 删除临时图片文件
        import os
        for img in image_files:
            os.remove(img)
            
        print(f"截图成功，并保存为PDF文件：{save_path}.pdf")
    
    except WebDriverException as e:
        print(f"截图失败：{e}")

# 调用截图函数
save_webpage_as_pdf("https://www.example.com", "example_webpage")

三、程式碼解析：

1. 呼叫webdriver.Chrome()方法建立Chrome瀏覽器實例，並使用chrome_options參數配置無介面模式。
2. 使用driver.set_window_size()方法設定瀏覽器視窗大小以適應網頁。
3. 使用driver.get()方法存取指定網址。
4. 使用driver.save_screenshot()方法截取目前瀏覽器頁面，並儲存為圖片檔案。
5. 使用PIL庫將截圖合併為PDF。

四、總結：
本文介紹如何使用Python和WebDriver實作網頁截圖，並將截圖儲存為PDF檔案。透過WebDriver的配置和使用，我們可以輕鬆地進行網頁截圖操作，並利用Python的圖像處理庫將多個截圖合併為PDF文件，實現了對網頁內容的完整保存和分享。此方法可廣泛應用於Web開發和測試過程中的調試、記錄和報告等場景，提升工作效率和準確性。

以上是使用Python和WebDriver實現網頁截圖並儲存為PDF文件的詳細內容。更多資訊請關注PHP中文網其他相關文章！