phantomjs+selenium

package zhaohui;

import java.io.BufferedReader;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.io.IOException;
import java.io.InputStreamReader;
import java.sql.DriverPropertyInfo;
import java.util.ArrayList;
import java.util.List;
import java.util.logging.Level;

import org.apache.commons.io.FileUtils;
import org.openqa.selenium.By;
import org.openqa.selenium.OutputType;
import org.openqa.selenium.TakesScreenshot;
import org.openqa.selenium.WebDriver;
import org.openqa.selenium.WebElement;
import org.openqa.selenium.logging.LogType;
import org.openqa.selenium.logging.LoggingPreferences;
import org.openqa.selenium.phantomjs.PhantomJSDriver;
import org.openqa.selenium.phantomjs.PhantomJSDriverService;
import org.openqa.selenium.remote.CapabilityType;
import org.openqa.selenium.remote.DesiredCapabilities;



public class Yaogeili {
    private String phantomjsPath = null;
    private WebDriver driver = null;
    private DesiredCapabilities caps = null;

    public Yaogeili(String path, String ua) {
        this.phantomjsPath = path;
        caps = DesiredCapabilities.phantomjs();
        caps.setJavascriptEnabled(true);
        caps.setCapability("takesScreenshot", true);
        caps.setCapability(
                PhantomJSDriverService.PHANTOMJS_EXECUTABLE_PATH_PROPERTY,
                this.phantomjsPath);
        caps.setCapability("phantomjs.page.settings.userAgent", ua);
        caps.setCapability("output-encoding", "UTF-8");
        caps.setCapability("script-encoding", "UTF-8");
        //关闭phantomjs的log，目前只能减少log信息，无法彻底关闭
        LoggingPreferences logPrefs = new LoggingPreferences();
        logPrefs.enable(LogType.BROWSER, Level.OFF);
        logPrefs.enable(LogType.DRIVER, Level.OFF);
        logPrefs.enable(LogType.CLIENT, Level.OFF);
        logPrefs.enable(LogType.PERFORMANCE, Level.OFF);
        logPrefs.enable(LogType.PROFILER, Level.OFF);
        logPrefs.enable(LogType.SERVER, Level.OFF);
        caps.setCapability(CapabilityType.LOGGING_PREFS, logPrefs);
        this.driver=new PhantomJSDriver(caps);
    }

    public void gethtml(String query){
        String url = " style="color:#0000CC;">+java.net.URLEncoder.encode(query)+"&type=m-map_bus2";
//        System.setProperty("webdriver.chrome.driver", "files/chromedriver.exe");
//        WebDriver driver = new ChromeDriver();
        this.driver.get(url);
        driver.manage().window().maximize();
        try {
            Thread.sleep(1000);
        } catch (InterruptedException e1) {
            // TODO Auto-generated catch block
            e1.printStackTrace();
        }
        try{
            WebElement webelement =this.driver.findElement(By.xpath("//*[@id='J-iframe-html']/iframe"));
            this.driver.switchTo().frame(webelement);
        }catch(Exception e){
            System.out.println(e);
        }
        try{
            this.driver.findElement(By.xpath("//*[@id='mohe-m-super-map-medicine-broad']"));
            System.out.println(query+"ok");
        }
        catch (Exception e){
            this.takeScreen("test-output/screen", query);
            System.out.println(query);
        }
//        this.driver.quit();
    }
    public String takeScreen(String screenParentPath,String screenName)
    {
        this.driver.getCurrentUrl();
        File file=((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
        String screenAdress=null;
        try {
            FileUtils.copyFile(file, new File(screenParentPath+System.getProperty("file.separator")+screenName+".jpg"));
        } catch (FileNotFoundException e) {
            e.printStackTrace();
        } catch (IOException e) {
            e.printStackTrace();
        }
        finally
        {
            screenAdress=screenName+".png";
        }
        return screenAdress;
    }

    @SuppressWarnings({ "rawtypes", "unchecked" })
    public List<String> getquery() throws IOException{
        File file = new File("files\\files.txt");
        String encoding="GBK";
        List list=new ArrayList();
        if (file.isFile()&& file.exists()){
            InputStreamReader read = new InputStreamReader(new FileInputStream(file),encoding);
            BufferedReader bufferedReader = new BufferedReader(read);
            String lineTxt = null;
            while((lineTxt = bufferedReader.readLine()) != null){
                list.add(lineTxt);
            }
            read.close();

        }else{
            System.out.println("query文件不存在");
        }
        return list;

    }


    public static void main(String[] args) throws IOException {
        String ua = "Mozilla/5.0 (Windows; U; Windows NT 5.2) AppleWebKit/525.13 (KHTML, like Gecko) Chrome/0.2.149.27 Safari/525.13 ";
        Yaogeili yl = new Yaogeili("src/zhaohui/phantomjs.exe", ua);
//        yl.gethtml("21金维他");
        List<String> list = yl.getquery();
        for(int i=0;i<list.size();i++){
            yl.gethtml(list.get(i));
        }
        yl.driver.quit();


    }

}