Explicitly pass in the config_entry in tami4 coordinator (#137912)

explicitly pass in the config_entry in coordinator
This commit is contained in:
Michael 2025-02-09 14:48:38 +01:00 committed by GitHub
parent 583b2e285b
commit d92e2194d0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 2 deletions

View File

@ -26,7 +26,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
except exceptions.TokenRefreshFailedException as ex:
raise ConfigEntryNotReady("Error connecting to API") from ex
coordinator = Tami4EdgeCoordinator(hass, api)
coordinator = Tami4EdgeCoordinator(hass, entry, api)
await coordinator.async_config_entry_first_refresh()
hass.data.setdefault(DOMAIN, {})[entry.entry_id] = {

View File

@ -7,6 +7,7 @@ import logging
from Tami4EdgeAPI import Tami4EdgeAPI, exceptions
from Tami4EdgeAPI.water_quality import WaterQuality
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
@ -36,11 +37,16 @@ class FlattenedWaterQuality:
class Tami4EdgeCoordinator(DataUpdateCoordinator[FlattenedWaterQuality]):
"""Tami4Edge water quality coordinator."""
def __init__(self, hass: HomeAssistant, api: Tami4EdgeAPI) -> None:
config_entry: ConfigEntry
def __init__(
self, hass: HomeAssistant, config_entry: ConfigEntry, api: Tami4EdgeAPI
) -> None:
"""Initialize the water quality coordinator."""
super().__init__(
hass,
_LOGGER,
config_entry=config_entry,
name="Tami4Edge water quality coordinator",
update_interval=timedelta(minutes=60),
)