Changes between Version 3 and Version 4 of DjangoGraphviz

Timestamp:

Aug 5, 2006, 4:42:40 AM (18 years ago)

Author:

Antonio Cavedoni <antonio@…>

Comment:

--

DjangoGraphviz

@@ -3 +3 @@
 Inspired by [http://www.hackdiary.com/archives/000093.html this nice hack by Matt Biddulph] I set out to create a similar thing for Django models. Below is an initial implementation, and here are some results generated with the OS X version of Graphviz:
 
- * [http://cavedoni.com/2006/08/camera.dot generated dot file]: resulting image in [http://cavedoni.com/2006/08/camera.pdf PDF]
- * [http://cavedoni.com/2006/08/mincer.dot generated dot file]: resulting image in [http://cavedoni.com/2006/08/mincer.pdf PDF] 
+ * [http://cavedoni.com/2006/08/camera-001.dot generated dot file]: resulting image in [http://cavedoni.com/2006/08/camera-001.pdf PDF]
+ * [http://cavedoni.com/2006/08/mincer-001.dot generated dot file]: resulting image in [http://cavedoni.com/2006/08/mincer-001.pdf PDF] 
 
 I hope people might find it useful.
@@ -10 +10 @@
 {{{
 #!python
+#!/usr/bin/python
 """Django model to DOT (Graphviz) converter
 by Antonio Cavedoni <antonio@cavedoni.org>
@@ -16 +17 @@
 the script like this:
 
-  $ python modelviz.py <appname>.<models module_name> > <filename>.dot
+  $ python modelviz.py <app_label> > <filename>.dot
 
 Example:
 
-  $ python modelviz.py camera.models > foo.dot
+  $ python modelviz.py camera > foo.dot
 """
 
-from django.db.models.base import ModelBase
+from django.db import models
+from django.db.models import get_models
 from django.db.models.fields.related import \
     ForeignKey, OneToOneField, ManyToManyField
 
-def generate_dot(model_module):
-   print "digraph model {"
+def generate_dot(app_label):
+   app = models.get_app(app_label)
 
-   for obj in dir(model_module):
-      o = getattr(model_module, obj)
-      if isinstance(o, ModelBase):
-         for field in o._meta.fields:
-            if type(field) in [ForeignKey, OneToOneField, ManyToManyField]:
-               print '  %s -> %s_%s [label="%s"];' % \
-                   (o.__name__, o.__name__, field.name, type(field).__name__)
-               print "  %s_%s -> %s" % \
-                   (o.__name__, field.name, field.rel.to.__name__)
-            else:
-               print "  %s -> %s_%s;" % (o.__name__, o.__name__, field.name)
+   print "digraph %s {" % ('"' + app.__name__ + '"')
+   print """  fontname = "Helvetica"
+  fontsize = 8
+      
+  node [
+    fontname = "Helvetica"
+    fontsize = 8
+    shape = "record"
+  ]
+   edge [
+    fontname = "Helvetica"
+    fontsize = 8
+  ]
+"""
+   for o in get_models(app):
+      nodes = []
+      nodes.append("""  %s [
+    label = "{%s | """ % (o.__name__, o.__name__))
+
+      # members
+      for field in o._meta.fields:
+         nodes.append("%s : %s\\l" % (field.name, type(field).__name__))
+         
+      for field in o._meta.many_to_many:
+         nodes.append("%s : %s\\l" % (field.name, type(field).__name__))
+
+      nodes.append(""" }"\n]\n""")
+      print "".join(nodes)
+
+      # relations
+      rel = []
+      for field in o._meta.fields:
+         if isinstance(field, ForeignKey):
+            _rel = '  %s -> %s [label="%s"];' % \
+                (o.__name__, field.rel.to.__name__, type(field).__name__)
+            if _rel not in rel:
+               rel.append(_rel)
+
+      for field in o._meta.many_to_many:
+         if isinstance(field, ManyToManyField):
+            _rel = '  %s -> %s [label="%s"] [arrowhead=normal arrowtail=normal];' % \
+                (o.__name__, field.rel.to.__name__, type(field).__name__)
+            if _rel not in rel:
+               rel.append(_rel)
+
+      print "\n".join(rel)
 
    print "}"
@@ -46 +83 @@
 if __name__ == "__main__":
    import sys
-   _models = __import__(sys.argv[1])
-   generate_dot(_models.models)
+   app_label = sys.argv[1]
+   generate_dot(app_label)
 }}}