快捷導(dǎo)航

Python+Selenium實(shí)現(xiàn)無頭瀏覽器網(wǎng)頁截圖

更新時間：2025年03月13日 09:01:45 作者：程序員Rocky

這篇文章主要為大家詳細(xì)介紹了Python+Selenium實(shí)現(xiàn)無頭瀏覽器網(wǎng)頁截圖的相關(guān)知識,文中的示例代碼講解詳細(xì),感興趣的小伙伴可以跟隨小編一起學(xué)習(xí)一下

一、背景

最近有個需求，就是需要登錄系統(tǒng)，針對系統(tǒng)的流程數(shù)據(jù)，有一個工作流程圖或者說叫審批流程圖。每個階段都有相對應(yīng)的人員審批、評論信息，以及是否通過等等。這些信息都需要進(jìn)行網(wǎng)頁截圖。

這些信息人事需要進(jìn)行歸檔和壓縮，如果是人工操作，里面的流程有將近10w+的數(shù)據(jù)條目。這如果讓人工實(shí)現(xiàn)實(shí)在是頭大，根本無法完成這個巨大的工作量。

自然這種事情落到了我的頭上，誰讓我們會搞程序呢，呵呵。我采用的Python3+Selenium的方式，模擬登錄比較麻煩需要驗(yàn)證碼各種，我直接使用人事賬號的登錄cookie即可實(shí)現(xiàn)登錄，那接下來就是訪問頁面就可以。訪問到對應(yīng)頁面進(jìn)行截圖保存，然后又繼續(xù)翻頁，繼續(xù)截圖，以此類推。

腦海里就浮現(xiàn)了我的實(shí)現(xiàn)邏輯。說干就干。

二、Chrome無頭瀏覽器+驅(qū)動下載安裝

注意事項(xiàng): google-chrome-stable和chromedriver最好保持一致的版本，否則可能會出現(xiàn)兼容性報錯的情況!!!!

1、google-chrome-stable下載安裝

#1、安裝google-chrome-stable的依賴
yum install -y atk at-spi2-atk libdrm mesa-libgbm gtk3 vulkan xdg-utils
#安裝中文字體
yum install -y wqy-microhei-fonts
 
#2、下載rpm安裝包
https://mirrors.aliyun.com/google-chrome/google-chrome/google-chrome-stable-110.0.5481.77-1.x86_64.rpm
 
#3、安裝
 
rpm -ivh google-chrome-stable-110.0.5481.77-1.x86_64.rpm
 
#4、查看版本信息
google-chrome-stable --version

顯示版本信息如下，表示google-chrome-stable安裝成功:

2、安裝chromedriver驅(qū)動

# 下載安裝包
wget "https://registry.npmmirror.com/-/binary/chromedriver/110.0.5481.77/chromedriver_linux64.zip"
 
# 解壓
unzip chromedriver_linux64.zip -d /usr/local/chromedriver
 
# 設(shè)置環(huán)境變量
 
export CHROME_DRIVER=/usr/local/chromedriver
export PATH=$PATH:$CHROME_DRIVER
 
# 驗(yàn)證chromedriver版本是否與google-chrome-stable版本完全一致,否則會出現(xiàn)兼容性問題
chromedriver --version

三、Python樣例代碼

1、pip3安裝依賴

pip3 install selenium
 
pip3 install pillow

2、樣例代碼

訪問https://qq.com 騰訊門戶網(wǎng)站，將騰訊網(wǎng)截圖保存下來。如果是自己的業(yè)務(wù)，那么傳遞cookie這些參考文檔傳遞即可。此時，您就當(dāng)你的程序就是一個chorme瀏覽器，chrome瀏覽器能做的事情，你的代碼就能做。只是平時我們使用chrome是鼠標(biāo)點(diǎn)點(diǎn)點(diǎn)，換到代碼就是調(diào)用API函數(shù)即可。

from selenium.webdriver import Chrome
from selenium.webdriver.chrome.options import Options
import time
import os.path
from PIL import Image
 
def mergeImage(output_path):
    images = [Image.open(os.path.join(output_path, f"screenshot_{i}.png")) for i in range(scrolls)]
    widths, heights = zip(*(i.size for i in images))
 
    total_width = max(widths)
    total_height = sum(heights)
 
    new_image = Image.new('RGB', (total_width, total_height))
    y_offset = 0
    for img in images:
        new_image.paste(img, (0, y_offset))
        y_offset += img.height
 
    final_screenshot_path = os.path.join(output_path, "final_screenshot.png")
    new_image.save(final_screenshot_path)
    print(f"Final screenshot saved to {final_screenshot_path}")
 
 
if __name__ == "__main__":
    # 設(shè)置瀏覽器
    options = Options()
    options.add_argument('--no-sandbox')
    options.add_argument('--disable-gpu')
    options.add_argument('--headless')  # 無頭參數(shù)
 
    output_path = "./screenshots"
    # 啟動瀏覽器
    driver = Chrome(options=options)
    driver.maximize_window()
    try:
        # 訪問頁面
        url = 'https://qq.com'
        driver.get(url)
        time.sleep(1)
        # 設(shè)置截屏整個網(wǎng)頁的寬度以及高度
        scroll_width = 2500
        scroll_height = 9500
        driver.set_window_size(scroll_width, scroll_height)
 
        # 獲取頁面高度
        total_height = driver.execute_script("return document.body.scrollHeight")
        viewport_height = driver.execute_script("return window.innerHeight")
        scrolls = int(total_height / viewport_height) + 1
 
        # 創(chuàng)建輸出目錄
        if not os.path.exists(output_path):
            os.makedirs(output_path)
 
        # 滾動并截屏
        for i in range(scrolls):
            y_offset = i * viewport_height
            driver.execute_script(f"window.scrollTo(0, {y_offset});")
            time.sleep(1)  # 等待內(nèi)容加載
            screenshot_path = os.path.join(output_path, f"screenshot_{i}.png")
            driver.get_screenshot_as_file(screenshot_path)
            print(f"Saved {screenshot_path}")
 
        # 合并圖片
        mergeImage(output_path)
 
        # 關(guān)閉瀏覽器
        driver.close()
        driver.quit()
 
    except Exception as e:
        print(e)