Implement for loops using desugar

1 files changed, 7 insertions, 1 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 477a26f..29e7274 100644
--- a/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java
+++ b/src/main/java/org/zwobble/couscous/backends/python/PythonCodeGenerator.java
@@ -11,6 +11,7 @@ import org.zwobble.couscous.ast.visitors.NodeTransformer;
 import org.zwobble.couscous.backends.naming.Names;
 import org.zwobble.couscous.backends.python.ast.*;
 import org.zwobble.couscous.transforms.DesugarForEachToWhile;
+import org.zwobble.couscous.transforms.DesugarForToWhile;
 import org.zwobble.couscous.transforms.DesugarSwitchToIfElse;
 import org.zwobble.couscous.types.ScalarType;
 import org.zwobble.couscous.types.Type;
@@ -61,8 +62,13 @@ import static org.zwobble.couscous.util.ExtraLists.*;
 public class PythonCodeGenerator {
     public static PythonModuleNode generateCode(TypeNode sugaredTypeNode) {
         NodeTransformer switchTransformer = DesugarSwitchToIfElse.transformer();
+        NodeTransformer forTransformer = DesugarForToWhile.transformer();
         NodeTransformer forEachTransformer = DesugarForEachToWhile.transformer();
-        TypeNode typeNode = forEachTransformer.transformTypeDeclaration(switchTransformer.transformTypeDeclaration(sugaredTypeNode));
+        TypeNode typeNode = forTransformer.transformTypeDeclaration(
+            forEachTransformer.transformTypeDeclaration(
+                switchTransformer.transformTypeDeclaration(sugaredTypeNode)
+            )
+        );
 
         Iterator<PythonImportNode> imports = generateImports(typeNode).iterator();