fixes #14 - tuple assignments in for loop

Author: craigbarratt

custom_components/pyscript/eval.py

@@ -447,10 +447,8 @@ async def ast_if(self, arg):
 
     async def ast_for(self, arg):
         """Execute for statement."""
-        loop_var = await self.aeval(arg.target)
-        loop_iter = await self.aeval(arg.iter)
-        for i in loop_iter:
-            self.sym_table[loop_var] = i
+        for loop_var in await self.aeval(arg.iter):
+            await self.recurse_assign(arg.target, loop_var)
             for arg1 in arg.body:
                 val = await self.aeval(arg1)
                 if isinstance(val, EvalStopFlow):

tests/custom_components/pyscript/test_unit_eval.py

@@ -180,6 +180,29 @@
     ["from math import sqrt as sqroot\nsqroot(1024)", 32],
     [
         """
+d = {"x": 1, "y": 2, "z": 3}
+s = []
+for k, v in d.items():
+    s.append(f"{k}: {v}")
+s
+""",
+        ["x: 1", "y: 2", "z: 3"],
+    ],
+    [
+        """
+d = {"x": 1, "y": 2, "z": 3}
+i = 0
+s = []
+k = [0, 0, 0]
+for k[i], v in d.items():
+    s.append([k.copy(), v])
+    i += 1
+s
+""",
+        [[["x", 0, 0], 1], [["x", "y", 0], 2], [["x", "y", "z"], 3]],
+    ],
+    [
+        """
 def foo(bar=6):
     if bar == 5:
         return