关于canvas保存成为图片基本都不能和selenium进行很好的配合,canvas保存成图片应用较多的是图片验证或滑动验证的图片保存,下面是利用selenium直接将canvas保存成图片的代码
public static void main(String[] args) {
// web驱动的存放路径
String webDriverDir = "web驱动的存放路径 ";
// 加载驱动
System.setProperty("webdriver.chrome.driver", webDriverDir);
// 设置浏览器参数
ChromeOptions options = new ChromeOptions();
// 创建驱动对象
WebDriver driver = new ChromeDriver(options);
// 打开链接
driver.get("链接地址");
// 查找画布这里用的是tagName用其他的id或className也完全可以
WebElement canvas = driver .findElement(By.tagName("canvas"));
// 这句到代码是关键执行完后返回的图片的base64 字符串编码
Object base64 = ((JavascriptExecutor) driver)
.executeScript("return arguments[0].toDataURL('image/png').substring(21);", canvas);
String base64Str = base64.toString();
// 将base64把字符串装换成图片
generateImage(base64Str , "要存放图片的路径");
}
/**
* @Description: 将base64编码字符串转换为图片
* @Author:
* @CreateTime:
* @param imgStr base64编码字符串
* @param path 图片路径-具体到文件
* @return
*/
public static boolean generateImage(String imgStr, String path) {
if (imgStr == null)
return false;
// 解密
try {
Decoder decoder = java.util.Base64.getDecoder();
byte[] b = decoder.decode(imgStr);
// 处理数据
for (int i = 0; i < b.length; ++i) {
if (b[i] < 0) {
b[i] += 256;
}
}
OutputStream out = new FileOutputStream(path);
out.write(b);
out.flush();
out.close();
return true;
} catch (IOException e) {
return false;
}
}