Better handling of failure screenshots for logs

Author: mdmintz

seleniumbase/core/log_helper.py

@@ -10,7 +10,10 @@
 def log_screenshot(test_logpath, driver):
     screenshot_name = settings.SCREENSHOT_NAME
     screenshot_path = "%s/%s" % (test_logpath, screenshot_name)
-    driver.get_screenshot_as_file(screenshot_path)
+    try:
+        driver.get_screenshot_as_file(screenshot_path)
+    except Exception:
+        print("WARNING: Unable to get screenshot for failure logs!")
 
 
 def log_test_failure_data(test, test_logpath, driver, browser):