使用Python识别和填充简单验证码

在网络爬虫和自动化测试中，经常会遇到需要识别和填充验证码的情况。本文将介绍如何使用Python和第三方库来识别简单的验证码，并自动填充到表单中。

1. 下载验证码图片
首先，我们需要下载验证码图片并保存到本地。这里以requests库为例，发送GET请求获取验证码图片。

import requests

# 下载验证码图片
url = "https://example.com/captcha.png"  # 替换为实际的验证码图片URL
response = requests.get(url)
with open("captcha.png", "wb") as f:
    f.write(response.content)
2. 预处理验证码图片
接下来，我们需要对验证码图片进行预处理，例如将其转换为灰度图像，以便后续识别。

from PIL import Image

# 打开验证码图片
captcha_img = Image.open("captcha.png")

# 转换为灰度图像
captcha_img = captcha_img.convert("L")

# 保存处理后的图片
captcha_img.save("captcha_gray.png")
3. 使用OCR库识别验证码数字
我们可以使用第三方OCR库，例如pytesseract，来识别验证码中的数字。

import pytesseract

# 识别验证码数字
captcha_text = pytesseract.image_to_string(captcha_img)

# 去除识别结果中的非数字字符
captcha_text = "".join(filter(str.isdigit, captcha_text))

print("识别结果：", captcha_text)
4. 填充验证码到表单
最后，我们将识别出的验证码数字填充到表单中的验证码输入框，并提交表单。

from selenium import webdriver
from selenium.webdriver.common.keys import Keys

# 初始化浏览器
driver = webdriver.Chrome()

# 打开目标网页
driver.get("https://example.com/login")

# 填充验证码到表单
captcha_input = driver.find_element_by_id("captcha-input")
captcha_input.send_keys(captcha_text)

# 提交表单
captcha_input.send_keys(Keys.ENTER)

更多内容联系q1436423940