Explicitly pass in the config_entry in prusalink coordinator (#138036)

explicitly pass in the config_entry in coordinator
pull/137224/head
Michael 2025-02-09 17:31:52 +01:00 committed by GitHub
parent f8169f1110
commit 685e882847
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 14 additions and 7 deletions

View File

@ -24,6 +24,7 @@ from .coordinator import (
InfoUpdateCoordinator,
JobUpdateCoordinator,
LegacyStatusCoordinator,
PrusaLinkUpdateCoordinator,
StatusCoordinator,
)
@ -47,11 +48,11 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
entry.data[CONF_PASSWORD],
)
coordinators = {
"legacy_status": LegacyStatusCoordinator(hass, api),
"status": StatusCoordinator(hass, api),
"job": JobUpdateCoordinator(hass, api),
"info": InfoUpdateCoordinator(hass, api),
coordinators: dict[str, PrusaLinkUpdateCoordinator] = {
"legacy_status": LegacyStatusCoordinator(hass, entry, api),
"status": StatusCoordinator(hass, entry, api),
"job": JobUpdateCoordinator(hass, entry, api),
"info": InfoUpdateCoordinator(hass, entry, api),
}
for coordinator in coordinators.values():
await coordinator.async_config_entry_first_refresh()

View File

@ -37,12 +37,18 @@ class PrusaLinkUpdateCoordinator(DataUpdateCoordinator[T], ABC):
config_entry: ConfigEntry
expect_change_until = 0.0
def __init__(self, hass: HomeAssistant, api: PrusaLink) -> None:
def __init__(
self, hass: HomeAssistant, config_entry: ConfigEntry, api: PrusaLink
) -> None:
"""Initialize the update coordinator."""
self.api = api
super().__init__(
hass, _LOGGER, name=DOMAIN, update_interval=self._get_update_interval(None)
hass,
_LOGGER,
config_entry=config_entry,
name=DOMAIN,
update_interval=self._get_update_interval(None),
)
async def _async_update_data(self) -> T: