XML report generation for Django unittests

bittentest.py

import os

from django.conf import settings

from south.management.commands import test

class Command(test.Command):
    help = test.Command.help.rstrip('.') + ' with special output for Bitten.'

    def handle(self, *test_labels, **options):
        os.environ['DJANGO_BITTEN_TEST'] = '1'
        return super(Command, self).handle(*test_labels, **options)

runner.py

class AtariTestSuiteRunner(DjangoTestSuiteRunner):

    def run_suite(self, suite, **kwargs):
        if os.environ.get('DJANGO_BITTEN_TEST') == '1':
            from bitten.util.testrunner import XMLTestRunner
            xml_output = settings.TEST_XML_OUTPUT
            output_dir = os.path.dirname(xml_output) or '.'
            if not os.path.exists(output_dir):
                os.makedirs(output_dir)
            xml_output_file = open(xml_output, 'w')
            return XMLTestRunner(xml_stream=xml_output_file).run(suite)
        else:
            return super(AtariTestSuiteRunner, self).run_suite(suite, **kwargs)