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

Call async added to hass super in Risco (#100444)

Browse source
This commit is contained in:
Joost Lekkerkerker 2023-09-26 20:02:00 +02:00 • committed by GitHub
parent ae7ede1253
commit bdfdeb2bc0
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
homeassistant/components/risco/entity.py
View file

@ -31,17 +31,10 @@ class RiscoCloudEntity(CoordinatorEntity[RiscoDataUpdateCoordinator]):
def _get_data_from_coordinator(self) -> None:
raise NotImplementedError
def _refresh_from_coordinator(self) -> None:
def _handle_coordinator_update(self) -> None:
self._get_data_from_coordinator()
self.async_write_ha_state()
# pylint: disable-next=hass-missing-super-call
async def async_added_to_hass(self) -> None:
"""When entity is added to hass."""
self.async_on_remove(
self.coordinator.async_add_listener(self._refresh_from_coordinator)
)
@property
def _risco(self):
"""Return the Risco API object."""

7
homeassistant/components/risco/sensor.py
View file

@ -86,15 +86,12 @@ class RiscoSensor(CoordinatorEntity[RiscoEventsDataUpdateCoordinator], SensorEnt
self._attr_name = f"Risco {self.coordinator.risco.site_name} {name} Events"
self._attr_device_class = SensorDeviceClass.TIMESTAMP
# pylint: disable-next=hass-missing-super-call
async def async_added_to_hass(self) -> None:
"""When entity is added to hass."""
await super().async_added_to_hass()
self._entity_registry = er.async_get(self.hass)
self.async_on_remove(
self.coordinator.async_add_listener(self._refresh_from_coordinator)
)
def _refresh_from_coordinator(self):
def _handle_coordinator_update(self):
events = self.coordinator.data
for event in reversed(events):
if event.category_id in self._excludes: