From bd3eec90ba609d3a6e45bdf0694088e43a2d0f37 Mon Sep 17 00:00:00 2001 From: Michael <35783820+mib1185@users.noreply.github.com> Date: Sun, 9 Feb 2025 17:33:38 +0100 Subject: [PATCH] Explicitly pass in the config_entry in microbees coordinator (#138090) explicitly pass in the config_entry in coordinator --- homeassistant/components/microbees/__init__.py | 2 +- homeassistant/components/microbees/coordinator.py | 8 +++++++- 2 files changed, 8 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/microbees/__init__.py b/homeassistant/components/microbees/__init__.py index 488988ab593..12c536121da 100644 --- a/homeassistant/components/microbees/__init__.py +++ b/homeassistant/components/microbees/__init__.py @@ -45,7 +45,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: raise ConfigEntryAuthFailed("Token not valid, trigger renewal") from ex raise ConfigEntryNotReady from ex microbees = MicroBees(token=session.token[CONF_ACCESS_TOKEN]) - coordinator = MicroBeesUpdateCoordinator(hass, microbees) + coordinator = MicroBeesUpdateCoordinator(hass, entry, microbees) await coordinator.async_config_entry_first_refresh() hass.data.setdefault(DOMAIN, {})[entry.entry_id] = HomeAssistantMicroBeesData( connector=microbees, diff --git a/homeassistant/components/microbees/coordinator.py b/homeassistant/components/microbees/coordinator.py index af207507e77..0094dc33e81 100644 --- a/homeassistant/components/microbees/coordinator.py +++ b/homeassistant/components/microbees/coordinator.py @@ -9,6 +9,7 @@ import logging import aiohttp from microBeesPy import Actuator, Bee, MicroBees, MicroBeesException, Sensor +from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.exceptions import ConfigEntryAuthFailed from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed @@ -28,11 +29,16 @@ class MicroBeesCoordinatorData: class MicroBeesUpdateCoordinator(DataUpdateCoordinator[MicroBeesCoordinatorData]): """MicroBees coordinator.""" - def __init__(self, hass: HomeAssistant, microbees: MicroBees) -> None: + config_entry: ConfigEntry + + def __init__( + self, hass: HomeAssistant, config_entry: ConfigEntry, microbees: MicroBees + ) -> None: """Initialize microBees coordinator.""" super().__init__( hass, _LOGGER, + config_entry=config_entry, name="microBees Coordinator", update_interval=timedelta(seconds=30), )