[Python] 怎么把HTML的报告转换为图片，利用无头浏览器

时间：2021-08-12 19:20:05

How to convert HTML Report to picture format in Email? So that we can see the automation report also at home or on mobile phone anywhere.

We tried to use phantomJs to get the full-page screenshot of HTML, it doesn't work well on some computers, then we found that the newest Chrome doesn't support it anymore, and Chrome has use its Headless mode to replace phantomJs.

Version 1 : phantomJs

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

import time

import os

from selenium import webdriver

jenkinsJobName=os.getenv("JOB_NAME")

url="http://10.2.3.3/testAgent/AutoAnaylsisReport.html"

print url

save_fn="buildNumResult.PNG"

driver = webdriver.PhantomJS()

driver.maximize_window()

driver.get(url) # Load page

time.sleep(30)

driver.save_screenshot(save_fn)

driver.close()

time.sleep(5)

os.system("taskkill /F /IM phantomjs.exe")

Version 2: Chrome Headless

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

import time

import os

from selenium import webdriver

from selenium.webdriver.common.keys import Keys

url="http://10.2.4.1/testAgent/BillingAnaylisisReport.html"

print url

save_fn="buildNumResult.PNG"

option = webdriver.ChromeOptions()

option.add_argument('--headless')

option.add_argument('--disable-gpu')

option.add_argument("--window-size=1280,1024")

option.add_argument("--hide-scrollbars")

driver = webdriver.Chrome(chrome_options=option)

driver.get(url)

print(driver.title)

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)

driver.save_screenshot(save_fn)

driver.quit()

Version 3 , 为了把截图放在邮件里直接展示，需要把图片截小一点。然后在邮件内容的HTML中加上一行：${FILE,path="report.html"}

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

import time,os

from PIL import Image

from selenium import webdriver

from selenium.webdriver.common.keys import Keys

time.sleep(10)

url="http://10.249.4.17/testRailAgent/BillingAnaylisisReport.html?jenkinsJobName=Billing_Office_Live&projectUATName=Billing_Office_UAT&type=1"

print url

save_fn="buildNumResult.PNG"

option = webdriver.ChromeOptions()

option.add_argument('--headless')

option.add_argument('--disable-gpu')

option.add_argument("--window-size=1280,1024")

option.add_argument("--hide-scrollbars")

driver = webdriver.Chrome(chrome_options=option)

driver.get(url)

time.sleep(30)

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)

driver.save_screenshot(save_fn)

time.sleep(5)

driver.quit()

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('<img src="report'+str(i)+'.png"></img>') #写入数据

fp.close() #关闭文件

　　

相关文章