From e44bf2c9e4108ab69a599acdbbf54a02b6984a07 Mon Sep 17 00:00:00 2001 From: Michael Williamson Date: Mon, 5 Oct 2015 18:47:04 +0100 Subject: Use slice syntax instead of __getitem__ --- .../couscous/backends/python/PythonSerializer.java | 16 ++++++++++++++++ 1 file changed, 16 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 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; @@ -95,6 +96,21 @@ public class PythonSerializer implements PythonNodeVisitor { writer.writeSymbol(")"); } + @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(() -> { -- cgit v1.2.3