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)
If you are interested to learn Selenium on a much deeper level and want to become a professional in the testing domain, check out Intellipaat’s Selenium certification!