selenium獲取圖片驗證碼

# encoding:utf-8
from PIL import Image
from selenium import webdriver

url = '網站地址'
driver = webdriver.Chrome()
driver.maximize_window()  # 將瀏覽器最大化
driver.get(url)
# 截取當前網頁並放到E盤下命名為printscreen，該網頁有我們需要的驗證碼
driver.save_screenshot('E:\\printscreen.png') 
imgelement = driver.find_element_by_xpath('//*[@id="loginForm"]/div/ul[2]/li[4]/div/div/div[3]/img')  # 定位驗證碼
location = imgelement.location  # 獲取驗證碼x,y軸坐標
size = imgelement.size  # 獲取驗證碼的長寬
rangle = (int(location['x']), int(location['y']), int(location['x'] + size['width']),
          int(location['y'] + size['height']))  # 寫成我們需要截取的位置坐標
i = Image.open("E:\\printscreen.png")  # 打開截圖
frame4 = i.crop(rangle)  # 使用Image的crop函數，從截圖中再次截取我們需要的區域
frame4.save('E:\\save.jpg') # 保存我們接下來的驗證碼圖片 進行打碼
driver.close()