selenium中页面截图和元素截图的方法

一、页面截图

  selenium中页面截图的方法比较简单,可以直接使用selenium自带的截图方式save_screenshot(‘filename’)。

from selenium import webdriver
driver = webdriver.Chrome()
driver.get("http://xlrz.chdi.com.cn/wssq/")
driver.save_screenshot("login.png")

  注意:save_screenshot的文件后缀名只能是png。

  get_screenshot_as_flie("文件路径"),与save_screenshot(‘filename’)功能相似。不过get_screenshot_as_flie("文件路径")可以指定文件路径,而save_screenshot(‘filename’)是默认在项目目录下生成图片。

二、元素截图

  元素截图需要先安装第三方pillow库,安装命令是“pip install pillow”.代码中需要先导入Image模组。

  例如获取学历认证登录页面,【登录】按钮具体代码如下:

from selenium import webdriver
from PIL import Image

driver = webdriver.Chrome()
driver.get("http://xlrz.chdi.com.cn/wssq/")
#获取登录页面的截图
driver.save_screenshot('login.png')
#定位登录按钮
imgcode=driver.find_element_by_name("submit")
#loaction方法获取元素坐标值X,y,且以字典的方式返回
left=imgcode.location['x']
top=imgcode.location['y']
#size方法获取元素的高度和宽度,以字典方式返回
right=left+imgcode.size['width']
bottom=top+imgcode.size['height']
#打开登录页面的截图
im=Image.open("login.png")
#获取登录按钮截图
mg=im.crop((left,top,right,bottom))
#保存截图,命名未ann
mg.save('ann.png')

  

上一篇:Unity 任意区域截图


下一篇:Mac截图