html2img_ping.py 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384
  1. from time import sleep
  2. from selenium import webdriver
  3. from selenium.webdriver.chrome.service import Service
  4. import img_pdf_png
  5. def scroll_page(driver):
  6. # 设置滚动条缓慢滚动
  7. page_height = driver.execute_script('return document.body.scrollHeight') # 页面高度
  8. print(page_height)
  9. # web_height = driver.execute_script("return window.innerHeight") # 网页可见内容的高 原先用页面高度作为滚动条的滑动高度,会导致页面错乱
  10. web_height = 300 # 网页可见内容的高 原先用页面高度作为滚动条的滑动高度,会导致页面错乱
  11. n1 = page_height // web_height
  12. last_height = page_height % web_height
  13. print(last_height)
  14. driver.execute_script("document.documentElement.style.overflowY = 'hidden'")
  15. if n1 > 0: # 判断是否需要滚动
  16. for j in range(n1):
  17. # print(r"j:{}".format(j))
  18. if j == n1 - 1 and last_height > 0: # 截取尾部
  19. driver.execute_script("window.scrollBy(0,{})".format(page_height)) # 滚动条滚到底
  20. print("滚到底了")
  21. else:
  22. driver.execute_script("window.scrollBy(0,{})".format(web_height))
  23. # print(web_height)
  24. sleep(1)
  25. new_page_height = driver.execute_script('return document.body.scrollHeight') # 页面高度
  26. if new_page_height > page_height:
  27. scroll_page(driver)
  28. def html2img(driver, image_folder):
  29. page_width = driver.execute_script('return document.documentElement.scrollWidth')
  30. # page_width = 800
  31. driver.execute_script("$('#app #tipsAlert').next().css('display', 'none')") # 隐藏置顶图标
  32. window_height = driver.execute_script("return window.screen.height") # 屏幕高度
  33. driver.set_window_size(page_width, window_height) # 设置浏览器宽高
  34. scroll_page(driver)
  35. driver.execute_script("document.body.scrollTop = document.documentElement.scrollTop = 0") # 滚动条滚到底
  36. page_height = driver.execute_script('return document.body.scrollHeight') # 页面高度
  37. print("浏览器宽{} 浏览器高度{} 网页实际总高度{}".format(page_width, window_height, page_height))
  38. web_height = driver.execute_script("return window.innerHeight") # 网页可见内容的高 原先用页面高度作为滚动条的滑动高度,会导致页面错乱
  39. n1 = page_height // web_height
  40. print("当前窗口网页高度{}".format(web_height))
  41. last_height = page_height % web_height
  42. print("底部多出来的部分{}".format(last_height))
  43. margin_flag = False
  44. if n1 == 0 or (n1 == 1 and last_height == 0): # 判断是否需要滚动
  45. driver.get_screenshot_as_file(r'{}/{}.png'.format(image_folder, '0')) # 指定截图保存位置
  46. else:
  47. j = 0
  48. while j <= n1:
  49. if j == 0:
  50. driver.execute_script("window.scrollBy(0,0)")
  51. # print(0)
  52. elif j == n1 and last_height > 0: # 截取尾部
  53. driver.execute_script("$('#app #resetcss').css('margin-bottom', '{}px')".format(web_height))
  54. driver.execute_script("window.scrollBy(0,{})".format(web_height)) # 滚动条滚到底
  55. print("拉到底拉")
  56. else:
  57. driver.execute_script("window.scrollBy(0,{})".format(web_height))
  58. sleep(1)
  59. driver.save_screenshot(r'{}/{}.png'.format(image_folder, j)) # 截屏
  60. j = j + 1
  61. # 调用截图函数
  62. if __name__ == "__main__":
  63. # 创建一个 Chrome WebDriver 实例
  64. options = webdriver.ChromeOptions()
  65. options.add_argument("headless")
  66. # options.add_argument(" window-size=1920,1080")
  67. # s = Service(executable_path='/Users/xiexiaoyuan/Downloads/chromedriver_mac64_111/chromedriver')
  68. s = Service(executable_path='E:/chromedriver-win64/chromedriver.exe')
  69. driver = webdriver.Chrome(service=s, options=options)
  70. # driver.maximize_window()
  71. driver.get('https://ficc.hzinsights.com/reportshare_crm_report?code=4e38d30e656da5ae9d3a425109ce9e04')
  72. sleep(5)
  73. image_folder = './imgs'
  74. output_file = './screenshot'
  75. file_name = 'output1'
  76. html2img(driver, image_folder)
  77. img_pdf_png.merge_images(image_folder, output_file, file_name)
  78. driver.quit()