feat(flow_parser.py): add parser to support loops, rename cond_parser…

… to flow_parser
aadhithya · May 16, 2022 · 78cc1da · 78cc1da
1 parent ab91aad
commit 78cc1da
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/rajinipp/parser/cond_parser.py → rajinipp/parser/flow_parser.py b/rajinipp/parser/cond_parser.py → rajinipp/parser/flow_parser.py
@@ -1,4 +1,6 @@
-from ..ast.control import IfCondition, IfElseCondition
+import pdb
+
+from ..ast.control import ForLoop, IfCondition, IfElseCondition
 
 
 class ConditionalParser:
@@ -17,3 +19,15 @@ def if_stmt(p):
         )
         def if_else_stmt(p):
             return IfElseCondition(p[1], p[3], p[7])
+
+
+class LoopParser:
+    """parse loops"""
+
+    def parse(self, pg):
+        @pg.production(
+            "statement : FOR_START forvar FOR_RANGE_START forvar FOR_RANGE_END"
+            + " L_BRACE statements R_BRACE END_BLOCK SEMI_COLON"
+        )
+        def for_loop(p):
+            return ForLoop(p[1], p[3], p[6])