From 4189eb24789bd5d1d9ef4ecea567f4119cbb325c Mon Sep 17 00:00:00 2001
From: Michael Williamson <mike@zwobble.org>
Date: Sun, 31 Jul 2016 19:13:45 +0100
Subject: Implement for loops using desugar

---
 .../org/zwobble/couscous/backends/python/PythonCodeGenerator.java | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'src/main/java/org/zwobble/couscous/backends/python')

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();
 
-- 
cgit v1.2.3