Webdriver Screenshot

Question

When taking a screenshot using Selenium Webdriver on windows with python, the screenshot is saved directly to the path of the program, is there a way to save the .png file to a specific directory?

unutbu · Answer 1 · 2014-11-04T00:43:25.520

88

Use driver.save_screenshot('/path/to/file') or driver.get_screenshot_as_file('/path/to/file'):

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.get_screenshot_as_file('/tmp/google.png') 
    # driver.save_screenshot('/tmp/google.png')

edited Nov 04 '14 at 00:43

answered Jan 17 '12 at 19:03

unutbu

842,883
184
1,785
1,677

1

Hi, `driver.save_screenshot('/path/to/file')` works on Windows, but `driver.get_screenshot_as_file('/path/to/file')` doesn't. (Yes, I changed for "\\"). But it helped, thank you. Have you got any idea guys how to check google's ReCaptcha with selenium? You will not be able to select any elements, even if HTML generates `
` or whatever... JS Scripts does not work too. **For clarification**, I do not mean solving images in reCaptcha, but only checking a checkbox "Im not a robot".
– Tommy L Apr 20 '18 at 16:10
1

@TommyL: [`save_screenshot` calls `get_screenshot_as_file`](https://github.com/SeleniumHQ/selenium/pull/3804/files), so if one worked, so should the other. – unutbu Apr 20 '18 at 20:30
1

@TommyL: Regarding recaptcha -- try googling something like "selenium click google recaptcha". There are a number of potential leads, such as [this one](https://stackoverflow.com/q/32446151/190597). If that doesn't work for you, you could consider posting a new question -- including your code so we understand what you've tried and what went wrong. – unutbu Apr 20 '18 at 20:36
One more thing that helped me immensely, if you need to change the image dimension, simply set the window size before you take the snapshot using `driver.set_window_size(1366, 728)` – srdg Jun 05 '19 at 18:41

score 35 · Answer 2 · edited May 23 '17 at 12:10

35

Inspired from this thread (same question for Java): Take a screenshot with Selenium WebDriver

from selenium import webdriver

browser = webdriver.Firefox()
browser.get('http://www.google.com/')
browser.save_screenshot('screenie.png')
browser.quit()

edited May 23 '17 at 12:10

Community

1
1

answered Dec 16 '13 at 12:23

Ran Adler

3,587
30
27

score 10 · Answer 3 · edited Dec 13 '12 at 08:49

10

Yes, we have a way to get screenshot extension of .png using python webdriver

use below code if you working in python webriver.it is very simple.

driver.save_screenshot('D\folder\filename.png')

edited Dec 13 '12 at 08:49

Sirko

72,589
19
149
183

answered Dec 13 '12 at 08:31

Kv.senthilkumar

936
2
16
29

score 7 · Answer 4 · answered May 08 '15 at 07:44

7

driver.save_screenshot("path to save \\screen.jpeg")

answered May 08 '15 at 07:44

Shaik

400
3
9

score 7 · Answer 5 · edited Nov 26 '18 at 04:53

7

This will take screenshot and place it in a directory of a chosen name.

import os
driver.save_screenshot(os.path.join(os.path.dirname(os.path.realpath(__file__)), 'NameOfScreenShotDirectory', 'PutFileNameHere'))

edited Nov 26 '18 at 04:53

hellpanderr

5,581
3
33
43

answered Jun 27 '18 at 14:35

Ger Mc

630
3
11
22

It's great to have a nicely formatted code for the answer but it's usually best practice to include a little explanation with it. – sniperd Jun 27 '18 at 15:15
2

"This will take screenshot and place it in a directory of a chosen name." is pretty clear for most people I'd imagine – Ger Mc Sep 27 '19 at 14:02
NameError: name '__file__' is not defined – youssef mhiri Feb 26 '21 at 23:09

score 5 · Answer 6 · edited May 23 '17 at 12:10

Sure it isn't actual right now but I faced this issue also and my way: Looks like 'save_screenshot' have some troubles with creating files with space in name same time as I added randomization to filenames for escaping override.

Here I got method to clean my filename of whitespaces (How do I replace whitespaces with underscore and vice versa?):

def urlify(self, s):
    # Remove all non-word characters (everything except numbers and letters)
    s = re.sub(r"[^\w\s]", '', s)
    # Replace all runs of whitespace with a single dash
    s = re.sub(r"\s+", '-', s)
    return s

then

driver.save_screenshot('c:\\pytest_screenshots\\%s' % screen_name)

where

def datetime_now(prefix):
    symbols = str(datetime.datetime.now())
    return prefix + "-" + "".join(symbols)

screen_name = self.urlify(datetime_now('screen')) + '.png'

score 5 · Answer 7 · answered Jun 07 '18 at 20:50

Here they asked a similar question, and the answer seems more complete, I leave the source:

How to take partial screenshot with Selenium WebDriver in python?

from selenium import webdriver
from PIL import Image
from io import BytesIO

fox = webdriver.Firefox()
fox.get('http://stackoverflow.com/')

# now that we have the preliminary stuff out of the way time to get that image :D
element = fox.find_element_by_id('hlogo') # find part of the page you want image of
location = element.location
size = element.size
png = fox.get_screenshot_as_png() # saves screenshot of entire page
fox.quit()

im = Image.open(BytesIO(png)) # uses PIL library to open image in memory

left = location['x']
top = location['y']
right = location['x'] + size['width']
bottom = location['y'] + size['height']


im = im.crop((left, top, right, bottom)) # defines crop points
im.save('screenshot.png') # saves new cropped image

Shubham Jain · Answer 8 · 2018-01-29T12:54:59.343

You can use below function for relative path as absolute path is not a good idea to add in script

Import

import sys, os

Use code as below :

ROOT_DIR = os.path.dirname(os.path.abspath(__file__))
screenshotpath = os.path.join(os.path.sep, ROOT_DIR,'Screenshots'+ os.sep)
driver.get_screenshot_as_file(screenshotpath+"testPngFunction.png")

make sure you create the folder where the .py file is present.

os.path.join also prevent you to run your script in cross-platform like: UNIX and windows. It will generate path separator as per OS at runtime. os.sep is similar like File.separtor in java

score 2 · Answer 9 · edited Jun 20 '19 at 12:33

2

TakeScreenShot screenshot=new TakeScreenShot();
screenshot.screenShot("screenshots//TestScreenshot//password.png");

it will work , please try.

edited Jun 20 '19 at 12:33

Md.Sukel Ali

2,987
5
22
34

answered Jun 20 '19 at 11:02

suriya kanagaraj

29
3

score 0 · Answer 10 · edited Jan 13 '21 at 06:18

Have a look on the below python script to take snap of FB homepage by using selenium package of Chrome web driver.

Script:

import selenium

from selenium import webdriver

import time

from time import sleep

chrome_browser = webdriver.Chrome()

chrome_browser.get('https://www.facebook.com/') # Enter to FB login page

sleep(5)

chrome_browser.save_screenshot('C:/Users/user/Desktop/demo.png') # To take FB homepage snap

chrome_browser.close() # To Close the driver connection

chrome_browser.quit() # To Close the browser

score 0 · Answer 11 · answered Mar 24 '22 at 19:51

It's quite simple, plz try this:

from selenium import webdriver

driver = webdriver.Chrome()

def take_screenshot(name):
    driver.get_screenshot_as_file(f"./screenshot/{name}.png")

#Or with datetime:
def take_screenshot_with_dt(name):       
    datetime.now()
    screenDatetime = datetime.now().strftime('%d-%m-%Y-%H-%M')
    driver.get_screenshot_as_file(f"./screenshot/{name}-{screenDatetime}.png")



take_screenshot("screenshotName")
take_screenshot_with_dt("screenshotNameDt")

score 0 · Answer 12 · answered May 30 '22 at 06:51

write this hook in conftest.py

@pytest.hookimpl(hookwrapper=True)
def pytest_runtest_makereport(item):
    pytest_html = item.config.pluginmanager.getplugin("html")
    outcome = yield
    report = outcome.get_result()
    extra = getattr(report, "extra", [])
    if report.when == "call":
        # always add url to report
        extra.append(pytest_html.extras.url(driver.current_url))
        xfail = hasattr(report, "wasxfail")
        if (report.skipped and xfail) or (report.failed and not xfail):
            report_directory = os.path.dirname(item.config.option.htmlpath)
            file_name = report.nodeid.replace("::", "_") + ".png"
            destination_file = os.path.join(report_directory, file_name)
            driver.save_screenshot(destination_file)
            if file_name:
                html = '<div><img src="%s" alt="screenshot" style="width:300px;height=200px"'\
                    'onclick="window.open(this.src)" align="right"/></div>'%file_name
            # only add additional html on failure
                extra.append(pytest_html.extras.html(html))
        report.extra = extra

score -2 · Answer 13 · edited Sep 11 '19 at 13:51

-2

WebDriver driver = new FirefoxDriver();
driver.get("http://www.google.com/");
File scrFile = ((TakesScreenshot)driver).getScreenshotAs(OutputType.FILE);
FileUtils.copyFile(scrFile, new File("c:\\NewFolder\\screenshot1.jpg"));

edited Sep 11 '19 at 13:51

slfan

8,950
115
65
78

answered Sep 11 '19 at 13:34

Dharit Mehta

25
3

3

Please provide some explanations to go along with your code. – francoisr Sep 11 '19 at 17:05

score -10 · Answer 14 · answered Oct 12 '12 at 06:57

I understand you are looking for an answer in python, but here is how one would do it in ruby..

http://watirwebdriver.com/screenshots/

If that only works by saving in current directory only.. I would first assign the image to a variable and then save that variable to disk as a PNG file.

eg:

 image = b.screenshot.png

 File.open("testfile.png", "w") do |file|
  file.puts "#{image}"
 end

where b is the browser variable used by webdriver. i have the flexibility to provide an absolute or relative path in "File.open" so I can save the image anywhere.

Webdriver Screenshot

14 Answers14

Linked