From 9e850b3a09ceebf6140e77982b45abab0c57ad6c Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Mon, 5 Oct 2015 17:59:27 +0100
Subject: Implement attribute access and calls in Python backend

---
 .../couscous/backends/python/PythonSerializer.java | 24 ++++++++++++++++++++++
 1 file changed, 24 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 7eaf2fb..1e10315 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java
@@ -1,8 +1,10 @@
 package org.zwobble.couscous.backends.python;
 
 import org.zwobble.couscous.backends.python.ast.PythonAssignmentNode;
+import org.zwobble.couscous.backends.python.ast.PythonAttributeAccessNode;
 import org.zwobble.couscous.backends.python.ast.PythonBlock;
 import org.zwobble.couscous.backends.python.ast.PythonBooleanLiteralNode;
+import org.zwobble.couscous.backends.python.ast.PythonCallNode;
 import org.zwobble.couscous.backends.python.ast.PythonClassNode;
 import org.zwobble.couscous.backends.python.ast.PythonConditionalExpressionNode;
 import org.zwobble.couscous.backends.python.ast.PythonExpressionNode;
@@ -71,6 +73,28 @@ public class PythonSerializer implements PythonNodeVisitor {
         writeParenthesised(conditional.getFalseValue());
     }
 
+    @Override
+    public void visit(PythonAttributeAccessNode attributeAccess) {
+        writeParenthesised(attributeAccess.getLeft());
+        writer.writeSymbol(".");
+        writer.writeIdentifier(attributeAccess.getAttributeName());
+    }
+
+    @Override
+    public void visit(PythonCallNode call) {
+        writeParenthesised(call.getCallee());
+        writer.writeSymbol("(");
+        if (!call.getArguments().isEmpty()) {
+            write(call.getArguments().get(0));
+            for (val argumentName : skip(call.getArguments(), 1)) {
+                writer.writeSymbol(",");
+                writer.writeSpace();
+                write(argumentName);
+            }
+        }
+        writer.writeSymbol(")");
+    }
+
     @Override
     public void visit(PythonReturnNode pythonReturn) {
         writer.writeStatement(() -> {
-- 
cgit v1.2.3