cdef another class in ParseTreeTransforms

diff --git a/Cython/Compiler/ParseTreeTransforms.pxd b/Cython/Compiler/ParseTreeTransforms.pxd
index 20011e94f04596c338916accced406c7b59da43c..77dd4a9ae72f5725ef213d8792ce130a0010d35b 100644 (file)
--- a/Cython/Compiler/ParseTreeTransforms.pxd
+++ b/Cython/Compiler/ParseTreeTransforms.pxd
@@ -1,10 +1,10 @@
 
 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

diff --git a/Cython/Compiler/ParseTreeTransforms.py b/Cython/Compiler/ParseTreeTransforms.py
index 3ab87d033428af1cf6eef07c7bcb965b2c830f1f..b9b0655471f731966d6e5b92fd57006ad71cf905 100644 (file)
--- 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):
     """