Explicitly pass in the config_entry in version coordinator (#137877)

explicitly pass in the config_entry in coordinator

homeassistant/components/version/__init__.py

@@ -6,7 +6,6 @@ import logging
 
 from pyhaversion import HaVersion
 
-from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
 
@@ -18,12 +17,10 @@ from .const import (
     CONF_SOURCE,
     PLATFORMS,
 )
-from .coordinator import VersionDataUpdateCoordinator
+from .coordinator import VersionConfigEntry, VersionDataUpdateCoordinator
 
 _LOGGER = logging.getLogger(__name__)
 
-type VersionConfigEntry = ConfigEntry[VersionDataUpdateCoordinator]
-
 
 async def async_setup_entry(hass: HomeAssistant, entry: VersionConfigEntry) -> bool:
     """Set up the version integration from a config entry."""
@@ -40,6 +37,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: VersionConfigEntry) -> b
 
     coordinator = VersionDataUpdateCoordinator(
         hass=hass,
+        config_entry=entry,
         api=HaVersion(
             session=async_get_clientsession(hass),
             source=entry.data[CONF_SOURCE],

homeassistant/components/version/binary_sensor.py

@@ -13,8 +13,8 @@ from homeassistant.const import CONF_NAME, EntityCategory, __version__ as HA_VER
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from . import VersionConfigEntry
 from .const import CONF_SOURCE, DEFAULT_NAME
+from .coordinator import VersionConfigEntry
 from .entity import VersionEntity
 
 HA_VERSION_OBJECT = AwesomeVersion(HA_VERSION)

homeassistant/components/version/coordinator.py

@@ -14,21 +14,25 @@ from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, Upda
 
 from .const import DOMAIN, LOGGER, UPDATE_COORDINATOR_UPDATE_INTERVAL
 
+type VersionConfigEntry = ConfigEntry[VersionDataUpdateCoordinator]
+
 
 class VersionDataUpdateCoordinator(DataUpdateCoordinator[None]):
     """Data update coordinator for Version entities."""
 
-    config_entry: ConfigEntry
+    config_entry: VersionConfigEntry
 
     def __init__(
         self,
         hass: HomeAssistant,
+        config_entry: VersionConfigEntry,
         api: HaVersion,
     ) -> None:
         """Initialize the coordinator."""
         super().__init__(
             hass=hass,
             logger=LOGGER,
+            config_entry=config_entry,
             name=DOMAIN,
             update_interval=UPDATE_COORDINATOR_UPDATE_INTERVAL,
         )

homeassistant/components/version/diagnostics.py

@@ -9,7 +9,7 @@ from attr import asdict
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers import device_registry as dr, entity_registry as er
 
-from . import VersionConfigEntry
+from .coordinator import VersionConfigEntry
 
 
 async def async_get_config_entry_diagnostics(

homeassistant/components/version/sensor.py

@@ -10,8 +10,8 @@ from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.typing import StateType
 
-from . import VersionConfigEntry
 from .const import CONF_SOURCE, DEFAULT_NAME
+from .coordinator import VersionConfigEntry
 from .entity import VersionEntity