Use slice syntax instead of __getitem__

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");