利用selenium模塊配合chromedriver把html文件轉換爲圖片

# -*- coding: utf-8 -*-
import time, os
from PIL import Image
from selenium import webdriver

def Html2Image(url,fname):
    v_url = url
    chrome_driver = """/root/chromedriver""" #這裏的chromium版本要和chromedriver版本兼容
    save_fn = fname + '.png'
    option = webdriver.ChromeOptions()
    option.add_argument('--headless')
    option.add_argument('--disable-gpu')
    option.add_argument("--window-size=900,1200")#image size
    option.add_argument("--hide-scrollbars")
    driver = webdriver.Chrome(options=option,executable_path=chrome_driver)
    driver.get(v_url)
    time.sleep(5)
    scroll_width = driver.execute_script('return document.body.parentNode.scrollWidth')
    scroll_height = driver.execute_script('return document.body.parentNode.scrollHeight')
    driver.set_window_size(scroll_width, scroll_height)
    #打開html文件截圖爲圖片
    driver.save_screenshot(save_fn)
    time.sleep(5)
    driver.quit()
    #生成html文件，嵌入上面生成的圖片
    im = Image.open(save_fn)
    w, h = im.size
    imgCount = h // 2000 + 1
    size = h / imgCount
    left = 0
    shang = 0
    index = 0
    for i in range(imgCount):
        i = i + 1
        shang += 1
        a = size * left
        b = size * (i - 1)
        c = w
        d = size * i
        region = im.crop((a, b, c, d))
        region.save("report%s.png" % i)

    fp = open("report.html", "w+b")  # 打開一個文本文件
    for i in range(1, imgCount + 1):
        fp.write(bytes('<img src="report' + str(i) + '.png"></img>','utf-8'))  # 寫入數據
    fp.close()  # 關閉文件
if __name__ == "__main__":
    Html2Image(url="file:///tools/scripts/oradbinfo/storage_report_HtmlResult.html",fname='storage_report_HtmlResult')