Add the parsers flow information as classes to parser.representation.

2014-11-03 18:25:10 +01:00
parent e25684d470
commit 55db65434c
2 changed files with 32 additions and 2 deletions
--- a/jedi/parser/pytree.py
+++ b/jedi/parser/pytree.py
@@ -84,6 +84,11 @@ def convert(grammar, raw_node):
        'global_stmt': pr.GlobalStmt,
        'nonlocal_stmt': pr.KeywordStatement,
        'assert_stmt': pr.KeywordStatement,
+        'if_stmt': pr.IfStmt,
+        'with_stmt': pr.WithStmt,
+        'for_stmt': pr.ForStmt,
+        'while_stmt': pr.WhileStmt,
+        'try_stmt': pr.TryStmt,
    }

    ast_mapping = dict((getattr(python_symbols, k), v) for k, v in _ast_mapping.items())