small correction for RebuildVisitor

--HG--
branch : _ast_compat

1 files changed, 9 insertions, 2 deletions

diff --git a/utils.py b/utils.py
index a305f069..09387fb5 100644
--- a/utils.py
+++ b/utils.py
@@ -201,10 +201,17 @@ class ASTVisitor(object):
     def visit_bool(self, node):
         """dummy method for visiting an Bool node"""
 
-REDIRECT = {'Expr': 'Discard',
-            'ImportFrom': 'From',
+REDIRECT = {
             'Attribute': 'Getattr',
+            'Call': 'CallFunc',
+            'ClassDef': 'Class',
             'comprehension': "ListCompFor",
+            'excepthandler': 'ExceptHandler',
+            'Expr': 'Discard',
+            'FunctionDef': 'Function',
+            'GeneratorExp': 'GenExpr',
+            'ImportFrom': 'From',
+            'keyword': 'Keyword',
             'Repr': 'Backquote',
 
             'Add': 'BinOp',