1 packagecom.dbyl.tests;2
3 importjava.awt.Rectangle;4 importjava.awt.image.BufferedImage;5 importjava.io.BufferedReader;6 importjava.io.File;7 importjava.io.FileInputStream;8 importjava.io.IOException;9 importjava.io.InputStreamReader;10 importjava.io.Reader;11 importjava.util.concurrent.TimeUnit;12
13 importjavax.imageio.ImageIO;14
15 importorg.apache.commons.io.FileUtils;16 importorg.openqa.selenium.By;17 importorg.openqa.selenium.OutputType;18 importorg.openqa.selenium.Point;19 importorg.openqa.selenium.TakesScreenshot;20 importorg.openqa.selenium.WebDriver;21 importorg.openqa.selenium.WebElement;22
23 importcom.dbyl.libarary.utils.DriverFactory;24
25 public classTesseractTest {26
27     public static void main(String[] args) throwsIOException,28 InterruptedException {29
30         WebDriver driver =DriverFactory.getChromeDriver();31         driver.get("file:///C:/Users/validation.html");32         driver.manage().timeouts().pageLoadTimeout(30, TimeUnit.SECONDS);33         WebElement element = driver.findElement(By.xpath("//img"));34
35         //take screen shot for element
36         screenShotForElement(driver, element, "D:\\Tesseract-OCR\\test.png");37
38 driver.quit();39
40         //use Tesseract to get strings
41         Runtime rt =Runtime.getRuntime();42         rt.exec("cmd.exe /C  tesseract.exe D:\\Tesseract-OCR\\test.png  D:\\Tesseract-OCR\\test -1 ");43
44         Thread.sleep(1000);45         //Read text
46         readTextFile("D:\\Tesseract-OCR\\test.txt");47 }48
49     /**
50 * This method for read TXT file51 *52 *@paramfilePath53      */
54     public static voidreadTextFile(String filePath) {55         try{56             String encoding = "GBK";57             File file = newFile(filePath);58             if (file.isFile() && file.exists()) { //判断文件是否存在
59                 InputStreamReader read = newInputStreamReader(60                         new FileInputStream(file), encoding);//考虑到编码格式
61                 BufferedReader bufferedReader = newBufferedReader(read);62                 String lineTxt = null;63                 while ((lineTxt = bufferedReader.readLine()) != null) {64 System.out.println(lineTxt);65 }66 read.close();67             } else{68                 System.out.println("找不到指定的文件");69 }70         } catch(Exception e) {71             System.out.println("读取文件内容出错");72 e.printStackTrace();73 }74 }75
76     /**
77 * This method for screen shot element78 *79 *@paramdriver80 *@paramelement81 *@parampath82 *@throwsInterruptedException83      */
84     public static voidscreenShotForElement(WebDriver driver,85             WebElement element, String path) throwsInterruptedException {86         File scrFile =((TakesScreenshot) driver)87 .getScreenshotAs(OutputType.FILE);88         try{89             Point p =element.getLocation();90             int width =element.getSize().getWidth();91             int height =element.getSize().getHeight();92             Rectangle rect = newRectangle(width, height);93             BufferedImage img =ImageIO.read(scrFile);94             BufferedImage dest =img.getSubimage(p.getX(), p.getY(),95 rect.width, rect.height);96             ImageIO.write(dest, "png", scrFile);97             Thread.sleep(1000);98             FileUtils.copyFile(scrFile, newFile(path));99         } catch(IOException e) {100 e.printStackTrace();101 }102 }103
104 }