1 files changed, 7 insertions, 0 deletions
diff --git a/crowbar_reference_compiler/ast.py b/crowbar_reference_compiler/ast.py
index 142fa8e..7ae7ff9 100644
--- a/crowbar_reference_compiler/ast.py
+++ b/crowbar_reference_compiler/ast.py
@@ -531,6 +531,13 @@ class ASTBuilder(NodeVisitor):
             assert kwd.type == 'else'
         return IfStatement(condition, then, els)
 
+    def visit_WhileStatement(self, node, visited_children):
+        kwd, lparen, condition, rparen, body = visited_children
+        assert kwd.type == 'while'
+        assert lparen.type == '('
+        assert rparen.type == ')'
+        return WhileStatement(condition, body)
+
     def visit_ReturnStatement(self, node, visited_children):
         ret, body, semi = visited_children
         assert ret.type == 'return'