java后台截图_Java+Selenium实现后台截图网页，可运行在树莓派上
前言：最近折騰了一個(gè)QQ機(jī)器人，突然有個(gè)靈感。當(dāng)用戶發(fā)送一個(gè)網(wǎng)頁鏈接時(shí)，我想使用Java對(duì)網(wǎng)頁進(jìn)行截圖，然后將截圖文件發(fā)回到QQ上，感覺這個(gè)功能很酷炫，于是昨天(2020-10-29)研究了一晚上，下面對(duì)截圖功能這個(gè)部分進(jìn)行總結(jié)。
一. 概述
首先我在網(wǎng)上查詢了相關(guān)資料(并不多)，最常見的兩種方案：
使用 Java 自帶的 Robot 類，對(duì)電腦屏幕進(jìn)行截圖，不建議使用。
使用 Selenium 工具，對(duì)游覽器進(jìn)行截圖。
由于最后想要運(yùn)行在樹莓派上(ARM 32位)，遇到了不少問題。
受限于樹莓派zero的性能，生成一張截圖可能要30s左右。
二. 準(zhǔn)備與安裝方案一：使用Java自帶Robot類
只需要 JDK1.6 及以上版本即可。
方案二：使用Selenium
第一步：
安裝 chrome 游覽器：
1
2
3#Linux 上建議使用chromium
sudo apt install chromium-browser
#Win和Mac可直接在官網(wǎng)下載
在 chrome 游覽器中，或者命令行執(zhí)行 chromium-browser --version 查看版本號(hào)
并在下面的鏈接中下載相應(yīng)版本號(hào)的 chromedriver，解壓后即可使用。
注意：如果你使用的是32位的Linux(例如樹莓派)或者沒有找到相應(yīng)的版本號(hào)，建議使用命令去安裝：
1
2
3
4#安裝chromedriver
sudo apt-get install chromium-chromedriver
#查看chromedriver的安裝位置
dpkg -L chromium-chromedriver
第二步：
添加 commons-io 與 selenium 依賴，以 Maven 為例。
1
2
3
4
5
6
7
8
9
10
11
12
org.seleniumhq.selenium
selenium-java
3.141.59
commons-io
commons-io
2.8.0
三. 代碼實(shí)現(xiàn)方案一：使用Java自帶Robot類
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15Desktop.getDesktop().browse(new URL(url).toURI());
Robot robot = new Robot();
Dimension d = new Dimension(Toolkit.getDefaultToolkit().getScreenSize());
int width = (int) d.getWidth();
int height = (int) d.getHeight();
// 最大化瀏覽器
robot.keyRelease(KeyEvent.VK_F11);
Image image = robot.createScreenCapture(new Rectangle(0, 0, width,
height));
BufferedImage bi = new BufferedImage(width, height,
BufferedImage.TYPE_INT_RGB);
Graphics g = bi.createGraphics();
g.drawImage(image, 0, 0, width, height, null);
// 保存圖片 注意替換文件路徑
ImageIO.write(bi, "jpg", new File("/Users/ming/Desktop/temp/" + System.currentTimeMillis()+".jpg"));
方案二：使用Selenium
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28public void screenshot(String url) throws IOException{
//設(shè)置系統(tǒng)參數(shù)，第二個(gè)參數(shù)要指向chromedriver的位置
System.setProperty("webdriver.chrome.driver", "/usr/bin/chromedriver");
ChromeOptions options = new ChromeOptions();
options.addArguments("--headless"); // 無窗口模式
options.addArguments("--disable-infobars"); // 禁言消息條(就上面經(jīng)常一條黃色的那個(gè))
options.addArguments("--disable-extensions"); // 禁用插件
options.addArguments("--disable-gpu"); // 禁用GPU
options.addArguments("--no-sandbox"); // 禁用沙盒模式否則會(huì)報(bào)錯(cuò)
options.addArguments("--disable-dev-shm-usage");
options.addArguments("--hide-scrollbars"); // 隱藏滾動(dòng)條
options.addArguments("--window-size=375,190"); // 設(shè)置開啟游覽器時(shí)的分辨率，如果在樹莓派運(yùn)行建議不要設(shè)置此項(xiàng)
//啟動(dòng)chrome實(shí)例
WebDriver driver = new ChromeDriver(options);
// 設(shè)置游覽器打開后調(diào)整大小
// driver.manage().window().setSize(new Dimension(1920, 1080));
//訪問指定url
driver.get(url);
//指定了OutputType.FILE做為參數(shù)傳遞給getScreenshotAs()方法，其含義是將截取的屏幕以文件形式返回。
File srcFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
//利用FileUtils工具類的copyFile()方法保存getScreenshotAs()返回的文件對(duì)象。
FileUtils.copyFile(srcFile, new File("/home/project/screenshot.png"));
//關(guān)閉瀏覽器
driver.quit();
}
注意：如果在樹莓派上運(yùn)行，要按照上面的參數(shù)去設(shè)置，否則會(huì)報(bào)一些奇奇怪怪的錯(cuò)誤。
如果想對(duì)頁面的 特定區(qū)域 進(jìn)行截圖呢？請(qǐng)往下繼續(xù)看。
四. Selenium中花里胡哨的操作
元素定位
首先，Selenium提供定位元素的方法：driver.findElement()
方式
示例
根據(jù)Id查找
driver.findElement(By.id(“id”))
根據(jù)name查找
driver.findElement(By.name(“name”))
根據(jù)鏈接的全部文本查找
driver.findElement(By.linkText(“l(fā)inkText”))
根據(jù)鏈接的部分文本查找
driver.findElement(By.partialLinkText(“partialLinkText”))
根據(jù)ClassName查找
driver.findElement(By.className(“className”))
根據(jù)tagName查找
driver.findElement(By.tagName(“tagName”))
根據(jù)xpath查找
driver.findElement(By.xpath(“xpath”))
表單輸入與提交
除此之外，還可以模擬在游覽器中輸入表單并提交
1
2
3driver.findElement(By.name("firstname")).sendKeys("Lei");//輸入字串
driver.findElement(By.name("lastname")).sendKeys("Hou");
driver.findElement(By.id("submit")).click();
截圖指定區(qū)域
下面直接上代碼。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66public File elementSnapshot(WebElement element) throws Exception{
//創(chuàng)建全屏截圖
File screen = ((TakesScreenshot) this.driver).getScreenshotAs(OutputType.FILE);
BufferedImage image = ImageIO.read(screen);
//獲取元素的高度、寬度
int width = element.getSize().getWidth();
int height = element.getSize().getHeight();
//創(chuàng)建一個(gè)矩形使用上面的高度，和寬度
Rectangle rect = new Rectangle(width, height);
//元素坐標(biāo)
Point p = element.getLocation();
//對(duì)前面的矩形進(jìn)行操作
//TODO 使用可以截全圖的方法(滾動(dòng)條)，暫未找到方式
int w = rect.width; //指定矩形區(qū)域的寬度
int h = rect.height;//指定矩形區(qū)域的高度
int x = p.getX(); //指定矩形區(qū)域左上角的X坐標(biāo)
int y = p.getY(); //指定矩形區(qū)域左上角的Y坐標(biāo)
//driver的分辨率，這里設(shè)置1920*1080
int w_driver = 1920;
int h_driver = 1080;
System.out.println("width:" + w);
System.out.println("height:" + h);
System.out.println("x:" + x);
System.out.println("y:" + y);
System.out.println("y+height:" + (y + h));
System.out.println("x+width:" + (x + w));
/**
* 如果Element的Y坐標(biāo)值加上高度超過driver的高度
* 就會(huì)報(bào)錯(cuò)(y + height) is outside or not
* 退而求其次，調(diào)整圖片的寬度和高度, 調(diào)整到適合driver的分辨率
* 此時(shí)會(huì)截圖driver可見的元素區(qū)域快照
* TODO 如果能找到跨滾動(dòng)條截圖的方式，可以不用裁剪
*/
try {
if (y + h > h_driver) {
h = h - (y + h - h_driver); //
System.out.println("修改后的height:" + h);
System.out.println("修改后的y+height：" + (y + h));
}
//(x + width) is outside or not
if (x + w > w_driver) {
w = x - (x + w - w_driver);
System.out.println("修改后的width：" + w);
System.out.println("修改后的x+width:" + (x + w));
}
BufferedImage img = image.getSubimage(x, y, w, h);
ImageIO.write(img, "png", screen);
System.out.println("Screenshot By element success");
} catch (IOException e) {
// TODO Auto-generated catch block
e.printStackTrace();
}
return screen;
}
那么傳進(jìn)去的參數(shù)： WebElement 怎么獲取呢？
1
2
3
4// 通過driver獲取頁面
driver.get(url);
// 定位到class名為pyqBox的標(biāo)簽
WebElement element = driver.findElement(By.className("pyqBox"));
我們可以通過元素定位的方法來截圖，是不是很方便？
未完待續(xù)
總結(jié)
以上是生活随笔為你收集整理的java后台截图_Java+Selenium实现后台截图网页，可运行在树莓派上的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。
- 上一篇： java字符串数组排序
- 下一篇： SVN服务端与客户端安装(汉化包)以及简