iPython Django extensions

ipython_magic_function_inspector.py

"""
An IPython magic function to pretty-print objects with syntax highlighting.

Updated to also pretty print the object's __dict__ if it's available.

See, "Defining your own magics":
http://ipython.org/ipython-doc/stable/interactive/reference.html#defining-your-own-magics
For more on Pygments:
http://pygments.org/docs/quickstart/
Usage
-----
Place this file in your IPython startup directory. The default location is::
    ~/.ipython/profile_default/startup/
NOTE for Django: Since django uses an embedded IPython shell, it may not
load your default IPython profile. You'll need to run:
    %run ~/.ipython/profile_default/startup/ipython_magic_function_inspector.py
    
License
-------
Original copyright (c) 2014, Brad Montgomery <brad@bradmontgomery.net>
Updated copyright (c) 2016, Brian Bugh
Released under the MIT License.
http://opensource.org/licenses/MIT
"""
from __future__ import print_function

from IPython.core.magic import Magics, magics_class, line_magic
from pprint import pformat
from pygments import highlight
from pygments.formatters import Terminal256Formatter  # Or TerminalFormatter
from pygments.lexers import PythonLexer


@magics_class
class PrettyColorfulInspector(Magics):
    """Implementation for a magic function that inpects a given python object,
    and then prints a syntax-highlighted and pretty-printed version of it."""

    @line_magic
    def i(self, line):
        self.inspect(line)

    @line_magic
    def inspect(self, line):
        if line:
            # Use Pygments to do syntax highlighting
            lexer = PythonLexer()
            formatter = Terminal256Formatter()

            # evaluate the line to get a python object
            python_object = self.shell.ev(line)

            # Pretty Print/Format the object
            formatted_object = pformat(python_object)

            # Print the output, but don't return anything (othewise, we'd
            # potentially get a wall of color-coded text.
            print(highlight(formatted_object, lexer, formatter).strip())

            try: 
                formatted_dict = pformat(python_object.__dict__)
                print(highlight(formatted_dict, lexer, formatter).strip())
            except:
                pass


# Register with IPython
ip = get_ipython()
ip.register_magics(PrettyColorfulInspector)

load_models.py

from django.db.models.loading import get_models, get_apps
import importlib

loaded_models = get_models()
for app_mod in get_apps():
    app_models = get_models(app_mod)
    if not app_models:
        continue
    module = importlib.import_module(app_mod.__name__)
    for model in app_models:
        try:
            klass = getattr(module, model.__name__)
            globals()[model.__name__] = klass
        except AttributeError, e:
            print "Failed to import '%s' from '%s' reason: %s" % (model.__name__, app_mod.__name__.split('.')[-2], str(e))
            continue