[Epydoc-commits] SF.net SVN: epydoc: [1222] trunk/epydoc/src/epydoc/docwriter/dotgraph.py

[<ed...@us...>]

Revision: 1222
Author:   edloper
Date:     2006-04-12 12:50:15 -0700 (Wed, 12 Apr 2006)
ViewCVS:  http://svn.sourceforge.net/epydoc/?rev=1222&view=rev

Log Message:
-----------
- Fixed base tree list in the case where a base is not a ClassDoc

Modified Paths:
--------------
    trunk/epydoc/src/epydoc/docwriter/dotgraph.py
Modified: trunk/epydoc/src/epydoc/docwriter/dotgraph.py
===================================================================
--- trunk/epydoc/src/epydoc/docwriter/dotgraph.py	2006-04-12 19:49:51 UTC (rev 1221)
+++ trunk/epydoc/src/epydoc/docwriter/dotgraph.py	2006-04-12 19:50:15 UTC (rev 1222)
@@ -1002,14 +1002,16 @@
     classes = set(bases)
     queue = list(bases)
     for cls in queue:
-        if cls.subclasses not in (None, UNKNOWN):
-            queue.extend(cls.subclasses)
-            classes.update(cls.subclasses)
+        if isinstance(cls, ClassDoc):
+            if cls.subclasses not in (None, UNKNOWN):
+                queue.extend(cls.subclasses)
+                classes.update(cls.subclasses)
     queue = list(bases)
     for cls in queue:
-        if cls.bases not in (None, UNKNOWN):
-            queue.extend(cls.bases)
-            classes.update(cls.bases)
+        if isinstance(cls, ClassDoc):
+            if cls.bases not in (None, UNKNOWN):
+                queue.extend(cls.bases)
+                classes.update(cls.bases)
 
     # Add a node for each cls.
     classes = [d for d in classes if isinstance(d, ClassDoc)
@@ -1052,7 +1054,8 @@
     # Create nodes for all class_doc's subclasses.
     queue = [class_doc]
     for cls in queue:
-        if cls.subclasses not in (None, UNKNOWN):
+        if (isinstance(cls, ClassDoc) and
+            cls.subclasses not in (None, UNKNOWN)):
             queue.extend(cls.subclasses)
             for cls in cls.subclasses:
                 if cls not in nodes:


This was sent by the SourceForge.net collaborative development platform, the world's largest Open Source development site.