selenium 截圖 添加時間戳

 

在自動化程序中運行的代碼報錯信息或者是相關日志有可能並無法直觀的判斷出錯信息。因此截圖是避免不了的。為了避免因為重復運行或者是圖片名稱相同導致截圖被覆蓋。

建議在截圖時使用時間戳，保證截圖圖片名稱的唯一性。

 

import java.io.File;
import java.io.IOException;
import java.text.DateFormat;
import java.text.SimpleDateFormat;
import java.util.Calendar;
import org.apache.commons.io.FileUtils;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.chrome.ChromeDriver;
import org.openqa.selenium.chrome.ChromeOptions;

public class GetImg {
    public static void main(String[] args) {
        System.setProperty("webdriver.chrome.driver", "D:/chromedriver_win32/chromedriver.exe");
        ChromeOptions Options = new ChromeOptions();
        Options.addArguments("user-data-dir=C:\\Users\\happy\\AppData\\Local\\Google\\Chrome\\User Data");
        WebDriver driver = new ChromeDriver(Options);
        driver.get("https://www.baidu.com/");
        // 生成以時間戳為名的截圖，避免圖片重復導致的覆蓋
        DateFormat df = new SimpleDateFormat("yyyyMMddHHmmss");
        Calendar calendar = Calendar.getInstance();
        String imageName = df.format(calendar.getTime());
        // 進行截圖 並把截圖保存在指定位置。
        File srcFile = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);
        try {
            FileUtils.copyFile(srcFile, new File("d:\\" + imageName + ".png"));
        } catch (IOException e) {

            e.printStackTrace();
        }
        driver.close();
        driver.quit();

    }
}