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

---

因为最近在搞一个购票的一个爬虫需要获取当前验证码的位置信息进行打码，因为是用的selenium测试工具所以在网上找了多个资料搞出来的，记录下一成果。

# encoding:utf-8from PIL import Imagefrom selenium import webdriverurl = '网站地址'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()