Python + Selenium操作一:截图详解 您所在的位置:网站首页 京东页面截图在哪里 Python + Selenium操作一:截图详解

Python + Selenium操作一:截图详解

2024-07-11 22:08| 来源: 网络整理| 查看: 265

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

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

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的配合了。今天所述内容,有误的地方,希望大家及时指出,我也及时改正。

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



【本文地址】

公司简介

联系我们

今日新闻

    推荐新闻

    专题文章
      CopyRight 2018-2019 实验室设备网 版权所有