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();
        
         }
   }
    

}