Add support for for loops without braces

maurolacy · maurolacy · commit f4e1db3c03b7 · 2016-07-09T11:57:34.000-03:00
diff --git a/java2python/compiler/visitor.py b/java2python/compiler/visitor.py
@@ -434,6 +434,7 @@ def acceptDo(self, node, memo):
         tokens.FOR_EACH,
         tokens.METHOD_CALL,
         tokens.ARGUMENT_LIST,
+        tokens.FOR,
     )
 
     def acceptExpr(self, node, memo):
@@ -452,7 +453,9 @@ def acceptFor(self, node, memo):
         else:
             whileStat.expr.walk(cond, memo)
         whileBlock = self.factory.methodContent(parent=self)
-        if not node.firstChildOfType(tokens.BLOCK_SCOPE).children:
+        if not node.firstChildOfType(tokens.BLOCK_SCOPE): # A for loop without braces
+            whileBlock.walk(node.children[3], memo)
+        elif not node.firstChildOfType(tokens.BLOCK_SCOPE).children:
             self.factory.expr(left='pass', parent=whileBlock)
         else:
             whileBlock.walk(node.firstChildOfType(tokens.BLOCK_SCOPE), memo)
diff --git a/test/ForLoop3.java b/test/ForLoop3.java
@@ -0,0 +1,7 @@
+class ForLoop3 {
+    public static void main(String[] args) {
+        for (int i = 0; i < 3; i++)
+            System.out.println(i);
+        System.out.println("outside of loop");
+    }
+}
diff --git a/test/ForLoop4.java b/test/ForLoop4.java
@@ -0,0 +1,8 @@
+class ForLoop4 {
+    public static void main(String[] args) {
+        for (int i = 0; i < 3; i++)
+            if (i > 0)
+                System.out.println(i);
+        System.out.println("outside of loop");
+    }
+}
