Explicitly pass in the config_entry in rabbitair coordinator (#137985)

explicitly pass in the config_entry in coordinator
This commit is contained in:
Michael 2025-02-09 15:52:46 +01:00 committed by GitHub
parent 106c5c661e
commit a8c4cc7269
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:
zeroconf_instance = await zeroconf.async_get_async_instance(hass)
device: Client = UdpClient(host, token, zeroconf=zeroconf_instance)
coordinator = RabbitAirDataUpdateCoordinator(hass, device)
coordinator = RabbitAirDataUpdateCoordinator(hass, entry, device)
await coordinator.async_config_entry_first_refresh()

View File

@ -7,6 +7,7 @@ from typing import Any, cast
from rabbitair import Client, State
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.debounce import Debouncer
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@ -42,12 +43,17 @@ class RabbitAirDebouncer(Debouncer[Coroutine[Any, Any, None]]):
class RabbitAirDataUpdateCoordinator(DataUpdateCoordinator[State]):
"""Class to manage fetching data from single endpoint."""
def __init__(self, hass: HomeAssistant, device: Client) -> None:
config_entry: ConfigEntry
def __init__(
self, hass: HomeAssistant, config_entry: ConfigEntry, device: Client
) -> None:
"""Initialize global data updater."""
self.device = device
super().__init__(
hass,
_LOGGER,
config_entry=config_entry,
name="rabbitair",
update_interval=timedelta(seconds=10),
request_refresh_debouncer=RabbitAirDebouncer(hass),