我们执行UI自动化操作时,大多数时间都是不在现场的,出现错误时,没有办法第一时间查看到,这时我们可以通过截图当时出错的场景保存下来,后面进行查看报错的原因,Selenium中提供了几种截图的方法,小编今天给大家介绍下。 get_screenshot_as_base64()整个函数,获取当前屏幕页面截图。
我们通过代码编写打开百度页面,然后进行进行通过base64进行截图,截图后再次进行转换成保存下来。 from selenium import webdriver
driver=webdriver.Chrome()
driver.get('https://www.baidu.com/')
x=driver.get_screenshot_as_base64()
image=base64.b64decode(x)
save_screenshot(filename)是属于Selenium自带的截图函数,其中filename表示名称,道理一样,直接打开百度网页进行保存。
from selenium import webdriver
driver=webdriver.Chrome()
driver.get('https://www.baidu.com/')
driver.save_screenshot('1.png')
get_screenshot_as_file(filename)获取截图信息,其中filename表示名称,这里需要注意的是,信息只能保存为png格式的,其他的格式,代码会出现报错。
from selenium import webdriver
driver=webdriver.Chrome()
driver.get('https://www.baidu.com/')
driver.get_screenshot_as_file('1.png')
get_screenshot_as_png()通过二进制的方式进行保存。
from selenium import webdriver
driver=webdriver.Chrome()
driver.get('https://www.baidu.com/')
x=driver.get_screenshot_as_png()
上述截图方法中均是截取的当前页面的,那么如果我想要整个浏览器显示的内容全部截图下来怎么操作呢?其实方法一样的,我们只需要获取到浏览器的高度,然后就可以将截图全部都截取下来了。
from selenium import webdriver
opt=webdriver.ChromeOptions()
opt.headless=True#把Chrome设置成可视化无界面模式
driver=webdriver.Chrome(options=opt)#创建Chrome无界面对象
driver.get('http://www.51testing.com/html/news.html')
height=driver.execute_script("return document.documentElement.scrollHeight")
driver.set_window_size(1960,height)
driver.save_screenshot('22.png')
需要注意的一点是,这里需要将浏览器设置成无页面模式,因为只有无页面形式才可以截全图,不然截的还是当前页面的,也就是浏览器的页面显示的。
执行完成代码后,发现我们已经将长图全部截取下来了。
本篇的内容可能对于一些小伙们来说很简单,其实小编主要想介绍如何通过Selenium进行截取长图或者说整个页面的,当然如何在公司项目中进行使用,这个可能就需要小伙们根据公司项目进行合理安排,是否有此需求。
最后:
添加微信:atstudy-js 或者扫描下方二维码,备注“博客”邀你进群