cdef another class in ParseTreeTransforms

a41594b4 · Stefan Behnel · 138e5cbc · a41594b4 · a41594b4
Commit a41594b4 authored Nov 28, 2010 by Stefan Behnel
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 5 deletions

Cython/Compiler/ParseTreeTransforms.pxd Cython/Compiler/ParseTreeTransforms.pxd +3 -3

Cython/Compiler/ParseTreeTransforms.py Cython/Compiler/ParseTreeTransforms.py +3 -2

No files found.
--- a/Cython/Compiler/ParseTreeTransforms.pxd
+++ b/Cython/Compiler/ParseTreeTransforms.pxd

 from Cython.Compiler.Visitor cimport (
-    CythonTransform, VisitorTransform,
+    CythonTransform, VisitorTransform, TreeVisitor,
    ScopeTrackingTransform, EnvTransform)

-#class NameNodeCollector(TreeVisitor):
-#    cdef list name_nodes
+cdef class NameNodeCollector(TreeVisitor):
+    cdef list name_nodes

 cdef class SkipDeclarations: # (object):
    pass

--- a/Cython/Compiler/ParseTreeTransforms.py
+++ b/Cython/Compiler/ParseTreeTransforms.py
@@ -26,11 +26,12 @@ class NameNodeCollector(TreeVisitor):
        super(NameNodeCollector, self).__init__()
        self.name_nodes = []

-    visit_Node = TreeVisitor.visitchildren
-
    def visit_NameNode(self, node):
        self.name_nodes.append(node)

+    def visit_Node(self, node):
+        self._visitchildren(node, None)
+

 class SkipDeclarations(object):
    """