jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 21:27:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix Swiss public transport initial data for attributes (#107452)

Browse Source 
faster initial data for attributes
This commit is contained in:
Cyrill Raccaud 2024-01-07 16:19:58 +01:00 committed by GitHub
parent da8ce7bbf3
commit 3139e92696
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
homeassistant/components/swiss_public_transport/sensor.py
View File

@ -122,15 +122,25 @@ class SwissPublicTransportSensor(
entry_type=DeviceEntryType.SERVICE, entry_type=DeviceEntryType.SERVICE,
) )
async def async_added_to_hass(self) -> None:
"""Prepare the extra attributes at start."""
self._async_update_attrs()
await super().async_added_to_hass()
@callback @callback
def _handle_coordinator_update(self) -> None: def _handle_coordinator_update(self) -> None:
"""Handle the state update and prepare the extra state attributes.""" """Handle the state update and prepare the extra state attributes."""
self._async_update_attrs()
return super()._handle_coordinator_update()
@callback
def _async_update_attrs(self) -> None:
"""Update the extra state attributes based on the coordinator data."""
self._attr_extra_state_attributes = { self._attr_extra_state_attributes = {
key: value key: value
for key, value in self.coordinator.data.items() for key, value in self.coordinator.data.items()
if key not in {"departure"} if key not in {"departure"}
} }
return super()._handle_coordinator_update()
@property @property
def native_value(self) -> str: def native_value(self) -> str: