feat(ConditionalParser): add support for if-else statements

aadhithya · May 16, 2022 · 2a219dc · 2a219dc
1 parent bcb55aa
commit 2a219dc
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/rajinipp/parser/cond_parser.py b/rajinipp/parser/cond_parser.py
@@ -1,4 +1,4 @@
-from ..ast.control import IfCondition
+from ..ast.control import IfCondition, IfElseCondition
 
 
 class ConditionalParser:
@@ -10,3 +10,10 @@ def parse(self, pg):
         )
         def if_stmt(p):
             return IfCondition(p[1], p[3])
+
+        @pg.production(
+            "statement : IF_COND logical_expression L_BRACE statements R_BRACE "
+            + "ELSE_COND L_BRACE statements R_BRACE END_BLOCK SEMI_COLON"
+        )
+        def if_else_stmt(p):
+            return IfElseCondition(p[1], p[3], p[7])