selenium获取手机验证码自动登录

为了自动登录，我们需要获取手机验证码。以下是获取验证码并自动登录的步骤：

1.使用Selenium打开登录页面，并输入用户名和密码。

from selenium import webdriver
import time

driver = webdriver.Chrome()
driver.get("https://example.com/login")
username = driver.find_element_by_name("username")
password = driver.find_element_by_name("password")
username.send_keys("your_username")
password.send_keys("your_password")

2.获取验证码位置并截取验证码保存。

from PIL import Image

# 获取验证码位置
code_element = driver.find_element_by_xpath("//img[@class='code-img']")
location = code_element.location
size = code_element.size
left = location['x']
top = location['y']
right = left + size['width']
bottom = top + size['height']

# 截取验证码并保存
driver.save_screenshot("screenshot.png")
image = Image.open("screenshot.png")
code_image = image.crop((left, top, right, bottom))
code_image.save("code.png")

3.调用百度API识别验证码。

from aip import AipOcr

# 设置百度API信息
APP_ID = 'your_app_id'
API_KEY = 'your_api_key'
SECRET_KEY = 'your_secret_key'
client = AipOcr(APP_ID, API_KEY, SECRET_KEY)

# 读取验证码图片
with open("code.png", "rb") as f:
    image = f.read()

# 调用百度API识别验证码
result = client.basicGeneral(image)
code = result['words_result'][0]['words']

4.输入验证码并点击登录按钮。

# 输入验证码
code_input = driver.find_element_by_name("code")
code_input.send_keys(code)

# 点击登录按钮
login_button = driver.find_element_by_xpath("//button[@class='login-btn']")
login_button.click()