From d9a17506f54718de304cc6eb2d093392e617d211 Mon Sep 17 00:00:00 2001 From: Michael <35783820+mib1185@users.noreply.github.com> Date: Sun, 9 Feb 2025 21:16:05 +0100 Subject: [PATCH] Explicitly pass in the config_entry in here_travel_time coordinator (#138155) explicitly pass in the config_entry in coordinator --- homeassistant/components/here_travel_time/__init__.py | 2 +- homeassistant/components/here_travel_time/coordinator.py | 9 +++++++++ 2 files changed, 10 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/here_travel_time/__init__.py b/homeassistant/components/here_travel_time/__init__.py index 1b99ba64827..132b12de4ce 100644 --- a/homeassistant/components/here_travel_time/__init__.py +++ b/homeassistant/components/here_travel_time/__init__.py @@ -56,7 +56,7 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b else: cls = HERERoutingDataUpdateCoordinator - data_coordinator = cls(hass, api_key, here_travel_time_config) + data_coordinator = cls(hass, config_entry, api_key, here_travel_time_config) hass.data.setdefault(DOMAIN, {})[config_entry.entry_id] = data_coordinator async def _async_update_at_start(_: HomeAssistant) -> None: diff --git a/homeassistant/components/here_travel_time/coordinator.py b/homeassistant/components/here_travel_time/coordinator.py index 65e1305e44e..a3345e78e4e 100644 --- a/homeassistant/components/here_travel_time/coordinator.py +++ b/homeassistant/components/here_travel_time/coordinator.py @@ -25,6 +25,7 @@ from here_transit import ( ) import voluptuous as vol +from homeassistant.config_entries import ConfigEntry from homeassistant.const import UnitOfLength from homeassistant.core import HomeAssistant from homeassistant.helpers import config_validation as cv @@ -44,9 +45,12 @@ _LOGGER = logging.getLogger(__name__) class HERERoutingDataUpdateCoordinator(DataUpdateCoordinator[HERETravelTimeData]): """here_routing DataUpdateCoordinator.""" + config_entry: ConfigEntry + def __init__( self, hass: HomeAssistant, + config_entry: ConfigEntry, api_key: str, config: HERETravelTimeConfig, ) -> None: @@ -54,6 +58,7 @@ class HERERoutingDataUpdateCoordinator(DataUpdateCoordinator[HERETravelTimeData] super().__init__( hass, _LOGGER, + config_entry=config_entry, name=DOMAIN, update_interval=timedelta(seconds=DEFAULT_SCAN_INTERVAL), ) @@ -168,9 +173,12 @@ class HERETransitDataUpdateCoordinator( ): """HERETravelTime DataUpdateCoordinator.""" + config_entry: ConfigEntry + def __init__( self, hass: HomeAssistant, + config_entry: ConfigEntry, api_key: str, config: HERETravelTimeConfig, ) -> None: @@ -178,6 +186,7 @@ class HERETransitDataUpdateCoordinator( super().__init__( hass, _LOGGER, + config_entry=config_entry, name=DOMAIN, update_interval=timedelta(seconds=DEFAULT_SCAN_INTERVAL), )