summaryrefslogtreecommitdiff
path: root/utils.py
diff options
context:
space:
mode:
Diffstat (limited to 'utils.py')
-rw-r--r--utils.py11
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',