Add Python array nodes

1 files changed, 14 insertions, 1 deletions

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 89711a2..0a7735d 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
@@ -46,7 +46,20 @@ public class PythonSerializer implements PythonNodeVisitor {
     public void visit(PythonVariableReferenceNode reference) {
         writer.writeIdentifier(reference.getName());
     }
-    
+
+    @Override
+    public void visit(PythonArrayNode array) {
+        writer.writeSymbol("[");
+        writer.writeWithSeparator(
+            array.getElements(),
+            element -> write(element),
+            () -> {
+                writer.writeSymbol(",");
+                writer.writeSpace();
+            });
+        writer.writeSymbol("]");
+    }
+
     @Override
     public void visit(PythonConditionalExpressionNode conditional) {
         writeParenthesised(conditional.getTrueValue(), conditional);