diff --git a/homeassistant/components/laundrify/__init__.py b/homeassistant/components/laundrify/__init__.py index b08624b6d23..7e3dd848348 100644 --- a/homeassistant/components/laundrify/__init__.py +++ b/homeassistant/components/laundrify/__init__.py @@ -13,7 +13,7 @@ from homeassistant.core import HomeAssistant from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady from homeassistant.helpers.aiohttp_client import async_get_clientsession -from .const import DEFAULT_POLL_INTERVAL, DOMAIN +from .const import DOMAIN from .coordinator import LaundrifyUpdateCoordinator _LOGGER = logging.getLogger(__name__) @@ -34,7 +34,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: except ApiConnectionException as err: raise ConfigEntryNotReady("Cannot reach laundrify API") from err - coordinator = LaundrifyUpdateCoordinator(hass, api_client, DEFAULT_POLL_INTERVAL) + coordinator = LaundrifyUpdateCoordinator(hass, entry, api_client) await coordinator.async_config_entry_first_refresh() diff --git a/homeassistant/components/laundrify/coordinator.py b/homeassistant/components/laundrify/coordinator.py index 22f68a7c5ae..928e30a9ed5 100644 --- a/homeassistant/components/laundrify/coordinator.py +++ b/homeassistant/components/laundrify/coordinator.py @@ -7,11 +7,12 @@ import logging from laundrify_aio import LaundrifyAPI, LaundrifyDevice from laundrify_aio.exceptions import ApiConnectionException, UnauthorizedException +from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.exceptions import ConfigEntryAuthFailed from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed -from .const import DOMAIN, REQUEST_TIMEOUT +from .const import DEFAULT_POLL_INTERVAL, DOMAIN, REQUEST_TIMEOUT _LOGGER = logging.getLogger(__name__) @@ -19,15 +20,21 @@ _LOGGER = logging.getLogger(__name__) class LaundrifyUpdateCoordinator(DataUpdateCoordinator[dict[str, LaundrifyDevice]]): """Class to manage fetching laundrify API data.""" + config_entry: ConfigEntry + def __init__( - self, hass: HomeAssistant, laundrify_api: LaundrifyAPI, poll_interval: int + self, + hass: HomeAssistant, + config_entry: ConfigEntry, + laundrify_api: LaundrifyAPI, ) -> None: """Initialize laundrify coordinator.""" super().__init__( hass, _LOGGER, + config_entry=config_entry, name=DOMAIN, - update_interval=timedelta(seconds=poll_interval), + update_interval=timedelta(seconds=DEFAULT_POLL_INTERVAL), ) self.laundrify_api = laundrify_api