add function states that returns all entities

Author: basnijholt

custom_components/pyscript/handler.py

@@ -30,6 +30,7 @@ def __init__(self, hass):
             "task.unique": self.task_unique,
             "service.call": self.service_call,
             "service.has_service": self.service_has_service,
+            "states": self.states,
         }
 
         #
@@ -53,6 +54,13 @@ def __init__(self, hass):
         #
         self.loggers = {}
 
+    async def states(self, domain=None):
+        return [
+            state
+            for state in self.hass.states.keys()
+            if state.split(".", 1)[0] == domain
+        ]
+
     async def async_sleep(self, duration):
         """Implement task.sleep()."""
         await asyncio.sleep(float(duration))