diff -r ebaad720f88b -r 526ebd3988b0 web/lib/django_extensions/management/commands/shell_plus.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/web/lib/django_extensions/management/commands/shell_plus.py Wed Jan 20 12:37:40 2010 +0100 @@ -0,0 +1,77 @@ +import os +from django.core.management.base import NoArgsCommand +from optparse import make_option + +class Command(NoArgsCommand): + option_list = NoArgsCommand.option_list + ( + make_option('--plain', action='store_true', dest='plain', + help='Tells Django to use plain Python, not IPython.'), + make_option('--no-pythonrc', action='store_true', dest='no_pythonrc', + help='Tells Django to use plain Python, not IPython.'), + ) + help = "Like the 'shell' command but autoloads the models of all installed Django apps." + + requires_model_validation = True + + def handle_noargs(self, **options): + # XXX: (Temporary) workaround for ticket #1796: force early loading of all + # models from installed apps. (this is fixed by now, but leaving it here + # for people using 0.96 or older trunk (pre [5919]) versions. + from django.db.models.loading import get_models, get_apps + loaded_models = get_models() + + use_plain = options.get('plain', False) + use_pythonrc = not options.get('no_pythonrc', True) + + # Set up a dictionary to serve as the environment for the shell, so + # that tab completion works on objects that are imported at runtime. + # See ticket 5082. + from django.conf import settings + imported_objects = {'settings': settings} + for app_mod in get_apps(): + app_models = get_models(app_mod) + if not app_models: + continue + model_labels = ", ".join([model.__name__ for model in app_models]) + print self.style.SQL_COLTYPE("From '%s' autoload: %s" % (app_mod.__name__.split('.')[-2], model_labels)) + for model in app_models: + try: + imported_objects[model.__name__] = getattr(__import__(app_mod.__name__, {}, {}, model.__name__), model.__name__) + except AttributeError, e: + print self.style.ERROR_OUTPUT("Failed to import '%s' from '%s' reason: %s" % (model.__name__, app_mod.__name__.split('.')[-2], str(e))) + continue + try: + if use_plain: + # Don't bother loading IPython, because the user wants plain Python. + raise ImportError + import IPython + # Explicitly pass an empty list as arguments, because otherwise IPython + # would use sys.argv from this script. + shell = IPython.Shell.IPShell(argv=[], user_ns=imported_objects) + shell.mainloop() + except ImportError: + # Using normal Python shell + import code + try: # Try activating rlcompleter, because it's handy. + import readline + except ImportError: + pass + else: + # We don't have to wrap the following import in a 'try', because + # we already know 'readline' was imported successfully. + import rlcompleter + readline.set_completer(rlcompleter.Completer(imported_objects).complete) + readline.parse_and_bind("tab:complete") + + # We want to honor both $PYTHONSTARTUP and .pythonrc.py, so follow system + # conventions and get $PYTHONSTARTUP first then import user. + if use_pythonrc: + pythonrc = os.environ.get("PYTHONSTARTUP") + if pythonrc and os.path.isfile(pythonrc): + try: + execfile(pythonrc) + except NameError: + pass + # This will import .pythonrc.py as a side-effect + import user + code.interact(local=imported_objects)