webdriver 截圖的方法實現工具類

package util;

import java.io.File;
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;

import org.apache.commons.io.FileUtils;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
/**
 * 該類用於實現截圖功能
 * sceenshot工具方法
 *
 */
public class Sceenshot {
	//定義截圖的文件名，去當前時間
	 public static SimpleDateFormat formatter=new SimpleDateFormat("yyyy-MM-dd-hh-mm-ss");	 
	 public static  String name=formatter.format(new Date());
	 //定義圖片保存的路徑
	 public static  String Dir="D:\\p_w_picpaths"; 
	 
       public static void sceenshot(WebDriver dr) {  
    	   //判定目錄是否存在，不存在則創建
    	   File dirpath=new File(Dir);
    	   if (! dirpath.isDirectory()){
    		   dirpath.mkdirs();
    	   }
  
    	//將截圖以文件的形式返回
   		File srcFile=((TakesScreenshot)dr).getScreenshotAs(OutputType.FILE);
   		//創建截圖的位置和文件名
   		File filepath=new File(Dir,name+".png");
   		try {
   			//使用copyFile()方法保存截圖到指定位置
   			FileUtils.copyFile(srcFile, filepath);
   		} catch (IOException e) {
   			// TODO Auto-generated catch block
   			System.out.println("sceenshot p_w_picpaths fail!!!!");
   		}
   	}
}