1 from django.core.management.base import BaseCommand

     2 from optparse import make_option

     3 import sys

     4 

     5 class Command(BaseCommand):

     6     option_list = BaseCommand.option_list + (

     7         make_option('--verbosity', action='store', dest='verbosity', default='1',

     8             type='choice', choices=['0', '1', '2'],

     9             help='Verbosity level; 0=minimal output, 1=normal output, 2=all output'),

    10         make_option('--noinput', action='store_false', dest='interactive', default=True,

    11             help='Tells Django to NOT prompt the user for input of any kind.'),

    12     )

    13     help = 'Runs the test suite for the specified applications, or the entire site if no apps are specified.'

    14     args = '[appname ...]'

    15 

    16     requires_model_validation = False

    17 

    18     def handle(self, *test_labels, **options):

    19         from django.conf import settings

    20         from django.db.models import get_app, get_apps

    21 

    22         verbosity = int(options.get('verbosity', 1))

    23         interactive = options.get('interactive', True)

    24     

    25         test_path = settings.TEST_RUNNER.split('.')

    26         # Allow for Python 2.5 relative paths

    27         if len(test_path) > 1:

    28             test_module_name = '.'.join(test_path[:-1])

    29         else:

    30             test_module_name = '.'

    31         test_module = __import__(test_module_name, {}, {}, test_path[-1])

    32         test_runner = getattr(test_module, test_path[-1])

    33 

    34         failures = test_runner(test_labels, verbosity=verbosity, interactive=interactive)

    35         if failures:

    36             sys.exit(failures)