Explicitly pass in the config_entry in prusalink coordinator (#138036)
explicitly pass in the config_entry in coordinatorpull/137224/head
parent
f8169f1110
commit
685e882847
|
@ -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()
|
||||
|
|
|
@ -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:
|
||||
|
|
Loading…
Reference in New Issue