feat(IfElseCondition): add node to interpret if-else statements

aadhithya · May 16, 2022 · 8ae2192 · 8ae2192
1 parent 56e7a88
commit 8ae2192
Showing 1 changed file with 15 additions and 1 deletion.
diff --git a/rajinipp/ast/control.py b/rajinipp/ast/control.py
@@ -8,6 +8,20 @@ def __init__(self, condition, value) -> None:
         self.value = value
 
     def eval(self):
-        if self.condition:
+        if self.condition.eval():
             return self.value.eval()
         return
+
+
+class IfElseCondition(Node):
+    def __init__(self, condition, true_value, false_value) -> None:
+        super().__init__()
+        self.condition = condition
+        self.true_value = true_value
+        self.false_value = false_value
+
+    def eval(self):
+        if self.condition.eval():
+            return self.true_value.eval()
+        else:
+            return self.false_value.eval()