Python + Selenium操作一:截图详解

在做测试的时候,空口无凭都是白掐,特别是自动化测试的时候,更需要图片来佐证自己发现的问题

话不多说,直接进入主题,技术就是这么直白

Webdriver自带截图功能,get_screenshot_as_file(),save_screenshot(),使用起来还是蛮方便的

截图一:get_screenshot_as_file()

截图二:save_screenshot()

get_screenshot_as_file()方式

该方式很简单,通过driver获取该方法,将截图要保存的路径写入就好,先看一张图,没运行代码的时候,Picture目录下还没有名为baidu.png的图片,建个Picture目录,是为了方便管理截图的图片

 

接下来运行代码,保存的图片如下:

可以看到,图片是截图成功了的,在对应文件夹下也是可以看到图片的,代码如下:

# coding=utf-8
from selenium import webdriver
import time

driver = webdriver.Chrome()
driver.get("https://www.baidu.com")
driver.maximize_window()
time.sleep(2)
try:
    picture_url=driver.get_screenshot_as_file('G:\\201801-\\python_code\\Demo\\Picture\\baidu.png')
    print("%s:截图成功!!!" % picture_url)
except BaseException as msg:
    print(msg)
driver.quit()

这里需要注意的是,文件存放的路径一定要正确,路径中的\\双斜杠,有一个是转义符,这个就不多说了

这种方式写法太过于死板,这样运行下去,图片名字会重名,达不到什么好效果,可以生成代码运行的当前时间,来做为图片名,这样就不会再出现该问题了,修改代码如下:

# coding=utf-8
from selenium import webdriver
import time

driver = webdriver.Chrome()
driver.get("https://www.baidu.com")
driver.maximize_window()
time.sleep(2)
picture_time = time.strftime("%Y-%m-%d-%H_%M_%S", time.localtime(time.time()))
print(picture_time)
try:
    picture_url=driver.get_screenshot_as_file('G:\\201801-\\python_code\\Demo\\Picture\\'+ picture_time +'.png')
    print("%s:截图成功!!!" % picture_url)
except BaseException as msg:
    print(msg)
driver.quit()

代码运行后,查看截图的效果,如下:

该方法使用起来更加方便,算是一个小小优化

save_screenshot()方式

save_screenshot()与get_screenshot_as_file()方式都是一样,先看一个小例子,代码如下:

# 截图方式二
# coding=utf-8
from selenium import webdriver
import time

driver = webdriver.Chrome()
driver.get("https://www.baidu.com")
driver.maximize_window()
time.sleep(2)
try:
    picture_url=driver.save_screenshot('.\\baidu1.png')
    print("%s :截图成功!!!" % picture_url)
except BaseException as msg:
    print("%s :截图失败!!!" % msg)
driver.quit()

运行该代码后,在当前目录下会保存成功一张名为baidu1.png的图片,如下:

一种方式学会,另一种是不是很简单了,接下来以这种截图方式,来做个优化吧,先看代码:

# 截图方式二
# coding=utf-8
from selenium import webdriver
import os
import time
# 生成年月日时分秒时间
picture_time = time.strftime("%Y-%m-%d-%H_%M_%S", time.localtime(time.time()))
directory_time = time.strftime("%Y-%m-%d", time.localtime(time.time()))
print(picture_time)
print(directory_time)
# 打印文件目录
print(os.getcwd())
# 获取到当前文件的目录,并检查是否有 directory_time 文件夹,如果不存在则自动新建 directory_time 文件
try:
    File_Path = os.getcwd() + '\\' + directory_time + '\\'
    if not os.path.exists(File_Path):
        os.makedirs(File_Path)
        print("目录新建成功:%s" % File_Path)
    else:
        print("目录已存在!!!")
except BaseException as msg:
    print("新建目录失败:%s" % msg)

driver = webdriver.Chrome()
driver.get("https://baidu.com/")
try:
    url=driver.save_screenshot('.\\' + directory_time + '\\' + picture_time + '.png')
    print("%s :截图成功!!!" % url)
except BaseException as pic_msg:
    print("截图失败:%s" % pic_msg)
time.sleep(2)
driver.quit()

 该代码的优化,是自动创建日期目录,并将截取的图片以年月日时分秒命名保存在当前日期目录下,这样就能很方便的查看图片保存路径,也方便管理自己的测试图片

 运行代码,我们来查看结果,如下:

这样运行代码后,就自动生成了名为2018-09-16的文件夹,截取的图片命名为2018-09-16-16_14_18.png,并保存在2018-09-16的文件夹下,如果明天运行该代码,就会生成新的文件夹,并命名为2018-09-17,这样就很好的区分了图片日期来源,如果再次运行该代码,就会提示文件夹已存在,图片依然截取成功,如下:

好了,这就是今天的截图内容了,还有可优化的地方,比如,截取指定大小的图片,截取元素的图片,如果页面很长,单单使用该两种方式是截取不全的,这就需要使用JS的配合了。今天所述内容,有误的地方,希望大家及时指出,我也及时改正。

学习的一点心得,大家都知道,遇到问题,不要逃避,更不要慌张,慢慢来,沉下心来,多度娘,多看博客,终究会解决的。今天在使用第二种方式优化代码截图的时候,由于没有新建目录的那一步,所以一直截图不成功,后面灵机一动,先给创建一个目录,再运行代码,结果就截图成功了,有点小喜悦。

posted @ 2018-09-16 15:42  温一壶清酒  阅读(47757)  评论(5编辑  收藏  举报