java+selenium+autoIt 实现下载(打印)功能

本贴最后更新于 2122 天前,其中的信息可能已经东海扬尘

java+selenium+autoIt 实现下载(打印)功能

selenium 是一个开源的自动化测试框架,它可以模拟用户对浏览器的操作,进行自动化的测试。但是,它不仅仅只能用来做测试。

AutoIt 是一个使用类似 BASIC 脚本语言免费软件,它设计用于 Windows GUI(图形用户界面)中进行自动化操作。它利用模拟键盘按键,鼠标移动和窗口/控件的组合来实现自动化任务。而这是其它语言不可能做到或无可靠方法实现的(例如 VBScriptSendKeys)。在这个项目中 AutoIt 主要用于在浏览器中点击下载后,保存文件时,对 windos 弹框进行操作。

因为业务需求,需要把一个 html 网页转化成 PDF,本来准备考虑用 iText 来对 html 转为 PDF,但是我们已经拥有 html 的网页了,用 iText 感觉太麻烦,然后又找了一个第三方 html->pdf,结果需要收费等一些其它原因,最终决定采用 selenium+chrome+java+autol 对网页进行导出 PDF。

即使这个是打印成 PDF,也可用于下载

环境:springboot v2.1.2 + maven + selenium 3.141.59 + chrome 71.0.3578.98

1 selenium

1.1 关于 selenium

selenium 是一个自动化测试框架,可以模拟用户的浏览器操作,可以和几门热门的语言相结合,例如 java,python,接下来我将使用 selenuim 来操作 chrome 来模拟用户将网页保存为 PDF。

1.2 selenium 连接 chrome

1.2.1 添加 selenium 的 jar 包
				<!-- Selenium dependency -->
        <dependency>
            <groupId>org.seleniumhq.selenium</groupId>
            <artifactId>selenium-java</artifactId>
            <version>3.141.59</version>
        </dependency>

        <dependency>
            <groupId>org.seleniumhq.selenium</groupId>
            <artifactId>selenium-server</artifactId>
            <version>3.141.59</version>
        </dependency>
1.2.2 下载谷歌浏览器驱动

谷歌浏览器驱动下载

因为我用的是最新的 chrome,所以下载的也是暂时最新的驱动,每个驱动对应的谷歌浏览器版本也不一样,你们可以查看相应的文档,查看自己所对应的驱动

下载完驱动之后,将驱动放置于 resource 目录下的 driver 文件夹内

1.2.3 加载 WebDriver

    /*获取浏览器的连接*/
    public static WebDriver openAccess() {

        //在idea运行的谷歌驱动路径
        System.setProperty("webdriver.chrome.driver",
                "src/main/resources/driver/chromedriver.exe");
        //打jar包后的谷歌驱动路径
		/*String driverPath = System.getProperty("user.dir")+File.separator+"driver"+ File.separator+"chromedriver.exe";*/
        HashMap<String, Object> chromePrefs = new HashMap<>();
        chromePrefs
                .put("profile.default_content_settings.popups", 0);//设置为禁止弹出下载窗口
        chromePrefs
                .put("download.default_directory", downloadFilepath);//设置为文件下载路径
        ChromeOptions options = new ChromeOptions();
        HashMap<String, Object> chromeOptionsMap = new HashMap<>();
        options.setExperimentalOption("prefs", chromePrefs);
        options.addArguments("--test-type");
        options.addArguments("disable-infobars");//取消Chrome正在受到自动测试软件的控制
       /*
        用户浏览器地址,用于加载浏览器的用户信息,
        这一步将增加浏览器的性能消耗,
        如果不加这一行,浏览器默认已访客模式进入浏览器,
        可根据自己的需求来选择是否使用
         */
        options.addArguments("user-data-dir=C:\\Users\\onegene\\AppData\\Local\\Google\\Chrome\\User Data");
        DesiredCapabilities cap = DesiredCapabilities.chrome();
        cap.setCapability(ChromeOptions.CAPABILITY, chromeOptionsMap);
        cap.setCapability(CapabilityType.ACCEPT_SSL_CERTS, true);
        cap.setCapability(ChromeOptions.CAPABILITY, options);
        WebDriver driver = null;
        boolean flag = true;
        while (flag) {
            try {
                flag = false;
                driver = new ChromeDriver(cap);
                //响应时间超过8秒,则重新开启浏览器连接
                driver.manage().timeouts().pageLoadTimeout(15, TimeUnit.SECONDS);
                driver.manage().window().maximize();
//                driver.get(url);

            } catch (Exception e) {
                flag = true;
                if (driver != null) {
                    driver.quit();
                }
                log.info("wait for connection browser ");
            }
        }
        return driver;
    }

System.setProperty("webdriver.chrome.driver","src/main/resources/driver/chromedriver.exe"); 这一步用于在 IDEA 中运行的是否加载驱动,如果,打成 jar 包之后,加载驱动的话,应该用下面一行加载驱动,驱动放在与 jar 包同目录的 dreiver 目录下

String driverPath = System.getProperty("user.dir")+File.separator+"driver"+ File.separator+"chromedriver.exe";
System.setProperty("webdriver.chrome.driver",driverPath);

options.addArguments("user-data-dir=" + DriverPathUtils.getChromePath()); 用户浏览器地址,用于加载浏览器的用户信息,这一步将增加浏览器的性能消耗,如果不加这一行,浏览器默认已访客模式进入浏览器,可根据自己的需求来选择是否使用

1.3 加载 url,进行元素操作

        webDriver.get(url);//自己定义
        Thread.sleep(1000);
				//根绝class寻找元素,并且点击
        WebElement addpBtn = webDriver.findElement(By.className("addp"));
        addpBtn.click();

        Thread.sleep(1000);
				//根据id寻找元素,并且点击
        WebElement genLayoutBtn = webDriver.findElement(By.id("genLayout"));
        genLayoutBtn.click();

注:selenium 切换 tab,并且关闭

String currentWin = webDriver.getWindowHandle();
Set<String> handles = webDriver.getWindowHandles();
for (String handle : handles) {
    if (currentWin.equals(handle)) continue;
    webDriver = webDriver.switchTo().window(handle);
}
webDriver.close();

1.4 调起打印事件

调起如下的谷歌浏览器打印事件

1548236262668png

调起浏览器的打印事件有以下两种方法:

① 点击鼠标右键,然后点击打印按钮

②按住键盘的`Ctrl`+`P`

显而易见,第 ② 中方法更实用,所以我们采用第 ② 种

我查看了 selenium 的键盘事件,发现是有键盘事件的,调起方法如下:

Actions action = new Actions(driver); 
action.keyDown(Keys.CONTROL);// 按下 Ctrl 键
action.sednKeys("p")//按下P键
action.keyUp(Keys.CONTROL);// 释放 Ctrl 键
action.perform()//发送组合按键

本来以为操作键盘的方法很简单,但是不知道为啥 chrome 一直无法调起键盘事件,谷歌了很久,也没有找到原因,网上也有很多人说无法调起。于是查看 jdk 文档,发现 java 提供了一个操作键盘事件的类。如下

1548236953864png

所以,我就用以下方法来实现键盘事件 Ctrl+P

        Robot robot = new Robot();
        robot.keyPress(KeyEvent.VK_CONTROL);
        robot.keyPress(KeyEvent.VK_P);
        robot.keyRelease(KeyEvent.VK_P);
        robot.keyRelease(KeyEvent.VK_CONTROL);

本以为事情终于解决了,然后才发现,SpringBoot 无法启动这个键盘事件,百度之后,才发现有两种实现的办法

① 启动 SpringBoot 钱,给虚拟机添加参数 -ea -Djava.awt.headless=false

② 在 SpringBoot 启动类中添加代码,开启 awt

public static void main(String[] args) {
		System.setProperty("java.awt.headless", "false");
		SpringApplication.run(SeleniumApplication.class, args);
}

1.5 进入保存界面

打印界面进入保存界面只需要按下回车键,所以只需按照上一步操作键盘点击回车键就好了

		Robot robot = new Robot();
        robot.keyPress(KeyEvent.VK_CONTROL);
        robot.keyPress(KeyEvent.VK_P);
        robot.keyRelease(KeyEvent.VK_P);
        robot.keyRelease(KeyEvent.VK_CONTROL);

		//睡眠7S,因为谷歌浏览器进入打印要先进行渲染,这个需要一段时间
        Thread.sleep(7000);
        robot.keyPress(KeyEvent.VK_ENTER);
        robot.keyRelease(KeyEvent.VK_ENTER);

2 autolt

AutoIt 目前最新是 v3 版本,这是一个使用类似 BASIC 脚本语言免费软件,它设计用于 Windows GUI(图形用户界面)中进行自动化操作。它利用模拟键盘按键,鼠标移动和窗口/控件的组合来实现自动化任务。而这是其它语言不可能做到或无可靠方法实现的

2.1 autolt 基本脚本语法

进入到了另存为得界面,这个时候 selenium 就基本上无法进行操作了,因为 selenium 只对谷歌浏览器进行操作,不对 window 窗口进行操作,所以这个时候就需要 autolt 进行处理了。autolt 是一个脚本语言,因为我们只需要保存一个文件,所以并不需要那么深入,所以下面我讲下 autolt 基本使用方法

2.1.1 安装完软件之后,各个文件的作用

1548292617073png
1548292152499png

  • 编辑器:用于编写脚本语言,并且进行调试,按住 F5 进行调试
  • 运行器:将脚本语言进行运行
  • 转脚本为 exe:将编写的脚本转换成 exe 文件,用于 java 或者 python 直接运行
  • 查看窗口信息:用于获取将要操作窗口的基本信息,例如将要操作按钮和编辑框的信息等等
2.1.2 查看窗口信息

1548292617073png

查看窗口信息的方法:按住 Finder Tool 的,然后拖拽到你要操作的位置,这个时候查看 Control 窗口,就能得到你要操作的元素信息

  • Title:当前窗口的标题
  • Calss:当前元素的类型,例如当前窗口为编辑框,则这里显示 Edit
  • Instance:这个是这个窗口的第一个 Edit,所以 Instance 为 1
  • ClassnameNN:可以作为当前元素的唯一 id,但是没有 AdvancedModel 准确

2.2 autolt 保存文件

#该脚本的语法是:  ;分号代表注释,为了博客好看,我先把;换成#
##ControlFocus ( "title", "窗口文本", controlID)   设置输入焦点到指定窗口的某个控件上
#WinWait ( "title题" , "窗口文本" , 超时时间 )  暂停脚本的执行直至指定窗口存在(出现)为止
#ControlSetText ( "title", "窗口文本", controlID, "新文本" )   修改指定控件的文本
#Sleep ( 延迟 )   使脚本暂停指定时间段
#ControlClick ( "title", "窗口文本", 控件ID , 按钮 , 点击次数 )   向指定控件发送鼠标点击命令
#其中,title即AutoIt Window Info识别出的Title字段,controlID即AutoIt Window Info识别
#出的Class和Instance的拼接,如上图拼接后的结果应为:Button1
#ControlClick(
#ControlClick("另存为","保存","Button2")

ControlFocus("另存为","text","1001")
WinWait("[CLASS:#32770]","",100)
ControlSetText("另存为","","[CLASS:Edit; INSTANCE:1]","$CmdLine[1]")
Sleep(1500)
ControlClick("另存为","保存","Button2")



2.3 autolt 打包成 exe

运行转脚本为 exe 的文件,将其转换成 exe 文件,放入 resource 目录下的 driver 下面

2.4 java 运行 exe 文件

//第一个参数是exe路径,后面是执行cmd得参数,这个也就是对应保存PDF的名称
String[] commandArray = {"src/main/resources/driver/pdf.exe", sample.getCode() + ".pdf"};
runtime.exec(commandArray);

3 结尾

虽然我写的这个博客是关于 java+selenium+autolt 将 html 打印成 PDF,但是也可以用到下载文件里边,原理是一样的。在进行爬虫的时候,经常要遇到下载文件的业务,这样可以很好的完成。

  • B3log

    B3log 是一个开源组织,名字来源于“Bulletin Board Blog”缩写,目标是将独立博客与论坛结合,形成一种新的网络社区体验,详细请看 B3log 构思。目前 B3log 已经开源了多款产品:SymSoloVditor思源笔记

    1063 引用 • 3453 回帖 • 203 关注
  • Java

    Java 是一种可以撰写跨平台应用软件的面向对象的程序设计语言,是由 Sun Microsystems 公司于 1995 年 5 月推出的。Java 技术具有卓越的通用性、高效性、平台移植性和安全性。

    3187 引用 • 8213 回帖
  • Selenium
    16 引用 • 13 回帖
  • autolt
    1 引用 • 1 回帖

相关帖子

欢迎来到这里!

我们正在构建一个小众社区,大家在这里相互信任,以平等 • 自由 • 奔放的价值观进行分享交流。最终,希望大家能够找到与自己志同道合的伙伴,共同成长。

注册 关于
请输入回帖内容 ...