java+selenium实现谷歌PDF自动化下载(打印)功能

selenium是一个开源的自动化测试框架,它可以模拟用户对浏览器的操作,进行自动化的操作。
接下来我将使用java+selenium+WebDriver来操作chrome来模拟用户将PDF保存到本地磁盘目录中。

1.添加selenium的jar包

<dependency>
    <groupId>org.seleniumhq.selenium</groupId>
    <artifactId>selenium-java</artifactId>
    <version>3.4.0</version>
</dependency>

2.模拟操作保存PDF等文件操作
在这里插入图片描述
调起浏览器的打印事件有以下两种方法:

① 点击如图所示的下载按钮,然后点击打印按钮(但是没有xpath路径,比较麻烦)

②按住键盘的Ctrl+s,直接就可以利用浏览器的默认设置来保存文件了。

这里采用第二种方式,代码如下,主要用Robot**模拟键盘操作**
ps: 注册驱动那些代码这里就没写了,单纯是模拟保存操作。

// 2023-05-26 不知道为啥PDF不能下载操作了,故新增下面自动化下载操作代码 linZT
// 切换到新窗口
String originalHandle = driver.getWindowHandle();
for (String handle : driver.getWindowHandles()) {
   if (!handle.equals(originalHandle)) {
      driver.switchTo().window(handle);
      break;
   }
}

System.out.println("在PDF预览窗口");
Thread.sleep(1000L);

// 使用快捷键模拟点击下载按钮
//Actions actions = new Actions(driver);
//actions.keyDown(Keys.CONTROL).sendKeys("s").keyUp(Keys.CONTROL).perform();
Robot robot = new Robot();
robot.keyPress(KeyEvent.VK_CONTROL);
robot.keyPress(KeyEvent.VK_S);
robot.keyRelease(KeyEvent.VK_S);
robot.keyRelease(KeyEvent.VK_CONTROL);
Thread.sleep(6000L);
System.out.println("执行Ctrl+s保存PDF文件操作");
// 切换到下载框
driver.switchTo().activeElement();
Thread.sleep(1000L);
// 模拟按下Enter键
robot.keyPress(KeyEvent.VK_ENTER);
robot.keyRelease(KeyEvent.VK_ENTER);
Thread.sleep(4000L);

// 关闭新窗口并切换回原始窗口
driver.close();
System.out.println("关闭PDF窗口并切换回原始窗口");
// 切换回原来的窗口
driver.switchTo().window(originalHandle);
System.out.println("切回原始窗口");

3.上述代码执行后没有效果的话,需要在启动项目时加入以下配置
①启动SpringBoot钱,给虚拟机java -jar后面添加参数-ea -Djava.awt.headless=false
②在SpringBoot启动类中添加代码,开启awt

@SpringBootApplication
public class ToolsApplication {
    public static void main(String[] args) {
        //System.setProperty("java.awt.headless", "false");
        SpringApplicationBuilder builder = new SpringApplicationBuilder(new Class[]{ToolsApplication.class});
        builder.headless(false).run(args);
    }
}

4.调用后成功保存到本地默认下载的磁盘目录
在这里插入图片描述

Logo

技术共进,成长同行——讯飞AI开发者社区

更多推荐