Python+selenium實(shí)現(xiàn)截圖圖片并保存截取的圖片

# -*- coding: utf-8 -*-
from selenium import webdriver
import unittest
import os,sys,time
import HTMLTestReport
#登錄
driver =webdriver.Firefox()
current_time = time.strftime("%Y-%m-%d-%H_%M_%S", time.localtime(time.time()))
current_time1 = time.strftime("%Y-%m-%d", time.localtime(time.time()))
print(current_time )
print(current_time1 )
# 必須打印圖片路徑HTMLTestRunner才能捕獲并且生成路徑，\image\**\\**.png 是獲取路徑的條件,必須這樣的目錄
#設(shè)置存儲(chǔ)圖片路徑，測試結(jié)果圖片可以按照每天進(jìn)行區(qū)分
#通過if進(jìn)行斷言判斷
driver.get("https://cache.yisu.com/upload/information/20200622/113/64783.html'
print(pic_path)
time.sleep(5)
print(driver.title)
#截取當(dāng)前url頁面的圖片，并將截取的圖片保存在指定的路徑下面（pic_path），注：以下兩種方法都可以
driver.save_screenshot(pic_path)
driver.save_screenshot('.\\result\\image\\' + current_time1+'\\' + current_time +'.png') 
if u'百度一下，你就知道' == driver.title:
  print ('Assertion test pass.') 
else:
  print ('Assertion test fail.')
 #通過try拋出異常進(jìn)行斷言判斷  
driver.get("https://baidu.com/")
driver.save_screenshot(pic_path)
try:
  assert u'百度一下，你就知道' == driver.title
  print ('Assertion test pass.') 
except Exception as e:
  print ('Assertion test fail.', format(e))
time.sleep(5)
driver.quit()