Selenium學習（11） 網頁截圖

 轉：https://www.cnblogs.com/fengyiru6369/p/7234840.html

 

save_screenshot方法實現了截圖功能，只須要傳入保存截圖的文件名就能夠了，十分方便；
也能夠使用 get_screenshot_as_file() 方法，（）中傳入路徑。

 

# -*- 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 是獲取路徑的條件,必須這樣的目錄

#設置存儲圖片路徑，測試結果圖片能夠按照天天進行區分

 

 

 

#經過if進行斷言判斷

driver.get(" https://baidu.com/")

#新建立路徑「.」表示當前整個.py文件的路徑所在的位置，「\\」路徑分割符，其中的一個是「\」表示轉義字符

pic_path = '.\\result\\image\\' + current_time1+'\\' + current_time +'.png'

print(pic_path)

time.sleep(5)

print(driver.title)

#截取當前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拋出異常進行斷言判斷   

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()