Auto Screenshot Selenium Test Failures (In Django)

class SeleniumTestCase(StaticLiveServerTestCase):  # StaticLiveServerTestCase from django.contrib.staticfiles
    live_server_url = 'http://localhost:8081'

    @classmethod
    def setUpClass(cls):
        '''
        One example of creating the test browser.  You could do it per-function as well, in setUp() instead.
        We'll be using cls.browser or self.browser later to get at the screenshot-saving mechanism.
        I'm omitting the corresponding tearDownClass or tearDown, which needs to close the browser.
        '''
        cls.browser = Browser()

    def run(self, result=None):
        '''
        Run is called on every test function, and an aggregate TestResult object
        is maintained across every test function.
        '''
        self.currentResult = result # remember result for use in tearDown
        super(SeleniumTestCase, self).run(result)

    def tearDown(self):
        '''
        tearDown is called after every test function.
        unittest.TestCase.run() handles errors and failures by appending each
        to the aggregate TestResult lists (TestResult.errors and TestResult.failures)
        so after a test just failed or errored, it'll be the last one in the list.
        '''
        result = self.currentResult  # currentResult contains running results for every test in a TestCase
        if result.errors:
            last_error = result.errors[-1]
            last_error_case, _ = last_error  # extract test method from tuple
            if last_error_case.id() == self.id():
                self._save_screenshot(last_error)
        elif result.failures:
            last_fail = result.failures[-1]
            last_fail_case, _ = last_fail  # extract test method from tuple
            if last_fail_case.id() == self.id():
                self._save_screenshot(last_fail)

        return super(SeleniumTestCase, self).tearDown()

    def _save_screenshot(self, error=None):
        '''
        At this point, we can use the browser object to save a screenshot.
        We can also do this directly from test functions.
        Screenshots will be saved with the python path to the test method
        as a filename.
        '''
        path = settings.SELENIUM_SCREENSHOTS_PATH
        import os
        if not os.path.exists(path):
            os.makedirs(path)
        filename = "%s.jpg" % self.id()
        full_path = os.path.join(path, filename)
        self.browser.save_screenshot(full_path)