python+selenium實現長截圖

Python selenium操作瀏覽器全屏截圖

最近碰到一個問題，需求是把做好的報告發到領導郵箱（只能發郵箱，不接受郵箱鏈接等方式），大家都知道outlook對js是不支持的，只支持最簡單的html樣式，最后選定的方法是截圖，但是報告可能很長啊，怎么能一次性全截下來呢，請大家往下看

首先本次實現是用的python3.6+selenium+chrome，python的安裝和python庫的安裝就不多說了。用代碼控制chrome的時候需要下一個driver，下載地址：http://chromedriver.storage.googleapis.com/index.html

下載之前要注意看看你電腦上裝的chrome版本號，把下好的driver放到跟chrome.exe一個文件夾就行了

---

接下來直接上代碼

# -*- coding: utf-8 -*-

from selenium import webdriver
from selenium.webdriver.chrome.options import Options
import os
import time

def get_image(url, pic_name):
#chromedriver的路徑
    chromedriver = r"C:\Users\name\AppData\Local\Google\Chrome\Application\chromedriver.exe"
    os.environ["webdriver.chrome.driver"] = chromedriver
#設置chrome開啟的模式，headless就是無界面模式
#一定要使用這個模式，不然截不了全頁面，只能截到你電腦的高度
    chrome_options = Options()
    chrome_options.add_argument('headless')
    driver = webdriver.Chrome(chromedriver,chrome_options=chrome_options)
#控制瀏覽器寫入並轉到鏈接
    driver.get(url)
    time.sleep(1)
#接下來是全屏的關鍵，用js獲取頁面的寬高，如果有其他需要用js的部分也可以用這個方法
    width = driver.execute_script("return document.documentElement.scrollWidth")
    height = driver.execute_script("return document.documentElement.scrollHeight")
    print(width,height)
#將瀏覽器的寬高設置成剛剛獲取的寬高
    driver.set_window_size(width, height)
    time.sleep(1)
#截圖並關掉瀏覽器
    driver.save_screenshot(pic_name)
    driver.close()

#你輸入的參數
url = 'https://movie.douban.com/top250'
pic_name = r'D:\image.png'
get_image(url, pic_name)

---

很簡單的幾行代碼，隨意截任何網站的圖片，通過這個方式也可以截網頁任意部位的圖，怎么樣你學會了嗎？覺得有用的話點贊關注喲

轉自 https://zhuanlan.zhihu.com/p/73255362