diff options
Diffstat (limited to 'src/main')
| -rw-r--r-- | src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java | 22 |
1 files changed, 19 insertions, 3 deletions
diff --git a/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java b/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java index 023d257..067e826 100644 --- a/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java +++ b/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java @@ -1,5 +1,6 @@ package org.zwobble.couscous.backends.python; +import java.util.List; import java.util.stream.Collectors; import org.zwobble.couscous.ast.AssignmentNode; @@ -159,17 +160,25 @@ public class PythonCodeGenerator { if (isPrimitive(methodCall.getReceiver())) { val isStringLength = methodCall.getReceiver().getType().equals(StringValue.REF) && methodCall.getMethodName() == "length"; + val isSubstring = methodCall.getReceiver().getType().equals(StringValue.REF) && + methodCall.getMethodName() == "substring"; if (isStringLength) { return pythonCall( pythonVariableReference("len"), asList(generateExpression(methodCall.getReceiver()))); + } else if (isSubstring) { + return pythonCall( + pythonAttributeAccess( + generateExpression(methodCall.getReceiver()), + "__getitem__"), + asList(pythonCall( + pythonVariableReference("slice"), + generateArguments(methodCall)))); } else { throw new UnsupportedOperationException(); } } else { - val arguments = methodCall.getArguments().stream() - .map(PythonCodeGenerator::generateExpression) - .collect(Collectors.toList()); + val arguments = generateArguments(methodCall); return pythonCall( pythonAttributeAccess( generateExpression(methodCall.getReceiver()), @@ -178,6 +187,13 @@ public class PythonCodeGenerator { } } + private List<PythonExpressionNode> generateArguments( + MethodCallNode methodCall) { + return methodCall.getArguments().stream() + .map(PythonCodeGenerator::generateExpression) + .collect(Collectors.toList()); + } + @Override public PythonExpressionNode visit(StaticMethodCallNode staticMethodCall) { throw new UnsupportedOperationException(); |