Implement attribute access and calls in Python backend

3 files changed, 39 insertions, 0 deletions

diff --git a/src/main/java/org/zwobble/couscous/backends/python/ast/PythonAttributeAccessNode.java b/src/main/java/org/zwobble/couscous/backends/python/ast/PythonAttributeAccessNode.java
new file mode 100644
index 0000000..ccd795f
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/backends/python/ast/PythonAttributeAccessNode.java
@@ -0,0 +1,16 @@
+package org.zwobble.couscous.backends.python.ast;
+
+import org.zwobble.couscous.backends.python.ast.visitors.PythonNodeVisitor;
+
+import lombok.Value;
+
+@Value(staticConstructor="pythonAttributeAccess")
+public class PythonAttributeAccessNode implements PythonExpressionNode {
+    PythonExpressionNode left;
+    String attributeName;
+    
+    @Override
+    public void accept(PythonNodeVisitor visitor) {
+        visitor.visit(this);
+    }
+}
diff --git a/src/main/java/org/zwobble/couscous/backends/python/ast/PythonCallNode.java b/src/main/java/org/zwobble/couscous/backends/python/ast/PythonCallNode.java
new file mode 100644
index 0000000..92c8b20
--- /dev/null
+++ b/src/main/java/org/zwobble/couscous/backends/python/ast/PythonCallNode.java
@@ -0,0 +1,19 @@
+package org.zwobble.couscous.backends.python.ast;
+
+import java.util.List;
+
+import org.zwobble.couscous.backends.python.ast.visitors.PythonNodeVisitor;
+
+import lombok.Value;
+
+@Value(staticConstructor="pythonCall")
+public class PythonCallNode implements PythonExpressionNode {
+    PythonExpressionNode callee;
+    List<PythonExpressionNode> arguments;
+    
+    
+    @Override
+    public void accept(PythonNodeVisitor visitor) {
+        visitor.visit(this);
+    }
+}
diff --git a/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonNodeVisitor.java b/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonNodeVisitor.java
index 0cd43ae..2e38d3b 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonNodeVisitor.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/ast/visitors/PythonNodeVisitor.java
@@ -1,7 +1,9 @@
 package org.zwobble.couscous.backends.python.ast.visitors;
 
 import org.zwobble.couscous.backends.python.ast.PythonAssignmentNode;
+import org.zwobble.couscous.backends.python.ast.PythonAttributeAccessNode;
 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.PythonFunctionDefinitionNode;
@@ -18,6 +20,8 @@ public interface PythonNodeVisitor {
     void visit(PythonBooleanLiteralNode booleanLiteral);
     void visit(PythonVariableReferenceNode reference);
     void visit(PythonConditionalExpressionNode conditional);
+    void visit(PythonAttributeAccessNode attributeAccess);
+    void visit(PythonCallNode call);
     
     void visit(PythonReturnNode pythonReturn);
     void visit(PythonPassNode pass);