From 38cc6849339b37c29af80e2affc1931479d61c05 Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Mon, 5 Oct 2015 13:31:20 +0100
Subject: Support assignment statements in the Python backend

---
 .../zwobble/couscous/backends/python/PythonSerializer.java  | 13 +++++++++++++
 1 file changed, 13 insertions(+)

(limited to 'src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java')

diff --git a/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
index 4893cb6..bc5fb19 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
@@ -1,5 +1,6 @@
 package org.zwobble.couscous.backends.python;
 
+import org.zwobble.couscous.backends.python.ast.PythonAssignmentNode;
 import org.zwobble.couscous.backends.python.ast.PythonBlock;
 import org.zwobble.couscous.backends.python.ast.PythonBooleanLiteralNode;
 import org.zwobble.couscous.backends.python.ast.PythonClassNode;
@@ -106,6 +107,18 @@ public class PythonSerializer implements PythonNodeVisitor<Void> {
         return null;
     }
 
+    @Override
+    public Void visit(PythonAssignmentNode assignment) {
+        writer.startStatement();
+        write(assignment.getTarget());
+        writer.writeSpace();
+        writer.writeSymbol("=");
+        writer.writeSpace();
+        write(assignment.getValue());
+        writer.endStatement();
+        return null;
+    }
+
     private void writeArgumentNames(PythonFunctionDefinitionNode functionDefinition) {
         if (!functionDefinition.getArgumentNames().isEmpty()) {
             writer.writeIdentifier(functionDefinition.getArgumentNames().get(0));
-- 
cgit v1.2.3