Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
homeassistant/components/prusalink/__init__.py
View File

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

10
homeassistant/components/prusalink/coordinator.py
View File

@ -37,12 +37,18 @@ class PrusaLinkUpdateCoordinator(DataUpdateCoordinator[T], ABC):
config_entry: ConfigEntry config_entry: ConfigEntry
expect_change_until = 0.0 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.""" """Initialize the update coordinator."""
self.api = api self.api = api
super().__init__( 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: async def _async_update_data(self) -> T: