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 | 16 |
1 files changed, 16 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 1e10315..abab1bf 100644 --- a/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java +++ b/src/main/java/org/zwobble/couscous/backends/python/PythonSerializer.java @@ -9,6 +9,7 @@ import org.zwobble.couscous.backends.python.ast.PythonClassNode; import org.zwobble.couscous.backends.python.ast.PythonConditionalExpressionNode; import org.zwobble.couscous.backends.python.ast.PythonExpressionNode; import org.zwobble.couscous.backends.python.ast.PythonFunctionDefinitionNode; +import org.zwobble.couscous.backends.python.ast.PythonGetSliceNode; import org.zwobble.couscous.backends.python.ast.PythonIntegerLiteralNode; import org.zwobble.couscous.backends.python.ast.PythonModuleNode; import org.zwobble.couscous.backends.python.ast.PythonNode; @@ -96,6 +97,21 @@ public class PythonSerializer implements PythonNodeVisitor { } @Override + public void visit(PythonGetSliceNode getSlice) { + writeParenthesised(getSlice.getReceiver()); + writer.writeSymbol("["); + // TODO: remove this duplication + if (!getSlice.getArguments().isEmpty()) { + write(getSlice.getArguments().get(0)); + for (val argumentName : skip(getSlice.getArguments(), 1)) { + writer.writeSymbol(":"); + write(argumentName); + } + } + writer.writeSymbol("]"); + } + + @Override public void visit(PythonReturnNode pythonReturn) { writer.writeStatement(() -> { writer.writeKeyword("return"); |
