From f8ac48fc7896c669d07e9ed6064ab336a37a594c Mon Sep 17 00:00:00 2001 From: Michael <35783820+mib1185@users.noreply.github.com> Date: Sat, 8 Feb 2025 13:15:37 +0100 Subject: [PATCH] Explicitly pass in the config_entry in flo coordinator (#137819) explicitly pass in the config_entry in coordinator --- homeassistant/components/flo/__init__.py | 4 +++- homeassistant/components/flo/coordinator.py | 10 +++++++++- 2 files changed, 12 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/flo/__init__.py b/homeassistant/components/flo/__init__.py index b619df91d59..6a497f5140d 100644 --- a/homeassistant/components/flo/__init__.py +++ b/homeassistant/components/flo/__init__.py @@ -37,7 +37,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: _LOGGER.debug("Flo user information with locations: %s", user_info) hass.data[DOMAIN][entry.entry_id]["devices"] = devices = [ - FloDeviceDataUpdateCoordinator(hass, client, location["id"], device["id"]) + FloDeviceDataUpdateCoordinator( + hass, entry, client, location["id"], device["id"] + ) for location in user_info["locations"] for device in location["devices"] ] diff --git a/homeassistant/components/flo/coordinator.py b/homeassistant/components/flo/coordinator.py index d0dd38bd490..f5dc34a50cd 100644 --- a/homeassistant/components/flo/coordinator.py +++ b/homeassistant/components/flo/coordinator.py @@ -10,6 +10,7 @@ from aioflo.api import API from aioflo.errors import RequestError from orjson import JSONDecodeError +from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed from homeassistant.util import dt as dt_util @@ -20,10 +21,16 @@ from .const import DOMAIN as FLO_DOMAIN, LOGGER class FloDeviceDataUpdateCoordinator(DataUpdateCoordinator): """Flo device object.""" + config_entry: ConfigEntry _failure_count: int = 0 def __init__( - self, hass: HomeAssistant, api_client: API, location_id: str, device_id: str + self, + hass: HomeAssistant, + config_entry: ConfigEntry, + api_client: API, + location_id: str, + device_id: str, ) -> None: """Initialize the device.""" self.hass: HomeAssistant = hass @@ -36,6 +43,7 @@ class FloDeviceDataUpdateCoordinator(DataUpdateCoordinator): super().__init__( hass, LOGGER, + config_entry=config_entry, name=f"{FLO_DOMAIN}-{device_id}", update_interval=timedelta(seconds=60), )