alvin/femcode
1
0
Fork 0
mirror of https://github.com/Alvin-Zilverstand/femcode.git synced 2026-03-06 13:23:39 +01:00
Code Issues Packages Projects Releases Wiki Activity

feat: Add conditional logic (if/else)

Browse Source
This commit is contained in:
Alvin
2025-07-22 16:17:32 +02:00
parent 735ae8e068
commit f3e4be5949
4 changed files with 155 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
src/interpreter.py
View File

@@ -18,15 +18,36 @@ class Interpreter:
def visit_Number(self, node):
return node.value
def visit_String(self, node):
return node.value
def visit_BinOp(self, node):
left_val = self.visit(node.left)
right_val = self.visit(node.right)
if node.op.type == 'PLUS':
return self.visit(node.left) + self.visit(node.right)
return left_val + right_val
elif node.op.type == 'MINUS':
return self.visit(node.left) - self.visit(node.right)
return left_val - right_val
elif node.op.type == 'MUL':
return self.visit(node.left) * self.visit(node.right)
return left_val * right_val
elif node.op.type == 'DIV':
return self.visit(node.left) / self.visit(node.right)
return left_val / right_val
def visit_Comparison(self, node):
left_val = self.visit(node.left)
right_val = self.visit(node.right)
if node.op.type == 'EQ':
return left_val == right_val
elif node.op.type == 'NEQ':
return left_val != right_val
elif node.op.type == 'GT':
return left_val > right_val
elif node.op.type == 'GTE':
return left_val >= right_val
elif node.op.type == 'LT':
return left_val < right_val
elif node.op.type == 'LTE':
return left_val <= right_val
def visit_Print(self, node):
value_to_print = self.visit(node.value)
@@ -43,3 +64,14 @@ class Interpreter:
return self.variables[var_name]
else:
raise NameError(f"name '{var_name}' is not defined")
def visit_Block(self, node):
for statement in node.statements:
self.visit(statement)
def visit_IfStatement(self, node):
condition_result = self.visit(node.condition)
if condition_result:
self.visit(node.if_block)
elif node.else_block:
self.visit(node.else_block)