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

时间:2024-05-22 22:42:39

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调起打印事件

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

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

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

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

②按住键盘的`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提供了一个操作键盘事件的类。如下

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

所以,我就用以下方法来实现键盘事件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 安装完软件之后,各个文件的作用

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

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

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

查看窗口信息的方法:按住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,但是也可以用到下载文件里边,原理是一样的。在进行爬虫的时候,经常要遇到下载文件的业务,这样可以很好的完成。