Changes between Version 7 and Version 8 of DjangoGraphviz

Timestamp:

Aug 6, 2006, 6:50:36 AM (18 years ago)

Author:

Antonio Cavedoni <antonio@…>

Comment:

Added model viz for the Zyons project, edited code to support GenericRelations and OneToOneFields

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-001.dot generated dot file]: resulting image in [http://cavedoni.com/2006/08/camera-001.pdf PDF], 
+ * Graphviz DOT files for the Zyons project: [http://cavedoni.com/2006/08/zyons-comment.dot comment] [http://cavedoni.com/2006/08/zyons-event.dot event] [http://cavedoni.com/2006/08/zyons-forum.dot forum] [http://cavedoni.com/2006/08/zyons-openid.dot openid] [http://cavedoni.com/2006/08/zyons-prefs.dot prefs] [http://cavedoni.com/2006/08/zyons-profile.dot profile] [http://cavedoni.com/2006/08/zyons-sessions.dot sessions] [http://cavedoni.com/2006/08/zyons-tag.dot tag]
+ * Resulting graphs in PDF: [http://cavedoni.com/2006/08/zyons-comment.pdf comment] [http://cavedoni.com/2006/08/zyons-event.pdf event] [http://cavedoni.com/2006/08/zyons-forum.pdf forum] [http://cavedoni.com/2006/08/zyons-openid.pdf openid] [http://cavedoni.com/2006/08/zyons-prefs.pdf prefs] [http://cavedoni.com/2006/08/zyons-profile.pdf profile] [http://cavedoni.com/2006/08/zyons-sessions.pdf sessions] [http://cavedoni.com/2006/08/zyons-tag.pdf tag]
+
+ * Other samples:
+   * [http://cavedoni.com/2006/08/camera-001.dot generated dot file]: resulting image in [http://cavedoni.com/2006/08/camera-001.pdf PDF], 
 {{{
 #!html
 <a href="http://code.djangoproject.com/attachment/wiki/DjangoGraphviz/camera-001.png">PNG</a>
 }}}
- * [http://cavedoni.com/2006/08/mincer-001.dot generated dot file]: resulting image in [http://cavedoni.com/2006/08/mincer-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], 
 {{{
 #!html
@@ -36 +40 @@
 from django.db.models.fields.related import \
     ForeignKey, OneToOneField, ManyToManyField
+from django.db.models.fields.generic import GenericRelation
 
 def generate_dot(app_label):
@@ -57 +62 @@
       nodes = []
       nodes.append("""  %s [
-    label = "{%s | """ % (o.__name__, o.__name__))
+    label = "{%s |""" % (o.__name__, o.__name__))
 
       # members
       for field in o._meta.fields:
-         nodes.append("%s : %s\\l" % (field.name, type(field).__name__))
+         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__))
+      if o._meta.many_to_many:
+         for field in o._meta.many_to_many:
+            nodes.append('%s : %s\\l' % (field.name, type(field).__name__))
+
+      if o._meta.one_to_one_field:
+         for field in o._meta.one_to_one_field:
+            nodes.append('%s : %s\\l' % (field.name, type(field).__name__))
 
       nodes.append(""" }"\n]\n""")
@@ -73 +83 @@
       for field in o._meta.fields:
          if isinstance(field, ForeignKey):
-            _rel = '  %s -> %s [label="%s"];' % \
+            _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)
+      if o._meta.many_to_many:
+         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)
+
+            if isinstance(field, GenericRelation):
+               _rel = '  %s -> %s [label="%s"] [style="dotted"] [arrowhead=normal arrowtail=normal];' % \
+                   (o.__name__, field.rel.to.__name__, type(field).__name__)
+               if _rel not in rel:
+                  rel.append(_rel)
+
+      if o._meta.one_to_one_field:
+         for field in o._meta.one_to_one_field:
+            if isinstance(field, OneToOneField):
+               _rel = '  %s -> %s [label="%s"] [arrowhead=none arrowtail=none];' % \
+                   (o.__name__, field.rel.to.__name__, type(field).__name__)
+               if _rel not in rel:
+                  rel.append(_rel)
 
       print "\n".join(rel)