From 2418ef8e8e39a2e2c915d6657fa237ad05fdace0 Mon Sep 17 00:00:00 2001
From: Michael <35783820+mib1185@users.noreply.github.com>
Date: Sun, 9 Feb 2025 15:45:51 +0100
Subject: [PATCH] Explicitly pass in the config_entry in refoss coordinator
 (#137978)

explicitly pass in the config_entry in coordinator
---
 homeassistant/components/refoss/__init__.py    | 2 +-
 homeassistant/components/refoss/bridge.py      | 8 ++++++--
 homeassistant/components/refoss/coordinator.py | 8 +++++++-
 3 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/homeassistant/components/refoss/__init__.py b/homeassistant/components/refoss/__init__.py
index 0f0c852b043..eb2085efda4 100644
--- a/homeassistant/components/refoss/__init__.py
+++ b/homeassistant/components/refoss/__init__.py
@@ -24,7 +24,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Set up Refoss from a config entry."""
     hass.data.setdefault(DOMAIN, {})
     discover = await refoss_discovery_server(hass)
-    refoss_discovery = DiscoveryService(hass, discover)
+    refoss_discovery = DiscoveryService(hass, entry, discover)
     hass.data[DOMAIN][DATA_DISCOVERY_SERVICE] = refoss_discovery
 
     await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
diff --git a/homeassistant/components/refoss/bridge.py b/homeassistant/components/refoss/bridge.py
index 11e92620fbb..a3ba9ea663d 100644
--- a/homeassistant/components/refoss/bridge.py
+++ b/homeassistant/components/refoss/bridge.py
@@ -6,6 +6,7 @@ from refoss_ha.device import DeviceInfo
 from refoss_ha.device_manager import async_build_base_device
 from refoss_ha.discovery import Discovery, Listener
 
+from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.dispatcher import async_dispatcher_send
 
@@ -16,9 +17,12 @@ from .coordinator import RefossDataUpdateCoordinator
 class DiscoveryService(Listener):
     """Discovery event handler for refoss devices."""
 
-    def __init__(self, hass: HomeAssistant, discovery: Discovery) -> None:
+    def __init__(
+        self, hass: HomeAssistant, config_entry: ConfigEntry, discovery: Discovery
+    ) -> None:
         """Init discovery service."""
         self.hass = hass
+        self.config_entry = config_entry
 
         self.discovery = discovery
         self.discovery.add_listener(self)
@@ -32,7 +36,7 @@ class DiscoveryService(Listener):
         if device is None:
             return
 
-        coordo = RefossDataUpdateCoordinator(self.hass, device)
+        coordo = RefossDataUpdateCoordinator(self.hass, self.config_entry, device)
         self.hass.data[DOMAIN][COORDINATORS].append(coordo)
         await coordo.async_refresh()
 
diff --git a/homeassistant/components/refoss/coordinator.py b/homeassistant/components/refoss/coordinator.py
index 929d1b3962b..381f64614b5 100644
--- a/homeassistant/components/refoss/coordinator.py
+++ b/homeassistant/components/refoss/coordinator.py
@@ -7,6 +7,7 @@ from datetime import timedelta
 from refoss_ha.controller.device import BaseDevice
 from refoss_ha.exceptions import DeviceTimeoutError
 
+from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
 
@@ -16,11 +17,16 @@ from .const import _LOGGER, DOMAIN, MAX_ERRORS
 class RefossDataUpdateCoordinator(DataUpdateCoordinator[None]):
     """Manages polling for state changes from the device."""
 
-    def __init__(self, hass: HomeAssistant, device: BaseDevice) -> None:
+    config_entry: ConfigEntry
+
+    def __init__(
+        self, hass: HomeAssistant, config_entry: ConfigEntry, device: BaseDevice
+    ) -> None:
         """Initialize the data update coordinator."""
         super().__init__(
             hass,
             _LOGGER,
+            config_entry=config_entry,
             name=f"{DOMAIN}-{device.device_info.dev_name}",
             update_interval=timedelta(seconds=15),
         )