Explicitly pass in the config_entry in swiss_public_transport coordinator (#137924)

explicitly pass in the config_entry in coordinator
This commit is contained in:
Michael 2025-02-08 21:14:47 +01:00 committed by GitHub
parent 022119e74f
commit a7dbcf72c2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 5 additions and 1 deletions

View File

@ -96,7 +96,9 @@ async def async_setup_entry(
}, },
) from e ) from e
coordinator = SwissPublicTransportDataUpdateCoordinator(hass, opendata, time_offset) coordinator = SwissPublicTransportDataUpdateCoordinator(
hass, entry, opendata, time_offset
)
await coordinator.async_config_entry_first_refresh() await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator entry.runtime_data = coordinator

View File

@ -61,6 +61,7 @@ class SwissPublicTransportDataUpdateCoordinator(
def __init__( def __init__(
self, self,
hass: HomeAssistant, hass: HomeAssistant,
config_entry: SwissPublicTransportConfigEntry,
opendata: OpendataTransport, opendata: OpendataTransport,
time_offset: dict[str, int] | None, time_offset: dict[str, int] | None,
) -> None: ) -> None:
@ -68,6 +69,7 @@ class SwissPublicTransportDataUpdateCoordinator(
super().__init__( super().__init__(
hass, hass,
_LOGGER, _LOGGER,
config_entry=config_entry,
name=DOMAIN, name=DOMAIN,
update_interval=timedelta(seconds=DEFAULT_UPDATE_TIME), update_interval=timedelta(seconds=DEFAULT_UPDATE_TIME),
) )