We can now also do cell assignment operations

Author: JessicaTegner

code_generator.py

@@ -1,5 +1,6 @@
 import logging
 
+from Token import Token
 from dictionary import Dictionary
 
 logger = logging.getLogger(__name__)
@@ -80,12 +81,15 @@ def generate_IfNode(self, node):
             return f'if {condition_code}:\n{body_code}'
 
     def generate_AssignmentNode(self, node):
-        code = f"{node.identifier.value} = "
-        if node.value.type == Dictionary.IDENTIFIER:
-            code += f"{node.value.value}"
+        if not isinstance(node.identifier, Token):
+            identifier_code = self.generate(node.identifier, is_root=False)
         else:
-            code += f"{self.generate(node.value, is_root=False)}"
-        return code
+            identifier_code = node.identifier.value
+        if not isinstance(node.value, Token):
+            value_code = self.generate(node.value, is_root=False)
+        else:
+            value_code = node.value.value
+        return f"{identifier_code} = {value_code}"
 
     def generate_IdentifierNode(self, node):
         return node.value

custom_parser.py

@@ -25,13 +25,13 @@ def _split_into_lines(self, tokens):
 
     def parse_line(self, line):
         """Parses a single line of tokens into a node."""
-        #print(f"Parsing line: {line}")
+        print(f"Parsing line: {line}")
         if not line:
             return None  # handle empty lines
         first_token = line[0]
         if first_token.type == Dictionary.KEYWORD and first_token.value == 'if':
             return self.parse_if_statement(line)
-        elif first_token.type == Dictionary.IDENTIFIER:
+        elif first_token.type == Dictionary.IDENTIFIER or first_token.type == Dictionary.CELL:
             return self.parse_assignment(line)
         else:
             return self.parse_expression(line)
@@ -134,7 +134,6 @@ def parse_if_statement(self, line):
 
     def parse_assignment(self, line):
         """Parse an assignment statement."""
-        # Example parsing assuming format `variable = expression`
         # remove whitespace tokens
         tokens = [token for token in line if token.type != Dictionary.WHITE_SPACE]
         # find the assignment token
@@ -146,6 +145,9 @@ def parse_assignment(self, line):
         if i != 1:
             raise ValueError("Invalid assignment statement")
         variable = tokens[0]
+        # if the variable is a cell reference, parse it
+        if variable.type == Dictionary.CELL:
+            variable = self.parse_cell_reference(variable)
         # everything after the assignment token is the expression
         variable_value = tokens[i+1:]
         # if we only have 1 token, and it's an identifier, we can return the variable

text_input.txt

@@ -3,4 +3,4 @@
 if cell a1  is not equal to 1
 	hello is yes
 
-variable is lol
+cell a1 is cell b3