python-selenium(webdriver)中的自动截屏并获取验证码的位置

---

因为最近在搞一个购票的一个爬虫需要获取当前验证码的位置信息进行打码，因为是用的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()