selenium2java爬虫示例

package wepractice;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Set;
import selenium.Library;
import selenium.Excel;

public class NeiHanjokes extends Library{
    public static void main(String[] args) {
        Library library = new Library();
        List<String[]> sheet = new ArrayList<String[]>();//新建list，用于存放每个测试用例的测试结果
        Map<Integer, List<String[]>> dateJoke = new HashMap<Integer, List<String[]>>();

        driver.get("http://neihanshequ.com/");
        String home = driver.getWindowHandle();
        library.findElementByXpathAndClick(".//*[@id='detail-list']/li[1]/div/div[2]/a/div/h1/p");
        Set<String> handles = driver.getWindowHandles();
        for(String handle : handles){
            if (!handle.equals(home)) {
                driver.switchTo().window(handle);
            }
        }
        for (int i = 0; i < 15; i++) {
            library.output(i);
            String joke = library.getTextByXpath("html/body/div[3]/div[1]/div/ul/li[1]/div/div[2]/a/div/h1/p");
            String[]  jokes = new String[1];
            jokes[0] = joke;
            sheet.add(jokes);
            library.findElementByIdAndClick("prevGroupLink");
        }

        dateJoke.put(1, sheet);
        Excel excel = new Excel();
        excel.writeXlsx(dateJoke);
        driver.close();//关闭窗口
        for(String handle : handles){//切换窗口
            if (handle.equals(home)) {
                driver.switchTo().window(handle);
            }
        }
        driver.quit();//退出浏览器
    }
}
技术类文章精选