feat(parser): Helper recognises Assign Expr

nyjc-computing · May 4, 2022 · 6f4a739 · 6f4a739
1 parent 9b0a410
commit 6f4a739
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/parser.py b/parser.py
@@ -25,14 +25,16 @@ def consume(tokens):
 def makeExpr(
     *,
     type=None, value=None,
-    frame=None, name=None,
+    frame=None, name=None, expr=None,
     left=None, oper=None, right=None,
     callable=None, args=None,
     token=None,
 ):
     if name is not None:
         if frame is not None:
             return Get(frame, name, token=token)
+        elif expr is not None:
+            return Assign(name, expr, token=token)
         else:
             return Name(name, token=token)
     if type is not None and value is not None:
@@ -168,6 +170,8 @@ def expression(tokens):
     expr = equality(tokens)
     return expr
 
+def assignment(tokens):
+
 # Statement parsers
 
 def outputStmt(tokens):