algorithm.{Visitor,Transformer}: don't die on lists of strings.

whitequark · whitequark · commit 2167c885d73b · 2015-06-06T17:43:36.000+03:00
diff --git a/pythonparser/algorithm.py b/pythonparser/algorithm.py
@@ -37,7 +37,7 @@ def visit(self, obj):
         """Visit a node or a list of nodes. Other values are ignored"""
         if isinstance(obj, list):
             for elt in obj:
-                self._visit_one(elt)
+                self.visit(elt)
         elif isinstance(obj, ast.AST):
             self._visit_one(obj)
 
@@ -79,7 +79,7 @@ def _visit_one(self, node):
     def visit(self, obj):
         """Visit a node or a list of nodes. Other values are ignored"""
         if isinstance(obj, list):
-            return list(filter(lambda x: x is not None, map(self._visit_one, obj)))
+            return list(filter(lambda x: x is not None, map(self.visit, obj)))
         elif isinstance(obj, ast.AST):
             return self._visit_one(obj)
 
