From 39bcef63bdca2a077df1f82edbed7399911856f3 Mon Sep 17 00:00:00 2001 From: Michael <35783820+mib1185@users.noreply.github.com> Date: Sun, 9 Feb 2025 21:17:11 +0100 Subject: [PATCH] Explicitly pass in the config_entry in hko coordinator (#138154) explicitly pass in the config_entry in coordinator --- homeassistant/components/hko/__init__.py | 2 +- homeassistant/components/hko/coordinator.py | 11 ++++++++++- 2 files changed, 11 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/hko/__init__.py b/homeassistant/components/hko/__init__.py index 5b06644580e..b7e21f731d8 100644 --- a/homeassistant/components/hko/__init__.py +++ b/homeassistant/components/hko/__init__.py @@ -25,7 +25,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: )[KEY_DISTRICT] websession = async_get_clientsession(hass) - coordinator = HKOUpdateCoordinator(hass, websession, district, location) + coordinator = HKOUpdateCoordinator(hass, entry, websession, district, location) await coordinator.async_config_entry_first_refresh() hass.data.setdefault(DOMAIN, {})[entry.entry_id] = coordinator diff --git a/homeassistant/components/hko/coordinator.py b/homeassistant/components/hko/coordinator.py index 566ba5dcf5e..5845e8831fe 100644 --- a/homeassistant/components/hko/coordinator.py +++ b/homeassistant/components/hko/coordinator.py @@ -26,6 +26,7 @@ from homeassistant.components.weather import ( ATTR_FORECAST_TEMP_LOW, ATTR_FORECAST_TIME, ) +from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed @@ -69,8 +70,15 @@ _LOGGER = logging.getLogger(__name__) class HKOUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]): """HKO Update Coordinator.""" + config_entry: ConfigEntry + def __init__( - self, hass: HomeAssistant, session: ClientSession, district: str, location: str + self, + hass: HomeAssistant, + config_entry: ConfigEntry, + session: ClientSession, + district: str, + location: str, ) -> None: """Update data via library.""" self.location = location @@ -80,6 +88,7 @@ class HKOUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]): super().__init__( hass, _LOGGER, + config_entry=config_entry, name=DOMAIN, update_interval=timedelta(minutes=15), )