Explicitly pass in the config_entry in sleepiq coordinator (#137946)

explicitly pass in the config_entry in coordinator
This commit is contained in:
Michael 2025-02-08 20:42:44 +01:00 committed by GitHub
parent 22d6913bc5
commit 1374fb23db
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 14 additions and 6 deletions

View File

@ -94,8 +94,8 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
await _async_migrate_unique_ids(hass, entry, gateway)
coordinator = SleepIQDataUpdateCoordinator(hass, gateway, email)
pause_coordinator = SleepIQPauseUpdateCoordinator(hass, gateway, email)
coordinator = SleepIQDataUpdateCoordinator(hass, entry, gateway)
pause_coordinator = SleepIQPauseUpdateCoordinator(hass, entry, gateway)
# Call the SleepIQ API to refresh data
await coordinator.async_config_entry_first_refresh()

View File

@ -7,6 +7,8 @@ import logging
from asyncsleepiq import AsyncSleepIQ
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import CONF_USERNAME
from homeassistant.core import HomeAssistant
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
@ -19,17 +21,20 @@ LONGER_UPDATE_INTERVAL = timedelta(minutes=5)
class SleepIQDataUpdateCoordinator(DataUpdateCoordinator[None]):
"""SleepIQ data update coordinator."""
config_entry: ConfigEntry
def __init__(
self,
hass: HomeAssistant,
config_entry: ConfigEntry,
client: AsyncSleepIQ,
username: str,
) -> None:
"""Initialize coordinator."""
super().__init__(
hass,
_LOGGER,
name=f"{username}@SleepIQ",
config_entry=config_entry,
name=f"{config_entry.data[CONF_USERNAME]}@SleepIQ",
update_interval=UPDATE_INTERVAL,
)
self.client = client
@ -45,17 +50,20 @@ class SleepIQDataUpdateCoordinator(DataUpdateCoordinator[None]):
class SleepIQPauseUpdateCoordinator(DataUpdateCoordinator[None]):
"""SleepIQ data update coordinator."""
config_entry: ConfigEntry
def __init__(
self,
hass: HomeAssistant,
config_entry: ConfigEntry,
client: AsyncSleepIQ,
username: str,
) -> None:
"""Initialize coordinator."""
super().__init__(
hass,
_LOGGER,
name=f"{username}@SleepIQPause",
config_entry=config_entry,
name=f"{config_entry.data[CONF_USERNAME]}@SleepIQPause",
update_interval=LONGER_UPDATE_INTERVAL,
)
self.client = client