Selenium - java - 屏幕截图

文档地址

Selenium 浏览器自动化项目 | Selenium

安装

        <dependency><groupId>org.seleniumhq.selenium</groupId><artifactId>selenium-java</artifactId><version>4.19.1</version></dependency>

使用

创建WebDriver实例

以chrome为例，创建

ChromeOptions co = new ChromeOptions();
co.setCapability("webSocketUrl", true);  //BiDi功能需要添加这个配置
co.addArguments("--headless");           //无头浏览器
WebDriver driver = new ChromeDriver(chromeOptions);/*********************************************
*            操作                            *
**********************************************/driver.quit();

打开目标页面

driver.get("https://www.selenium.dev/selenium/web/web-form.html");

捕捉截图

首先要开启bidi功能

截图可以选择截取一个页面的图，也可以选择某一元素进行截图

先看一个截取某一页面的代码例子：

// 首先创建一个BrowserContext对象BrowsingContext bc = new BrowsingContext(driver, driver.getWindowHandle());
String base64Str = bc.captureScreenshot();

截图返回的数据是一个图片的base64编码字符串

如果将截图文件保存到一个文件中，可以参考如下代码

        byte[] decode = Base64.getDecoder().decode(base64Str);try(FileOutputStream fos = new FileOutputStream(new File("test.png"))){fos.write(decode);fos.close();} catch (FileNotFoundException e) {throw new RuntimeException(e);} catch (IOException e) {throw new RuntimeException(e);}

这段代码会把截图文件保存为test.png文件中

如果想要截取某一个元素节点截图，我们可以使用如下代码

WebElement element = driver.findElement(new By.ByClassName("py-2"));
String base64Str = bc.captureElementScreenshot(((RemoteWebElement)element).getId());