- EdgeCollection can now go away

- fix reflection test

1 files changed, 25 insertions, 31 deletions

diff --git a/lib/sqlalchemy/topological.py b/lib/sqlalchemy/topological.py
index 07b1ba03e..fbde7c601 100644
--- a/lib/sqlalchemy/topological.py
+++ b/lib/sqlalchemy/topological.py
@@ -10,46 +10,26 @@ from sqlalchemy.exc import CircularDependencyError
 from sqlalchemy import util
 
 
-__all__ = ['sort']
-
-class _EdgeCollection(object):
-    """A collection of directed edges."""
-
-    def __init__(self, edges):
-        self.parent_to_children = util.defaultdict(set)
-        self.child_to_parents = util.defaultdict(set)
-        for parentnode, childnode in edges:
-            self.parent_to_children[parentnode].add(childnode)
-            self.child_to_parents[childnode].add(parentnode)
-            
-    def outgoing(self, node):
-        return self.parent_to_children[node]
-    
-    def incoming(self, node):
-        return self.child_to_parents[node]
-        
-    def __iter__(self):
-        for parent in self.parent_to_children:
-            for child in self.outgoing(parent):
-                yield (parent, child)
-
-    def __repr__(self):
-        return repr(list(self))
+__all__ = ['sort', 'sort_as_subsets', 'find_cycles']
 
 def sort_as_subsets(tuples, allitems):
     output = set()
 
     todo = set(allitems)
-    edges = _EdgeCollection(tuples)
+
+    edges = util.defaultdict(set)
+    for parent, child in tuples:
+        edges[child].add(parent)
+    
     while todo:
         for node in list(todo):
-            if not todo.intersection(edges.incoming(node)):
+            if not todo.intersection(edges[node]):
                 output.add(node)
 
         if not output:
             raise CircularDependencyError(
-                    "Circular dependency detected: cycles: %r all edges: %r" % 
-                    (find_cycles(tuples, allitems), edges))
+                    "Circular dependency detected: cycles: %r all edges: %s" % 
+                    (find_cycles(tuples, allitems), _dump_edges(edges, True)))
 
         todo.difference_update(output)
         yield output
@@ -68,7 +48,10 @@ def sort(tuples, allitems):
 def find_cycles(tuples, allitems):
     # straight from gvr with some mods
     todo = set(allitems)
-    edges = _EdgeCollection(tuples)
+
+    edges = util.defaultdict(set)
+    for parent, child in tuples:
+        edges[parent].add(child)
 
     output = set()
     
@@ -77,7 +60,7 @@ def find_cycles(tuples, allitems):
         stack = [node]
         while stack:
             top = stack[-1]
-            for node in edges.outgoing(top):
+            for node in edges[top]:
                 if node in stack:
                     cyc = stack[stack.index(node):]
                     todo.difference_update(cyc)
@@ -91,3 +74,14 @@ def find_cycles(tuples, allitems):
                 node = stack.pop()
     return output
 
+def _dump_edges(edges, reverse):
+    l = []
+    for left in edges:
+        for right in edges[left]:
+            if reverse:
+                l.append((right, left))
+            else:
+                l.append((left, right))
+    return repr(l)
+
+