#1、安裝google-chrome-stable的依賴(lài)
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

# 下載安裝包
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版本完全一致,否則會(huì)出現(xiàn)兼容性問(wèn)題
chromedriver --version

pip3 install selenium
 
pip3 install pillow

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')  # 無(wú)頭參數(shù)
 
    output_path = "./screenshots"
    # 啟動(dòng)瀏覽器
    driver = Chrome(options=options)
    driver.maximize_window()
    try:
        # 訪問(wèn)頁(yè)面
        url = 'https://qq.com'
        driver.get(url)
        time.sleep(1)
        # 設(shè)置截屏整個(gè)網(wǎng)頁(yè)的寬度以及高度
        scroll_width = 2500
        scroll_height = 9500
        driver.set_window_size(scroll_width, scroll_height)
 
        # 獲取頁(yè)面高度
        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)
 
        # 滾動(dòng)并截屏
        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)