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‘)

selenium - 截取頁面圖片和截取某個元素的圖