破解验证码的实用技巧：使用 Scala 和 Selenium 实现

在现代网页应用中，图形验证码经常作为登录的一部分，给自动化测试带来了挑战。本文将介绍两种有效的方法来处理验证码：获取验证码图片链接和使用 Selenium 进行截屏和裁剪。

方法一：获取验证码图片地址并下载
我们可以尝试获取验证码的图片链接并下载它。以下是 Scala 语言的实现代码：

scala

import org.openqa.selenium.{WebDriver, WebElement}
import org.openqa.selenium.chrome.ChromeDriver
import java.io.{File, FileOutputStream}
import java.net.URL
import scala.util.Random

object CaptchaDownloader {
def main(args: Array[String]): Unit = {
// 设置 ChromeDriver 路径
System.setProperty("webdriver.chrome.driver", "path/to/chromedriver")
val driver: WebDriver = new ChromeDriver()

try {
// 打开登录页面
driver.get("http://example.com/login")

// 获取验证码图片链接
val captchaElement: WebElement = driver.findElement(By.id("captchaImage"))
val captchaSrc: String = captchaElement.getAttribute("src")
val imgUrl = s"$captchaSrc.png"
val fileName = s"img/login/${Random.nextInt(100000)}.png" // 随机文件名

// 下载验证码图片
val url = new URL(imgUrl)
val connection = url.openConnection()
val inputStream = connection.getInputStream
val outputStream = new FileOutputStream(new File(fileName))

val buffer = new ArraybytesRead = 0
while ( {
bytesRead = inputStream.read(buffer)
bytesRead != -1
}) {
outputStream.write(buffer, 0, bytesRead)
}

outputStream.close()
inputStream.close()
println(s"验证码图片保存成功: $fileName")
} finally {
driver.quit()
}
}
}
方法二：使用 Selenium 截屏并裁剪验证码
如果获取链接不可靠，我们可以通过截屏获取验证码。以下是 Scala 的实现代码：

scala

import org.openqa.selenium.{By, WebDriver, WebElement}
import org.openqa.selenium.chrome.ChromeDriver
import org.openqa.selenium.OutputType
import org.apache.commons.io.FileUtils
import java.io.File
import javax.imageio.ImageIO
import java.awt.image.BufferedImage
import java.awt.Rectangle

object CaptchaCropper {
def main(args: Array[String]): Unit = {
// 设置 ChromeDriver 路径
System.setProperty("webdriver.chrome.driver", "path/to/chromedriver")
val driver: WebDriver = new ChromeDriver()
更多内容访问ttocr.com或联系1436423940
try {
// 打开登录页面
driver.get("http://example.com/login")

// 截取整个页面
val fullScreenshot: File = driver.getScreenshotAs(OutputType.FILE)

// 定位验证码元素
val captchaElement: WebElement = driver.findElement(By.id("captchaImage"))
val location = captchaElement.getLocation
val size = captchaElement.getSize

// 裁剪验证码
val img = ImageIO.read(fullScreenshot)
val captchaImg = img.getSubimage(location.getX, location.getY, size.getWidth, size.getHeight)
val captchaFileName = s"img/login/${Random.nextInt(100000)}_captcha.png"
ImageIO.write(captchaImg, "png", new File(captchaFileName))

println(s"验证码裁剪并保存成功: $captchaFileName")
} finally {
driver.quit()
}
}
}
输入验证码并登录
最后，我们可以使用 Tesseract OCR 库识别验证码的内容，并将其输入到登录框中。以下是示例代码：

scala

import org.openqa.selenium.{By, WebDriver}
import org.openqa.selenium.chrome.ChromeDriver
import net.sourceforge.tess4j.{Tesseract, TesseractException}
import java.io.File

object CaptchaLogin {
def main(args: Array[String]): Unit = {
// 设置 ChromeDriver 路径
System.setProperty("webdriver.chrome.driver", "path/to/chromedriver")
val driver: WebDriver = new ChromeDriver()

try {
// 打开登录页面
driver.get("http://example.com/login")

// 输入用户名和密码
driver.findElement(By.id("username")).sendKeys("your_username")
driver.findElement(By.id("password")).sendKeys("your_password")

// 识别验证码并输入
val captchaFilePath = "img/login/your_captcha_file.png" // 替换为实际文件路径
val tesseract = new Tesseract()
tesseract.setDatapath("path/to/tessdata")
var captchaText = ""
try {
captchaText = tesseract.doOCR(new File(captchaFilePath))
} catch {
case e: TesseractException => println(s"错误：${e.getMessage}")
}

driver.findElement(By.id("captchaInput")).sendKeys(captchaText.trim)

// 提交登录
driver.findElement(By.id("loginButton")).click()

// 检查登录状态
if (driver.getPageSource.contains("欢迎")) {
println("登录成功！")
} else {
println("登录失败，验证码可能输入错误。")
}
} finally {
driver.quit()
}
}
}