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