Java和Selenium：完美实现HTML页面截图，告别内容丢失

在使用Java和Selenium进行Web自动化测试时，将HTML页面转换为图片用于报告生成或调试非常常见。然而，当页面过长时，完整截图常常面临挑战。本文提供一种解决方案，确保HTML页面完整截图，避免内容丢失。

挑战：长页面截图难题

将HTML页面转换为图片时，我们可能遇到以下问题：

1. 页面较短： 直接截图即可。
2. 页面过长： 即使设置了很大的高度，也可能无法一次性截取完整页面。使用window.scrollby(0,x)方法分段截图，容易导致部分内容丢失。

解决方案：分段截图+图像拼接

为了解决内容丢失问题，我们采用分段截图并拼接的方法。关键在于控制每次滑动的距离，确保截图之间存在重叠，从而避免内容遗漏。

代码实现

以下Java代码使用Selenium实现HTML页面截图，并解决内容丢失问题：

AI舌诊面诊问诊医疗系统+诊疗商城系统源码

AI 舌诊面诊问诊医疗系统 + 诊疗商城系统源码是一套结合 AI 辅助诊断与商品转化的健康类程序，基于浩创云、ThinkPHP（PHP）开发，支持微信服务号与 H5 端，适合做中医舌诊面诊问答、健康建议推送以及配套诊疗商品的运营。

下载

立即学习“Java免费学习笔记（深入）”；

import org.openqa.selenium.*;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;
import org.openqa.selenium.support.ui.ExpectedConditions;
import org.openqa.selenium.support.ui.WebDriverWait;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

public class HtmlToImage_Selenium {

    public static void main(String[] args) {

        // 路径设置 (请替换为您的实际路径)
        String chromePath = "D:\htmlToImg\Selenium\chromedriver-win64\chromedriver.exe";
        String mergePath = "D:\htmlToImg\Selenium\output\merge.png";
        String tempPath = "D:\htmlToImg\Selenium\output\screenshot_";

        System.setProperty("webdriver.chrome.driver", chromePath);

        ChromeOptions options = new ChromeOptions();
        options.addArguments("--headless"); // 无头模式
        options.addArguments("--window-size=2160,1440"); // 窗口大小

        WebDriver driver = new ChromeDriver(options);

        try {
            driver.get("file:///D:/htmlToImg/Test.html");

            WebDriverWait wait = new WebDriverWait(driver, 15);
            wait.until(ExpectedConditions.visibilityOfElementLocated(By.tagName("body")));

            JavascriptExecutor js = (JavascriptExecutor) driver;
            long pageHeight = (long) js.executeScript("return document.body.scrollHeight;");
            System.out.println("页面高度为" + pageHeight);

            int captureHeight = 5000; // 每次截取的高度
            int overlapHeight = 2000; // 重叠高度
            int numberOfScreenshots = (int) Math.ceil((double) pageHeight / (captureHeight - overlapHeight));

            List<BufferedImage> images = new ArrayList<>();
            for (int i = 0; i < numberOfScreenshots; i++) {
                long scrollPosition = i * (captureHeight - overlapHeight);
                js.executeScript("window.scrollTo(0," + scrollPosition + ")");

                // 添加等待，确保页面滚动完成
                wait.until(ExpectedConditions.stalenessOf(driver.findElement(By.tagName("body"))));

                File screenshot = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
                BufferedImage image = ImageIO.read(screenshot);
                images.add(image);
            }

            // (此处需要添加图像拼接代码，根据实际情况选择合适的图像拼接库)
            // ...  图像拼接逻辑 ...

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            driver.quit();
        }
    }
}

注意： 代码中省略了图像拼接部分。 你需要选择一个合适的图像处理库（例如，Java的java.awt.image包或第三方库）来实现图像拼接功能。 拼接逻辑需要考虑重叠部分的处理，以确保最终图片的完整性和无缝衔接。

通过分段截图和图像拼接，可以有效解决长页面截图时内容丢失的问题，确保生成完整、高质量的HTML页面截图。 记住替换代码中的路径为您的实际路径。