java-selenium下载百度图片

package download;

import java.io.DataInputStream;
import java.io.File;
import java.io.FileOutputStream;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;

import org.apache.commons.io.input.ReaderInputStream;
import org.openqa.selenium.By;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.chrome.ChromeDriver;


public class Geturl {
        
        public static List<String> geturls(String baseUrl){
                List<String> urllist=new ArrayList<String>();
                System.setProperty("webdriver.chrome.driver", "E:\\\\webDriver\\\\chromedriverV2.28.exe");
                
                WebDriver driver = new ChromeDriver();
                driver.get(baseUrl);
                
                //获取所有img标签
                List<WebElement> imgList = driver.findElements(By.tagName("img"));
                System.out.println(imgList.size());
                
                try {
                        Thread.sleep(10000);
                } catch (Exception e) {
                        e.printStackTrace();
                }
                //便利所有标签
                try {
                        for (WebElement a : imgList) {
                                //System.out.println(a.getText());
                                System.out.println(a.getAttribute("src"));//获取img标签中的data-imgurl  data-imgurl
                                
                                //获取img标签data-imgurl属性值
                                String urlStr = a.getAttribute("src");
                                /*if(urlStr.contains(".jpg")) {
                                        urllist.add(urlStr);
                                }*/
                                
                        }
                } catch (Exception e) {
                        e.printStackTrace();
                }
                
                
                return urllist;
                
        }
        

        public static void downloadImg(List<String> urllist) throws Exception{
                URL url=null;
                int imageNumber = 0;
                for(String urlString:urllist) {
                        url = new URL(urlString);
                        DataInputStream dis = new DataInputStream(url.openStream());
                        String imageName ="C:\\Users\\0\\Pictures\\插画\\photos"+ imageNumber +".jpg";
                        FileOutputStream fos =new FileOutputStream(new File(imageName));
                        
                        byte[] buffer = new byte[1024];
                        int length;
                        while((length = dis.read(buffer))>0) {
                                fos.write(buffer, 0, length);
                        }
                        dis.close();
                        fos.close();
                        imageNumber++;
                }
        }
        
        
        
        public static void main(String[] args) {
                
                List<String> urllist=geturls("https://image.baidu.com/search/index?tn=baiduimage&ct=201326592&lm=-1&cl=2&ie=gbk&word=%B2%E5%BB%AD&fr=ala&ala=1&alatpl=adress&pos=0&hs=2&xthttps=111111");
                System.out.println(urllist.size());
                
                try {
                        downloadImg(urllist);
                } catch (Exception e) {
                        // TODO Auto-generated catch block
                        e.printStackTrace();
                }
                
        }
}