How can I take a screenshot with Selenium WebDriver?

Question

Is it possible to take a screenshot using Selenium WebDriver?

(Note: Not Selenium Remote Control)

Answer 1

There are multiple methods through Selenium's Java and Python client to take a screenshot using Selenium WebDriver.

---

Java Methods

The following are the different Java methods to take a screenshot:

• Using getScreenshotAs() from the TakesScreenshot interface:

• Code block:

       package screenShot;
  
       import java.io.File;
       import java.io.IOException;
  
       import org.apache.commons.io.FileUtils;
       import org.openqa.selenium.OutputType;
       import org.openqa.selenium.TakesScreenshot;
       import org.openqa.selenium.WebDriver;
       import org.openqa.selenium.firefox.FirefoxDriver;
       import org.openqa.selenium.support.ui.ExpectedConditions;
       import org.openqa.selenium.support.ui.WebDriverWait;
  
       public class Firefox_takesScreenshot {
  
           public static void main(String[] args) throws IOException {
  
               System.setProperty("webdriver.gecko.driver", "C:\\Utility\\BrowserDrivers\\geckodriver.exe");
               WebDriver driver =  new FirefoxDriver();
               driver.get("https://login.bws.birst.com/login.html/");
               new WebDriverWait(driver, 20).until(ExpectedConditions.titleContains("Birst"));
               File scrFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
               FileUtils.copyFile(scrFile, new File(".\\Screenshots\\Mads_Cruz_screenshot.png"));
               driver.quit();
           }
       }
  

• Screenshot:

  

• If the webpage is jQuery enabled, you can use ashot from the pazone/ashot library:

• Code block:

       package screenShot;
  
       import java.io.File;
       import javax.imageio.ImageIO;
       import org.openqa.selenium.WebDriver;
       import org.openqa.selenium.firefox.FirefoxDriver;
       import org.openqa.selenium.support.ui.ExpectedConditions;
       import org.openqa.selenium.support.ui.WebDriverWait;
  
       import ru.yandex.qatools.ashot.AShot;
       import ru.yandex.qatools.ashot.Screenshot;
       import ru.yandex.qatools.ashot.shooting.ShootingStrategies;
  
       public class ashot_CompletePage_Firefox {
  
           public static void main(String[] args) throws Exception {
  
               System.setProperty("webdriver.gecko.driver", "C:\\Utility\\BrowserDrivers\\geckodriver.exe");
               WebDriver driver =  new FirefoxDriver();
               driver.get("https://jquery.com/");
               new WebDriverWait(driver, 20).until(ExpectedConditions.titleContains("jQuery"));
               Screenshot myScreenshot = new AShot().shootingStrategy(ShootingStrategies.viewportPasting(100)).takeScreenshot(driver);
               ImageIO.write(myScreenshot.getImage(),"PNG",new File("./Screenshots/firefoxScreenshot.png"));
               driver.quit();
           }
       }
  

• Screenshot:

  

• Using selenium-shutterbug from assertthat/selenium-shutterbug library:

• Code block:

       package screenShot;
  
       import org.openqa.selenium.WebDriver;
       import org.openqa.selenium.firefox.FirefoxDriver;
       import com.assertthat.selenium_shutterbug.core.Shutterbug;
       import com.assertthat.selenium_shutterbug.utils.web.ScrollStrategy;
  
       public class selenium_shutterbug_fullpage_firefox {
  
           public static void main(String[] args) {
  
               System.setProperty("webdriver.gecko.driver", "C:\\Utility\\BrowserDrivers\\geckodriver.exe");
               WebDriver driver =  new FirefoxDriver();
               driver.get("https://www.google.co.in");
               Shutterbug.shootPage(driver, ScrollStrategy.BOTH_DIRECTIONS).save("./Screenshots/");
               driver.quit();
           }
       }
  

• Screenshot:

  

---

Python Methods

The following are the different Python methods to take a screenshot:

• Using save_screenshot() method:

• Code block:

       from selenium import webdriver
  
       driver = webdriver.Chrome(r'C:\Utility\BrowserDrivers\chromedriver.exe')
       driver.get("http://google.com")
       driver.save_screenshot('./Screenshots/save_screenshot_method.png')
       driver.quit()
  

• Screenshot:

  

• Using the get_screenshot_as_file() method:

• Code block:

       from selenium import webdriver
  
       driver = webdriver.Chrome(r'C:\Utility\BrowserDrivers\chromedriver.exe')
       driver.get("http://google.com")
       driver.get_screenshot_as_file('./Screenshots/get_screenshot_as_file_method.png')
       driver.quit()
  

• Screenshot:

  

• Using get_screenshot_as_png() method:

• Code block:

       from selenium import webdriver
  
       driver = webdriver.Chrome(r'C:\Utility\BrowserDrivers\chromedriver.exe')
       driver.get("http://google.com")
       screenPnG = driver.get_screenshot_as_png()
  
       # Crop it back to the window size (it may be taller)
       box = (0, 0, 1366, 728)
       im = Image.open(BytesIO(screenPnG))
       region = im.crop(box)
       region.save('./Screenshots/get_screenshot_as_png_method.png', 'PNG', optimize=True, quality=95)
       driver.quit()
  

• Screenshot: