evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Async speedup add_device callback (#3910)

Browse source 
* Speed up entities processing from add_device callback

* fix lint

* fix bug
This commit is contained in:
Pascal Vizeli 2016-10-17 00:35:57 +02:00 • committed by GitHub
parent 6581dc2381
commit 7484152be1
1 changed files with 9 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
homeassistant/helpers/entity_component.py
View file

@ -289,11 +289,9 @@ class EntityPlatform(object):
This method must be run in the event loop.
"""
for entity in new_entities:
ret = yield from self.component.async_add_entity(entity, self)
if ret:
self.platform_entities.append(entity)
tasks = [self._async_process_entity(entity) for entity in new_entities]
yield from asyncio.gather(*tasks, loop=self.component.hass.loop)
yield from self.component.async_update_group()
if self._async_unsub_polling is not None or \
@ -305,6 +303,13 @@ class EntityPlatform(object):
self.component.hass, self._update_entity_states,
second=range(0, 60, self.scan_interval))
@asyncio.coroutine
def _async_process_entity(self, new_entity):
"""Add entities to StateMachine."""
ret = yield from self.component.async_add_entity(new_entity, self)
if ret:
self.platform_entities.append(new_entity)
@asyncio.coroutine
def async_reset(self):
"""Remove all entities and reset data.