# -*- 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')
利用selenium模块配合chromedriver把html文件转换为图片
最新推荐文章于 2024-07-14 03:17:52 发布