Automatic screenshots when test fail by Selenium Webdriver in Python

Question

I want to automatic capturing screenshots if my webdriver tests failed (any exception or assertion error). I am using a Python unit test and Selenium Webdriver. Does anyone have any solution to this problem?

Answer 1

Use the following code using Firefox WebDriver which saves a screenshot on any exception to a dated image file:

 from datetime import datetime

from selenium import webdriver

browser = webdriver.Firefox()

try:

    # do some webdriver stuff here

except Exception as e:

    print e

    now = datetime.now().strftime('%Y-%m-%d_%H-%M-%S')

    browser.get_screenshot_as_file('screenshot-%s.png' % now)