Merge pull request #13 from basnijholt/states

add function entity_ids that returns all entities

Author: craigbarratt

README.md

@@ -265,6 +265,9 @@ In cases where you need to compute the name of the state variable dynamically, o
 set the state attributes, you can use the built-in functions `state.get(name)` and
 `state.set(name, value, attr=None)`; see below.
 
+The function `entity_ids(domain=None)` returns a list of all `entity_id`s of a domain. If domain
+is not specified, it returns all entities.
+
 Also, service names (which are called as functions) take priority over state variable names,
 so if a component has a state variable name that collides with one of its services, you'll
 need to use `state.get(name)` to access that state variable.

custom_components/pyscript/handler.py

@@ -34,6 +34,7 @@ def init(hass):
             "task.unique": Handler.task_unique,
             "service.call": Handler.service_call,
             "service.has_service": Handler.service_has_service,
+            "entity_ids": Handler.entity_ids,
         }
 
         #
@@ -57,6 +58,9 @@ def init(hass):
         #
         Handler.loggers = {}
 
+    async def entity_ids(domain=None):
+        return Handler.hass.states.async_entity_ids(domain)
+
     async def async_sleep(duration):
         """Implement task.sleep()."""
         await asyncio.sleep(float(duration))