Explicitly pass in the config_entry in tautulli coordinator (#137911)

explicitly pass in the config_entry in coordinator

homeassistant/components/tautulli/__init__.py

@@ -4,15 +4,13 @@ from __future__ import annotations
 
 from pytautulli import PyTautulli, PyTautulliHostConfiguration
 
-from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_API_KEY, CONF_URL, CONF_VERIFY_SSL, Platform
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.aiohttp_client import async_get_clientsession
 
-from .coordinator import TautulliDataUpdateCoordinator
+from .coordinator import TautulliConfigEntry, TautulliDataUpdateCoordinator
 
 PLATFORMS = [Platform.SENSOR]
-type TautulliConfigEntry = ConfigEntry[TautulliDataUpdateCoordinator]
 
 
 async def async_setup_entry(hass: HomeAssistant, entry: TautulliConfigEntry) -> bool:
@@ -27,7 +25,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: TautulliConfigEntry) ->
         session=async_get_clientsession(hass, entry.data[CONF_VERIFY_SSL]),
     )
     entry.runtime_data = TautulliDataUpdateCoordinator(
-        hass, host_configuration, api_client
+        hass, entry, host_configuration, api_client
     )
     await entry.runtime_data.async_config_entry_first_refresh()
     await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)

homeassistant/components/tautulli/coordinator.py

@@ -4,7 +4,6 @@ from __future__ import annotations
 
 import asyncio
 from datetime import timedelta
-from typing import TYPE_CHECKING
 
 from pytautulli import (
     PyTautulli,
@@ -18,14 +17,14 @@ from pytautulli.exceptions import (
 )
 from pytautulli.models.host_configuration import PyTautulliHostConfiguration
 
+from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import ConfigEntryAuthFailed
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
 
 from .const import DOMAIN, LOGGER
 
-if TYPE_CHECKING:
-    from . import TautulliConfigEntry
+type TautulliConfigEntry = ConfigEntry[TautulliDataUpdateCoordinator]
 
 
 class TautulliDataUpdateCoordinator(DataUpdateCoordinator[None]):
@@ -36,6 +35,7 @@ class TautulliDataUpdateCoordinator(DataUpdateCoordinator[None]):
     def __init__(
         self,
         hass: HomeAssistant,
+        config_entry: TautulliConfigEntry,
         host_configuration: PyTautulliHostConfiguration,
         api_client: PyTautulli,
     ) -> None:
@@ -43,6 +43,7 @@ class TautulliDataUpdateCoordinator(DataUpdateCoordinator[None]):
         super().__init__(
             hass=hass,
             logger=LOGGER,
+            config_entry=config_entry,
             name=DOMAIN,
             update_interval=timedelta(seconds=10),
         )

homeassistant/components/tautulli/sensor.py

@@ -26,9 +26,8 @@ from homeassistant.helpers.entity import EntityDescription
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType, StateType
 
-from . import TautulliConfigEntry
 from .const import ATTR_TOP_USER, DOMAIN
-from .coordinator import TautulliDataUpdateCoordinator
+from .coordinator import TautulliConfigEntry, TautulliDataUpdateCoordinator
 from .entity import TautulliEntity