diff -r ebaad720f88b -r 526ebd3988b0 web/lib/django_extensions/management/commands/graph_models.py --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/web/lib/django_extensions/management/commands/graph_models.py Wed Jan 20 12:37:40 2010 +0100 @@ -0,0 +1,60 @@ +from django.core.management.base import BaseCommand, CommandError +from optparse import make_option +from django_extensions.management.modelviz import generate_dot + +class Command(BaseCommand): + option_list = BaseCommand.option_list + ( + make_option('--disable-fields', '-d', action='store_true', dest='disable_fields', + help='Do not show the class member fields'), + make_option('--group-models', '-g', action='store_true', dest='group_models', + help='Group models together respective to there application'), + make_option('--all-applications', '-a', action='store_true', dest='all_applications', + help='Automaticly include all applications from INSTALLED_APPS'), + make_option('--output', '-o', action='store', dest='outputfile', + help='Render output file. Type of output dependend on file extensions. Use png or jpg to render graph to image.'), + make_option('--layout', '-l', action='store', dest='layout', default='dot', + help='Layout to be used by GraphViz for visualization. Layouts: circo dot fdp neato nop nop1 nop2 twopi'), + ) + + help = ("Creates a GraphViz dot file for the specified app names. You can pass multiple app names and they will all be combined into a single model. Output is usually directed to a dot file.") + args = "[appname]" + label = 'application name' + + requires_model_validation = True + can_import_settings = True + + def handle(self, *args, **options): + if len(args) < 1 and not options['all_applications']: + raise CommandError("need one or more arguments for appname") + + dotdata = generate_dot(args, **options) + if options['outputfile']: + self.render_output(dotdata, **options) + else: + self.print_output(dotdata) + + def print_output(self, dotdata): + print dotdata + + def render_output(self, dotdata, **kwargs): + try: + import pygraphviz + except ImportError, e: + raise CommandError("need pygraphviz python module ( apt-get install python-pygraphviz )") + + vizdata = ' '.join(dotdata.split("\n")).strip() + version = pygraphviz.__version__.rstrip("-svn") + try: + if [int(v) for v in version.split('.')]<(0,36): + # HACK around old/broken AGraph before version 0.36 (ubuntu ships with this old version) + import tempfile + tmpfile = tempfile.NamedTemporaryFile() + tmpfile.write(vizdata) + tmpfile.seek(0) + vizdata = tmpfile.name + except ValueError: + pass + + graph = pygraphviz.AGraph(vizdata) + graph.layout(prog=kwargs['layout']) + graph.draw(kwargs['outputfile'])