Java+Selenium——利用Robot类截图——整个桌面截图

前面我们介绍了Selenium中TakeScreenshot类来截图，得到的图片是浏览器窗口内的截图。

有时候，只截浏览器窗口内的图是不够的，而且TakeScreenshot截图只针对浏览器的web事件，

假如你在运行脚本过程，windows上有一个其他软件弹出了一个购物的弹窗页面，这个时候就会干扰你截图。

所以，有时候我们需要整个屏幕截图，这个时候我们就需要用到Robot这个类。

 

原理大概是，从电脑屏幕左上角画一个长方形，一直画到屏幕右下角，然后得到File对象，在把这个File是用png还是jpg保存，

复制到一个具体路径，这个就是截图的整个流程。下面我们用脚本来实现这个过程。

package rjcs;

import java.io.File;
import java.awt.Rectangle;
import java.awt.Robot;
import java.awt.Toolkit;
import java.awt.image.BufferedImage;
import java.awt.image.BufferedImage;

import org.apache.commons.io.FileUtils;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.firefox.FirefoxDriver;

import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Date;

import javax.imageio.ImageIO;

public class zmjsj 
{
    
    public static void main(String[] args)
    
    {
         System.setProperty("webdriver.firefox.bin","C:\\Program Files (x86)\\Mozilla Firefox\\firefox.exe");    //设置火狐的安装路径，防止系统找不到
            
         FirefoxDriver driver = new FirefoxDriver();        //初始化FireFox浏览器实例，并打开浏览器
         
        try
        {
             driver.manage().window().maximize();         //最大化窗口
             
             Thread.sleep(3000);        
             
             driver.manage().window().maximize();         //最大化窗口
             
             Thread.sleep(3000);
             
             driver.get("https://www.baidu.com/");  
             
             Thread.sleep(3000);

             
             BufferedImage image = new Robot().createScreenCapture(new Rectangle(Toolkit.getDefaultToolkit().getScreenSize()));    
             ImageIO.write(image, "png", new File("C:\\Users\\del\\Desktop\\screen_robot.png")); 
             
              Thread.sleep(5000);
             
        }catch (Exception e) 
        {
            e.printStackTrace();
        }finally 
        {
            driver.quit();
        
         }
   }
    

}