Support assignment statements in the Python backend

1 files changed, 22 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java b/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java
new file mode 100644
index 0000000..861e847
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/ast/AssignmentNode.java
@@ -0,0 +1,22 @@
+package org.zwobble.couscous.ast;
+
+import org.zwobble.couscous.ast.visitors.ExpressionNodeVisitor;
+
+import static org.zwobble.couscous.ast.VariableReferenceNode.reference;
+
+import lombok.Value;
+
+@Value
+public class AssignmentNode implements ExpressionNode {
+    public static AssignmentNode assign(VariableNode target, ExpressionNode value) {
+        return new AssignmentNode(reference(target), value);
+    }
+    
+    VariableReferenceNode target;
+    ExpressionNode value;
+    
+    @Override
+    public <T> T accept(ExpressionNodeVisitor<T> visitor) {
+        return visitor.visit(this);
+    }
+}