.jpg)
Java和Selenium截取超長html頁面完整截圖
使用java和selenium進(jìn)行web自動(dòng)化測試時(shí)，常常需要將html頁面轉(zhuǎn)換為圖片。對于高度較大的頁面，簡單的截圖方法往往無法捕捉完整內(nèi)容。本文改進(jìn)了一種方案，解決使用window.scrollby()方法導(dǎo)致部分內(nèi)容缺失的問題。
問題在于，直接使用window.scrollby()逐段截圖，由于頁面渲染和滾動(dòng)延遲，可能導(dǎo)致部分內(nèi)容未被捕獲。 改進(jìn)后的方法通過控制每次滾動(dòng)的距離，并使相鄰截圖之間有重疊區(qū)域，最終拼接成完整圖片。
以下是一個(gè)簡單的HTML測試頁面示例：
1<br> 2<br> 3<br> 4<br> ... 800<br>
改進(jìn)后的代碼如下（需補(bǔ)充圖片拼接邏輯，此處僅提供核心滾動(dòng)截圖部分）：
立即學(xué)習(xí)“Java免費(fèi)學(xué)習(xí)筆記（深入）”；
import org.openqa.selenium.By; import org.openqa.selenium.JavascriptExecutor; import org.openqa.selenium.webdriver; 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) throws IOException { // ChromeDriver路徑及輸出路徑 (請?zhí)鎿Q為您的實(shí)際路徑) String chromePath = "D:htmlToImgSeleniumchromedriver-win64chromedriver.exe"; String outputDir = "D:htmlToImgSeleniumoutput"; System.setProperty("webdriver.chrome.driver", chromePath); ChromeOptions options = new ChromeOptions(); options.addArguments("--headless"); // 無頭模式 options.addArguments("--window-size=1920,1080"); // 設(shè)置窗口大小 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 = 1000; int overlapHeight = 250; List<BufferedImage> images = new ArrayList<>(); int currentScroll = 0; while (currentScroll < pageHeight) { js.executeScript("window.scrollTo(0," + currentScroll + ")"); wait.until(ExpectedConditions.stalenessOf(driver.findElement(By.tagName("body")))); // 等待頁面滾動(dòng)完成 File screenshot = new File(outputDir + "screenshot_" + currentScroll + ".png"); driver.getScreenshotAs(OutputType.FILE).copy(screenshot); images.add(ImageIO.read(screenshot)); currentScroll += (captureHeight - overlapHeight); } //此處需要添加圖片拼接邏輯，將images列表中的圖片拼接成一張完整的圖片 // ... 圖片拼接代碼 ... } finally { driver.quit(); } } // ... 圖片拼接方法 (需要自行實(shí)現(xiàn)) ... }
這段代碼的關(guān)鍵改進(jìn)在于：
- 引入重疊區(qū)域: overlapHeight 變量控制每次截圖的重疊部分，避免因滾動(dòng)導(dǎo)致內(nèi)容丟失。
- 等待頁面渲染: 使用 wait.until(ExpectedConditions.stalenessOf(…)) 確保頁面完全滾動(dòng)到指定位置后再進(jìn)行截圖。
- 分段截圖: 循環(huán)滾動(dòng)并截圖，每次滾動(dòng)距離為 captureHeight – overlapHeight。
注意: 代碼中省略了圖片拼接部分，這需要根據(jù)實(shí)際情況選擇合適的圖片拼接算法實(shí)現(xiàn)。 可以使用第三方庫，例如 java.awt.image 包中的方法，或者其他圖像處理庫來完成拼接。 選擇合適的 captureHeight 和 overlapHeight 值也很重要，需要根據(jù)頁面內(nèi)容和性能進(jìn)行調(diào)整。 確保您已正確配置 ChromeDriver 并安裝了必要的依賴庫。
? 版權(quán)聲明
文章版權(quán)歸作者所有，未經(jīng)允許請勿轉(zhuǎn)載。
THE END
.png)
