From 12072b625c25f6e262306e940389780ff8f50d09 Mon Sep 17 00:00:00 2001
From: Michael <35783820+mib1185@users.noreply.github.com>
Date: Sat, 8 Feb 2025 21:28:48 +0100
Subject: [PATCH] Explicitly pass in the config_entry in solaredge coordinator
 (#137941)

explicitly pass in the config_entry in coordinator
---
 .../components/solaredge/coordinator.py       | 35 +++++++++++++++----
 homeassistant/components/solaredge/sensor.py  | 20 +++++++----
 2 files changed, 42 insertions(+), 13 deletions(-)

diff --git a/homeassistant/components/solaredge/coordinator.py b/homeassistant/components/solaredge/coordinator.py
index d37cf355fce..44f015eedeb 100644
--- a/homeassistant/components/solaredge/coordinator.py
+++ b/homeassistant/components/solaredge/coordinator.py
@@ -4,7 +4,7 @@ from __future__ import annotations
 
 from abc import ABC, abstractmethod
 from datetime import date, datetime, timedelta
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from aiosolaredge import SolarEdge
 from stringcase import snakecase
@@ -21,13 +21,22 @@ from .const import (
     POWER_FLOW_UPDATE_DELAY,
 )
 
+if TYPE_CHECKING:
+    from .types import SolarEdgeConfigEntry
+
 
 class SolarEdgeDataService(ABC):
     """Get and update the latest data."""
 
     coordinator: DataUpdateCoordinator[None]
 
-    def __init__(self, hass: HomeAssistant, api: SolarEdge, site_id: str) -> None:
+    def __init__(
+        self,
+        hass: HomeAssistant,
+        config_entry: SolarEdgeConfigEntry,
+        api: SolarEdge,
+        site_id: str,
+    ) -> None:
         """Initialize the data object."""
         self.api = api
         self.site_id = site_id
@@ -36,6 +45,7 @@ class SolarEdgeDataService(ABC):
         self.attributes: dict[str, Any] = {}
 
         self.hass = hass
+        self.config_entry = config_entry
 
     @callback
     def async_setup(self) -> None:
@@ -43,6 +53,7 @@ class SolarEdgeDataService(ABC):
         self.coordinator = DataUpdateCoordinator(
             self.hass,
             LOGGER,
+            config_entry=self.config_entry,
             name=str(self),
             update_method=self.async_update_data,
             update_interval=self.update_interval,
@@ -174,9 +185,15 @@ class SolarEdgeInventoryDataService(SolarEdgeDataService):
 class SolarEdgeEnergyDetailsService(SolarEdgeDataService):
     """Get and update the latest power flow data."""
 
-    def __init__(self, hass: HomeAssistant, api: SolarEdge, site_id: str) -> None:
+    def __init__(
+        self,
+        hass: HomeAssistant,
+        config_entry: SolarEdgeConfigEntry,
+        api: SolarEdge,
+        site_id: str,
+    ) -> None:
         """Initialize the power flow data service."""
-        super().__init__(hass, api, site_id)
+        super().__init__(hass, config_entry, api, site_id)
 
         self.unit = None
 
@@ -234,9 +251,15 @@ class SolarEdgeEnergyDetailsService(SolarEdgeDataService):
 class SolarEdgePowerFlowDataService(SolarEdgeDataService):
     """Get and update the latest power flow data."""
 
-    def __init__(self, hass: HomeAssistant, api: SolarEdge, site_id: str) -> None:
+    def __init__(
+        self,
+        hass: HomeAssistant,
+        config_entry: SolarEdgeConfigEntry,
+        api: SolarEdge,
+        site_id: str,
+    ) -> None:
         """Initialize the power flow data service."""
-        super().__init__(hass, api, site_id)
+        super().__init__(hass, config_entry, api, site_id)
 
         self.unit = None
 
diff --git a/homeassistant/components/solaredge/sensor.py b/homeassistant/components/solaredge/sensor.py
index 4b2398d15c2..004335b644b 100644
--- a/homeassistant/components/solaredge/sensor.py
+++ b/homeassistant/components/solaredge/sensor.py
@@ -206,7 +206,7 @@ async def async_setup_entry(
     """Add an solarEdge entry."""
     # Add the needed sensors to hass
     api = entry.runtime_data[DATA_API_CLIENT]
-    sensor_factory = SolarEdgeSensorFactory(hass, entry.data[CONF_SITE_ID], api)
+    sensor_factory = SolarEdgeSensorFactory(hass, entry, entry.data[CONF_SITE_ID], api)
     for service in sensor_factory.all_services:
         service.async_setup()
         await service.coordinator.async_refresh()
@@ -222,14 +222,20 @@ async def async_setup_entry(
 class SolarEdgeSensorFactory:
     """Factory which creates sensors based on the sensor_key."""
 
-    def __init__(self, hass: HomeAssistant, site_id: str, api: SolarEdge) -> None:
+    def __init__(
+        self,
+        hass: HomeAssistant,
+        config_entry: SolarEdgeConfigEntry,
+        site_id: str,
+        api: SolarEdge,
+    ) -> None:
         """Initialize the factory."""
 
-        details = SolarEdgeDetailsDataService(hass, api, site_id)
-        overview = SolarEdgeOverviewDataService(hass, api, site_id)
-        inventory = SolarEdgeInventoryDataService(hass, api, site_id)
-        flow = SolarEdgePowerFlowDataService(hass, api, site_id)
-        energy = SolarEdgeEnergyDetailsService(hass, api, site_id)
+        details = SolarEdgeDetailsDataService(hass, config_entry, api, site_id)
+        overview = SolarEdgeOverviewDataService(hass, config_entry, api, site_id)
+        inventory = SolarEdgeInventoryDataService(hass, config_entry, api, site_id)
+        flow = SolarEdgePowerFlowDataService(hass, config_entry, api, site_id)
+        energy = SolarEdgeEnergyDetailsService(hass, config_entry, api, site_id)
 
         self.all_services = (details, overview, inventory, flow, energy)