Explicitly pass in the config_entry in here_travel_time coordinator (#138155)

explicitly pass in the config_entry in coordinator
This commit is contained in:
Michael 2025-02-09 21:16:05 +01:00 committed by GitHub
parent 7c9d30eb06
commit d9a17506f5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 10 additions and 1 deletions

View File

@ -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:

View File

@ -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),
)