Explicitly pass in the config_entry in pure_energie coordinator (#138035)

explicitly pass in the config_entry in coordinator

homeassistant/components/pure_energie/__init__.py

@@ -2,22 +2,19 @@
 
 from __future__ import annotations
 
-from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import Platform
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import ConfigEntryNotReady
 
-from .coordinator import PureEnergieDataUpdateCoordinator
+from .coordinator import PureEnergieConfigEntry, PureEnergieDataUpdateCoordinator
 
 PLATFORMS: list[Platform] = [Platform.SENSOR]
 
-type PureEnergieConfigEntry = ConfigEntry[PureEnergieDataUpdateCoordinator]
-
 
 async def async_setup_entry(hass: HomeAssistant, entry: PureEnergieConfigEntry) -> bool:
     """Set up Pure Energie from a config entry."""
 
-    coordinator = PureEnergieDataUpdateCoordinator(hass)
+    coordinator = PureEnergieDataUpdateCoordinator(hass, entry)
     try:
         await coordinator.async_config_entry_first_refresh()
     except ConfigEntryNotReady:

homeassistant/components/pure_energie/coordinator.py

@@ -14,6 +14,8 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
 
 from .const import DOMAIN, LOGGER, SCAN_INTERVAL
 
+type PureEnergieConfigEntry = ConfigEntry[PureEnergieDataUpdateCoordinator]
+
 
 class PureEnergieData(NamedTuple):
     """Class for defining data in dict."""
@@ -25,16 +27,18 @@ class PureEnergieData(NamedTuple):
 class PureEnergieDataUpdateCoordinator(DataUpdateCoordinator[PureEnergieData]):
     """Class to manage fetching Pure Energie data from single eindpoint."""
 
-    config_entry: ConfigEntry
+    config_entry: PureEnergieConfigEntry
 
     def __init__(
         self,
         hass: HomeAssistant,
+        config_entry: PureEnergieConfigEntry,
     ) -> None:
         """Initialize global Pure Energie data updater."""
         super().__init__(
             hass,
             LOGGER,
+            config_entry=config_entry,
             name=DOMAIN,
             update_interval=SCAN_INTERVAL,
         )

homeassistant/components/pure_energie/diagnostics.py

@@ -9,7 +9,7 @@ from homeassistant.components.diagnostics import async_redact_data
 from homeassistant.const import CONF_HOST
 from homeassistant.core import HomeAssistant
 
-from . import PureEnergieConfigEntry
+from .coordinator import PureEnergieConfigEntry
 
 TO_REDACT = {
     CONF_HOST,

homeassistant/components/pure_energie/sensor.py

@@ -18,9 +18,12 @@ from homeassistant.helpers.device_registry import DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
-from . import PureEnergieConfigEntry
 from .const import DOMAIN
-from .coordinator import PureEnergieData, PureEnergieDataUpdateCoordinator
+from .coordinator import (
+    PureEnergieConfigEntry,
+    PureEnergieData,
+    PureEnergieDataUpdateCoordinator,
+)
 
 
 @dataclass(frozen=True, kw_only=True)