selenium + python实现截图并且保存图片的方法

在自动化测试过程中，是有必要截图的，特别是遇到错误的时候进行截图，截图可以帮助我们直观的定位错误、记录测试步骤。。截图对于测试人员来说应该是较为重要的一个技能。

webdriver的截图功能十分强悍，无论页面多长，webdriver都能比较完美的截到完整的页面。

Webdriver自带截图功能：

（1）get_screenshot_as_file()

该方式很简单，通过driver获取该方法，将截图要保存的路径写入就好，

（2）save_screenshot()，与get_screenshot_as_file()方式一样，使用起来都很简单

看以下实例，打开csdn首页进行截图

# -*- coding: utf-8 -*-
from selenium import webdriver
import unittest
import os,sys,time
from pathlib import Path

#初始化实例
driver = webdriver.Chrome()

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 )

#设置存储图片路径，测试结果图片可以按照每天进行区分

#通过if进行断言判断
driver.get("https://www.csdn.net/")
#新创建路径“.”表示当前整个.py文件的路径所在的位置，“\\”路径分割符，其中的一个是"\"转义符
pfilename = u'.\\image'
pic_path = pfilename+ '\\' + current_time1 + '_' + current_time + '.png'
#判断文件夹是否存在，不存在就新建一个新的
if Path(pfilename).is_dir():
    pass
else:
    Path(pfilename).mkdir()

print(pic_path)
time.sleep(2)
print(driv