jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-11-12 12:30:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

Entity to handle updates via events (#24733)

Browse Source 
* Entity to handle updates via events

* Fix a bug

* Update entity.py
This commit is contained in:
Paulus Schoutsen
2019-06-26 09:22:51 -07:00
committed by GitHub
parent 9e0636eefa
commit 06af6f19a3
5 changed files with 71 additions and 66 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/helpers/restore_state.py
View File

@@ -186,18 +186,18 @@ class RestoreStateData():
class RestoreEntity(Entity):
"""Mixin class for restoring previous entity state."""
async def async_added_to_hass(self) -> None:
async def async_internal_added_to_hass(self) -> None:
"""Register this entity as a restorable entity."""
_, data = await asyncio.gather(
super().async_added_to_hass(),
super().async_internal_added_to_hass(),
RestoreStateData.async_get_instance(self.hass),
)
data.async_restore_entity_added(self.entity_id)
async def async_will_remove_from_hass(self) -> None:
async def async_internal_will_remove_from_hass(self) -> None:
"""Run when entity will be removed from hass."""
_, data = await asyncio.gather(
super().async_will_remove_from_hass(),
super().async_internal_will_remove_from_hass(),
RestoreStateData.async_get_instance(self.hass),
)
data.async_restore_entity_removed(self.entity_id)