Taking Screenshot using Selenium WebDriver

Question

While taking the screenshot using Selenium Webdriver on windows browser and using python programming language, the screenshot is saved directly to the path of the program, is there a way to save the .png file to a specific directory, if yes then kindly tell me the solution?

Answer 1

Using driver.save_screenshot('/path/to/file') we will be able to do it as shown below:

import selenium.webdriver as webdriver
import contextlib

@contextlib.contextmanager
def quitting(thing):
    yield thing
    thing.quit()

with quitting(webdriver.Firefox()) as driver:
    driver.implicitly_wait(10)
    driver.get('http://www.google.com')
    driver.save_screenshot('/tmp/google.png')

---------------