2016-09-21 10 views
1

ウェブページにあるすべての要素のスクリーンショットをキャプチャしようとしていますが、これを以下のコードを書いたディスクに保存します。ページ上にあるすべての要素のスクリーンショットをキャプチャします

唯一の問題は、このコードは最初の繰り返しでのみ動作し、その後は予期しないことが起こっていることです。

List<WebElement> eleId = driver.findElements(By.xpath("//*[@id]")); //fetch all the elements with ID attribute 
    System.out.println(eleId.size()); 

    for (int i = 0;i < eleId.size();i++) { 

//   Get entire page screenshot 
      File screenshot = ((TakesScreenshot) driver).getScreenshotAs(OutputType.FILE); 
      BufferedImage fullImg = ImageIO.read(screenshot); 

//   Get the location of element on the page 
      Point point = eleId.get(i).getLocation(); 

//   Get width and height of the element 
      int eleWidth = eleId.get(i).getSize().getWidth(); 
      int eleHeight = eleId.get(i).getSize().getHeight(); 

//   Crop the entire page screenshot to get only element screenshot 
      BufferedImage eleScreenshot = fullImg.getSubimage(point.getX(), point.getY(), eleWidth, eleHeight); 
      ImageIO.write(eleScreenshot, "png", screenshot); 

//   Creating variables name for image to be stores in the disk 
      String fileName = eleId.get(i).getAttribute("id"); 
      String imageLocation = "D:\\" + fileName + ".png"; 
//   System.out.println(imageLocation); 

//   Copy the element screenshot to disk 
      File screenshotLocation = new File(imageLocation); 
      FileUtils.copyFile(screenshot, screenshotLocation); 

      System.out.println("Screenshot has been stored."); 
} 
+0

予期せぬものはどういう意味ですか?スクリプトでエラーがスローされますか? –

+0

エラーは発生せず、1回の繰り返しでのみ実行されます。 –

答えて

1

こんにちはSandeepは以下のコードを試してください。それは私のために働いています。画像の高さと幅を確認するための条件を1つ追加しました。

@Test(enabled=true) 
public void getIndividualElementScreenShot() throws IOException{ 
    WebDriver driver = new FirefoxDriver(); 
    driver.get("http://www.google.com/"); 
    driver.manage().timeouts().pageLoadTimeout(30, TimeUnit.SECONDS); 
    driver.manage().window().maximize(); 
    List<WebElement> eles = driver.findElements(By.xpath("//*[@id]")); 
    System.out.println(eles.size()); 
    for(WebElement ele : eles){ 
     //Get Entire page screen shot 
     File screenShot = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE); 
     BufferedImage fullImage = ImageIO.read(screenShot); 

     //Get the location on the page 
     Point point = ele.getLocation(); 

     //Get width and height of an element 
     int eleWidth = ele.getSize().getWidth(); 
     int eleHeight = ele.getSize().getHeight(); 

     //Cropping the entire page screen shot to have only element screen shot 
     if(eleWidth != 0 && eleHeight != 0){ 
      BufferedImage eleScreenShot = fullImage.getSubimage(point.getX(), point.getY(), eleWidth, eleHeight); 
      ImageIO.write(eleScreenShot, "png", screenShot); 


      //Creating variable name for image to be store in disk 
      String fileName = ele.getAttribute("id"); 
      String imageLocation = "F:\\ElementImage\\"+fileName+".png"; 
      System.out.println(imageLocation); 

      //Copy the element screenshot to disk 
      File screenShotLocation = new File(imageLocation); 
      org.apache.commons.io.FileUtils.copyFile(screenShot, screenShotLocation); 

      System.out.println("Screen shot has beed stored"); 

     } 
    } 
    driver.close(); 
    driver.quit(); 
} 
+0

Ty @Sandipan Pramanik、問題は、私がアクセスしようとしていた要素の幅と高さであったと思う... tysm :) –

0

コードはうまくいきます。しかし、各反復でファイルを上書きしています。わかりましたか? fileName変数に新しいファイル名を割り当てることによって、各反復でファイル名を変更します。以下のコードを使用してください:

 List<WebElement> eleId = driver.findElements(By.xpath("//*[@id]")); //fetch all the elements with ID attribute 
     System.out.println(eleId.size()); 

     for (int i = 0;i < eleId.size();i++) { 

//   Get entire page screenshot 
       File screenshot = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE); 
       BufferedImage fullImg = ImageIO.read(screenshot); 

//   Get the location of element on the page 
       Point point = eleId.get(i).getLocation(); 

//   Get width and height of the element 
       int eleWidth = eleId.get(i).getSize().getWidth(); 
       int eleHeight = eleId.get(i).getSize().getHeight(); 

//   Crop the entire page screenshot to get only element screenshot 
       BufferedImage eleScreenshot = fullImg.getSubimage(point.getX(), point.getY(), eleWidth, eleHeight); 
       ImageIO.write(eleScreenshot, "png", screenshot); 

//   Creating variables name for image to be stores in the disk 
       String fileName = eleId.get(i).getAttribute("id"); 
       String imageLocation = "D:/" + fileName + i + ".png"; 
//   System.out.println(imageLocation); 

//   Copy the element screenshot to disk 
       File screenshotLocation = new File(imageLocation); 
       FileUtils.copyFile(screenshot, screenshotLocation); 

       System.out.println("Screenshot has been stored."); 
    } 
+0

私もこれを試しましたが、同じ出力が残っていますが、最初の反復後に停止しています。 –

+0

編集した回答のコード部分をコピー&ペーストします。私はこれが助けになると思います。 :) –

+0

コピーの貼り付けも試していますが、反復は1回だけです。 –

関連する問題