Selenium遇到异常自动截图

public void onException(java.lang.Throwable throwable,

WebDriver driver){

Throwable cause = throwable.getCause();

/*

String cause = throwable.getClass().getName();

ScreenshotException ec = new ScreenshotException(cause);

*/

System.out.println(throwable.getClass().getName());

System.out.println("onException=" + cause);

Date currentTime = new Date();

SimpleDateFormat formatter = new SimpleDateFormat("yyyy-MM-dd-hh-mm-ss");

String dateString = formatter.format(currentTime);

File scrFile = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE);

try {

File screenshot = new File("D:/ddd/"

+ dateString  + ".png");

FileUtils.copyFile(scrFile,screenshot);

} catch (IOException e) {

e.printStackTrace();

}

}