selenium 定位截图

//由于selenium没法像htmlunit一样获取HtmlImage对象，获取不了流，所以用截图。把验证码给截图出来
package com.cyyun.test;

 

import java.awt.Toolkit;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.net.URL;
import java.util.concurrent.TimeUnit;

import javax.imageio.ImageIO;

import org.apache.commons.io.FileUtils;
import org.cyberneko.html.HTMLElements;
import org.openqa.selenium.By;
import org.openqa.selenium.Dimension;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.Point;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
//import org.openqa.selenium.WebDriver.Navigation;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.firefox.FirefoxWebElement;
import org.openqa.selenium.remote.Augmenter;
import org.openqa.selenium.remote.DesiredCapabilities;
import org.openqa.selenium.remote.RemoteWebDriver;

import com.chonseng.webmon.net.HtmlTools;
import com.cyyun.imgapp.util.URLDownloader;
import com.gargoylesoftware.htmlunit.html.HtmlImage;
import com.gargoylesoftware.htmlunit.html.HtmlImageInput;
 

 

public class SeleniumSinaTwitter  {
    
    

  public static void snapshot(TakesScreenshot drivername, String filename)
  {
      // this method will take screen shot ,require two parameters ,one is driver name, another is file name
      
      
        File scrFile = drivername.getScreenshotAs(OutputType.FILE);
        // Now you can do whatever you need to do with it, for example copy somewhere
        try {
            System.out.println("save snapshot path is:E:/"+filename);
            FileUtils.copyFile(scrFile, new File("E:\\"+filename));
        } catch (IOException e) {
            // TODO Auto-generated catch block
            System.out.println("Can't save screenshot");
            e.printStackTrace();
        } 
        finally
        {
            System.out.println("screen shot finished");
        }
  }
  
  public static void snapshot2(WebDriver drivername, String filename)
  {
      // this method will take screen shot ,require two parameters ,one is driver name, another is file name
      
      
      //  File scrFile = drivername.getScreenshotAs(OutputType.FILE);
        // Now you can do whatever you need to do with it, for example copy somewhere
        try {
        	  WebDriver augmentedDriver = new Augmenter().augment(drivername);
              File screenshot = ((TakesScreenshot) augmentedDriver).getScreenshotAs(OutputType.FILE);
              File file = new File("E:\\"+filename);
              FileUtils.copyFile(screenshot, file);
        } catch (IOException e) {
            // TODO Auto-generated catch block
            System.out.println("Can't save screenshot");
            e.printStackTrace();
        } 
        finally
        {
            System.out.println("screen shot finished");
        }
  }
  
  public static byte[] takeScreenshot(WebDriver driver) throws IOException {
		WebDriver augmentedDriver = new Augmenter().augment(driver);
      return ((TakesScreenshot) augmentedDriver).getScreenshotAs(OutputType.BYTES);
		//TakesScreenshot takesScreenshot = (TakesScreenshot) driver;
		//return takesScreenshot.getScreenshotAs(OutputType.BYTES);
		}

	 public static BufferedImage createElementImage(WebDriver driver,WebElement webElement)
	    		throws IOException {
	    		// 获得webElement的位置和大小。
	    		Point location = webElement.getLocation();
	    		Dimension size = webElement.getSize();
	    		// 创建全屏截图。
	    		BufferedImage originalImage =
	    		ImageIO.read(new ByteArrayInputStream(takeScreenshot(driver)));
	    		// 截取webElement所在位置的子图。
	    		BufferedImage croppedImage = originalImage.getSubimage(
	    		location.getX(),
	    		location.getY(),
	    		size.getWidth(),
	    		size.getHeight());
	    		return croppedImage;
	    		}
  
   public static void login(WebDriver driver)throws Exception{
	   //新浪微博自己发出的评论地址
	   //http://weibo.com/comment/outbox?wvr=1
	   String login = "https://login.sina.com.cn/signup/signin.php?entry=homepage";
	   driver.get(login);
	   Thread.sleep(2000);
	   WebElement keyWord = driver.findElement(By.xpath("//input[@name='username']"));
	   keyWord.clear();
	   keyWord.sendKeys("nbr987654321@163.com");
	   keyWord = driver.findElement(By.xpath("//input[@name='password']"));
	   keyWord.clear();
	   keyWord.sendKeys("XXX");
	   try{
		   keyWord = driver.findElement(By.xpath("//img[@id='check_img']"));
		   if(keyWord!=null){
			   
			   String src =  keyWord.getAttribute("src");
			   System.out.println("验证码地址 ====="+src);
			   if(src!=null && !"".equals(src)){
				   BufferedImage inputbig = createElementImage(driver,keyWord);
				   ImageIO.write(inputbig, "png", new File("e://testasdasd.png")); 
				   String key= "";
				   //验证码输入框
				   keyWord = driver.findElement(By.xpath("//input[@id='door']"));   
				   keyWord.clear();
				   keyWord.sendKeys(key);
			   }
			   
		   }
	   }catch(Exception e){
		   e.printStackTrace();
	   }
	   keyWord = driver.findElement(By.xpath("//input[@class='btn_submit_m']"));
	   keyWord.click();
	   Thread.sleep(2000);
	   try{
		   keyWord = driver.findElement(By.xpath("//img[@id='check_img']"));
		   if(keyWord!=null){
			   String src =  keyWord.getAttribute("src");
			   System.out.println("验证码地址 ====="+src);
			   if(src!=null && !"".equals(src)){
				   snapshot2(driver,"open_baidu_sinaweibo1111111111.png");
				   
				   BufferedImage inputbig = createElementImage(driver,keyWord);
				   ImageIO.write(inputbig, "png", new File("e://testasdasd.png")); 
				   
				   String key= "";
				   //验证码输入框
				   keyWord = driver.findElement(By.xpath("//input[@id='door']"));   
				   keyWord.clear();
				   keyWord.sendKeys(key);
				   //在提交
				   keyWord = driver.findElement(By.xpath("//input[@class='btn_submit_m']"));
				   keyWord.click();
			   }
			   
		   }
	   }catch(Exception e){
		   e.printStackTrace();
	   }
	   Thread.sleep(2000);
   }
   
 

    public static void main (String [] args) throws Exception
    {
         
    	 String serverUrl = "http://192.168.1.107:4444/wd/hub";
        String URL="http://weibo.com/comment/outbox?wvr=1";//http://coral.qq.com/1008591939
        //System.setProperty("webdriver.chrome.driver", "d:\\ie\\chromedriver.exe"); 
        //WebDriver driver = new ChromeDriver();
        DesiredCapabilities capability = DesiredCapabilities.firefox();
        capability.setJavascriptEnabled(true);
        WebDriver driver = new RemoteWebDriver(new URL(serverUrl), capability);
        //如果3s内还定位不到则抛出异常
        driver.manage().timeouts().implicitlyWait(3, TimeUnit.SECONDS);
        //页面加载超时时间设置为5s
       // driver.manage().timeouts().pageLoadTimeout(20, TimeUnit.SECONDS);
        //driver.manage.script_timeout = 3 #seconds
        driver.manage().timeouts().setScriptTimeout(3, TimeUnit.SECONDS);
        login(driver);
        snapshot2(driver,"open_baidu_sinaweibo1.png");
       //跳转新页面
       // driver.navigate().to(URL);
        driver.get(URL);
        //max size the browser
        driver.manage().window().maximize(); 
 /*
        Navigation navigation = driver.navigate();
         navigation.to(URL);*/
         Thread.sleep(2000);
         //snapshot((TakesScreenshot)driver,"open_baidu.png");
         snapshot2(driver,"open_baidu_sinaweibo.png");
       
     
         
          driver.quit();
           
       
    }
    
}