diff options
Diffstat (limited to 'src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java')
| -rw-r--r-- | src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java | 24 |
1 files changed, 24 insertions, 0 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 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; @@ -72,6 +74,28 @@ public class PythonSerializer implements PythonNodeVisitor { } @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(() -> { writer.writeKeyword("return"); |
