implement getting states using self.hass.states.async_entity_ids

Author: basnijholt

custom_components/pyscript/handler.py

@@ -55,11 +55,7 @@ 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
-        ]
+        return self.hass.states.async_entity_ids(domain)
 
     async def async_sleep(self, duration):
         """Implement task.sleep()."""