前面我们介绍了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();
}
}
}