selenium截取12306验证码图片

from selenium import webdriver
from PIL import Image
from time import sleep

browser = webdriver.Edge()
browser.maximize_window()

# 进入12306登陆页面
browser.get('https://kyfw.12306.cn/otn/resources/login.html')
sleep(1)
# 选择账号密码登录
browser.execute_script('document.querySelector("body > div.login-panel > div.login-box > ul > li.login-hd-account > a").click()')
# 截屏并保存
sleep(1)
browser.save_screenshot('./img.png')
# 定位图片位置
sleep(2)
img_ele = browser.find_element_by_xpath('//*[@id="J-loginImgArea"]')#定位验证码元素
img_location = img_ele.location#验证码图片左上角坐标（[x,y]）
img_size = img_ele.size #验证码长，宽（[x,y]）
#print(img_location,img_size)

#得到验证码图片的准确坐标
x1 = int(img_location['x'])
y1 = int(img_location['y'])
x2 = x1 + int(img_size['width'])
y2 = y1 + int(img_size['height'])

#截图并保存
box = (x1,y1,x2,y2)
print(box)
img = Image.open('./img.png')
img.crop(box).save('./img1.png')#按尺寸截图
browser.quit()