fixed for visitors when node.init is a FunctionCall

duilio · May 7, 2012 · 3832095 · 3832095
1 parent 71f4940
commit 3832095
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/slimit/visitors/ecmavisitor.py b/src/slimit/visitors/ecmavisitor.py
@@ -107,7 +107,7 @@ def visit_For(self, node):
             s += self.visit(node.init)
         if node.init is None:
             s += ' ; '
-        elif isinstance(node.init, (ast.Assign, ast.Comma)):
+        elif isinstance(node.init, (ast.Assign, ast.Comma, ast.FunctionCall)):
             s += '; '
         else:
             s += ' '

diff --git a/src/slimit/visitors/minvisitor.py b/src/slimit/visitors/minvisitor.py
@@ -140,7 +140,8 @@ def visit_For(self, node):
             s += self.visit(node.init)
         if node.init is None:
             s += ';'
-        elif isinstance(node.init, (ast.Assign, ast.Comma, ast.Conditional)):
+        elif isinstance(node.init, (ast.Assign, ast.Comma, ast.Conditional,
+                                    ast.FunctionCall)):
             s += ';'
         else:
             s += ''