--- /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'])