1.截取页面图片并保存

• 在测试过程中，是有必要截图，特别是遇到错误的时候进行截图。

# coding:utf-8

from time import sleep

from PIL import Image

from selenium import webdriver

driver = webdriver.Firefox()

driver.get("https://www.baidu.com")

sleep(2)

driver.get_screenshot_as_file('D:\\baidu.png')  # 把截取的图片存放在D盘

driver.quit()

2.截取某个元素的图

• 先安装pillow

pip install pillow

# coding:utf-8

from time import sleep

from PIL import Image

from selenium import webdriver

driver = webdriver.Firefox()

driver.get("https://www.baidu.com")

sleep(2)

driver.save_screenshot('baidu.png')  # 截取当前页面全图

element = driver.find_element_by_id("su")  # 百度一下的按钮

print("获取元素坐标：")

location = element.location

print(location)

print("获取元素大小：")

size = element.size

print(size)

# 计算出元素上、下、左、右 位置

left = element.location['x']

top = element.location['y']

right = element.location['x'] + element.size['width']

bottom = element.location['y'] + element.size['height']

im = Image.open('baidu.png')

im = im.crop((left, top, right, bottom))

im.save('D:\\baidu.png')