Ticket #7857: testshell.diff

django/core/management/commands/testshell.py

@@ +1 @@
+import os
+from django.core.management import call_command
+from django.core.management.base import BaseCommand
+from optparse import make_option
+
+class Command(BaseCommand):
+        option_list = BaseCommand.option_list + (
+                make_option('--plain', action='store_true', dest='plain',
+                        help='Tells Django to use plain Python, not IPython.'),
+        )
+        help = "Runs a Python interactive interpreter. Tries to use IPython, if it's available."
+        shells = ['ipython', 'bpython']
+        requires_model_validation = False
+
+        def handle(self, *args, **options):
+                print 'Using test database'
+                from django.db import connection
+                connection.creation.create_test_db()
+
+                use_plain = options.get('plain', False)
+
+                call_command('shell', plain=use_plain)