added repeat assignments, and added assert statement

Author: craigbarratt

custom_components/pyscript/eval.py

@@ -655,7 +655,9 @@ async def recurse_assign(self, lhs, val):
 
     async def ast_assign(self, arg):
         """Execute assignment statement."""
-        await self.recurse_assign(arg.targets[0], await self.aeval(arg.value))
+        rhs = await self.aeval(arg.value)
+        for target in arg.targets:
+            await self.recurse_assign(target, rhs)
 
     async def ast_augassign(self, arg):
         """Execute augmented assignment statement (lhs <BinOp>= value)."""
@@ -733,6 +735,13 @@ async def ast_delete(self, arg):
             else:
                 raise NotImplementedError(f"unknown target type {arg1} in del")
 
+    async def ast_assert(self, arg):
+        """Execute assert statement."""
+        if not await self.aeval(arg.test):
+            if arg.msg:
+                raise AssertionError(await self.aeval(arg.msg))
+            raise AssertionError
+
     async def ast_attribute_collapse(self, arg):
         """Combine dotted attributes to allow variable names to have dots."""
         # collapse dotted names, eg:

custom_components/pyscript/handler.py

@@ -59,6 +59,7 @@ def init(hass):
         Handler.loggers = {}
 
     async def entity_ids(domain=None):
+        """Implement entity_ids."""
         return Handler.hass.states.async_entity_ids(domain)
 
     async def async_sleep(duration):

tests/custom_components/pyscript/test_unit_eval.py

@@ -140,6 +140,7 @@
         [10, 11, 12, 13, 14, 15, 16, 17],
     ],
     ["(x, y) = (1, 2); [x, y]", [1, 2]],
+    ["a, b = (x, y) = (1, 2); [a, b, x, y]", [1, 2, 1, 2]],
     ["y = [1,2]; (x, y[0]) = (3, 4); [x, y]", [3, [4, 2]]],
     ["((x, y), (z, t)) = ((1, 2), (3, 4)); [x, y, z, t]", [1, 2, 3, 4]],
     [
@@ -697,6 +698,11 @@ def test_eval(hass):
         "(x, y) = 1",
         "Exception in test line 1 column 9: cannot unpack non-iterable object",
     ],
+    [
+        "assert 1 == 0, 'this is an error'",
+        "Exception in test line 1 column 15: this is an error",
+    ],
+    ["assert 1 == 0", "Exception in test line 1 column 12: "],
     [
         "import math; math.sinXYZ",
         "Exception in test line 1 column 13: module 'math' has no attribute 'sinXYZ'",