From 6266a4153d28ed9feb91a7a8c307b639f7654030 Mon Sep 17 00:00:00 2001
From: G Johansson <goran.johansson@shiftit.se>
Date: Mon, 30 Dec 2024 23:18:22 +0100
Subject: [PATCH] Explicitly set config entry in Trafikverket Train coordinator
(#134304)
---
.../components/trafikverket_train/__init__.py | 2 +-
.../components/trafikverket_train/coordinator.py | 13 ++++++-------
2 files changed, 7 insertions(+), 8 deletions(-)
diff --git a/homeassistant/components/trafikverket_train/__init__.py b/homeassistant/components/trafikverket_train/__init__.py
index b2945f7b626..d09077dd01a 100644
--- a/homeassistant/components/trafikverket_train/__init__.py
+++ b/homeassistant/components/trafikverket_train/__init__.py
@@ -19,7 +19,7 @@ _LOGGER = logging.getLogger(__name__)
async def async_setup_entry(hass: HomeAssistant, entry: TVTrainConfigEntry) -> bool:
"""Set up Trafikverket Train from a config entry."""
- coordinator = TVDataUpdateCoordinator(hass)
+ coordinator = TVDataUpdateCoordinator(hass, entry)
await coordinator.async_config_entry_first_refresh()
entry.runtime_data = coordinator
diff --git a/homeassistant/components/trafikverket_train/coordinator.py b/homeassistant/components/trafikverket_train/coordinator.py
index 49d4e1ded74..c4e1a418371 100644
--- a/homeassistant/components/trafikverket_train/coordinator.py
+++ b/homeassistant/components/trafikverket_train/coordinator.py
@@ -74,22 +74,21 @@ class TVDataUpdateCoordinator(DataUpdateCoordinator[TrainData]):
from_station: StationInfoModel
to_station: StationInfoModel
- def __init__(self, hass: HomeAssistant) -> None:
+ def __init__(self, hass: HomeAssistant, config_entry: TVTrainConfigEntry) -> None:
"""Initialize the Trafikverket coordinator."""
super().__init__(
hass,
_LOGGER,
+ config_entry=config_entry,
name=DOMAIN,
update_interval=TIME_BETWEEN_UPDATES,
)
self._train_api = TrafikverketTrain(
- async_get_clientsession(hass), self.config_entry.data[CONF_API_KEY]
- )
- self._time: time | None = dt_util.parse_time(self.config_entry.data[CONF_TIME])
- self._weekdays: list[str] = self.config_entry.data[CONF_WEEKDAY]
- self._filter_product: str | None = self.config_entry.options.get(
- CONF_FILTER_PRODUCT
+ async_get_clientsession(hass), config_entry.data[CONF_API_KEY]
)
+ self._time: time | None = dt_util.parse_time(config_entry.data[CONF_TIME])
+ self._weekdays: list[str] = config_entry.data[CONF_WEEKDAY]
+ self._filter_product: str | None = config_entry.options.get(CONF_FILTER_PRODUCT)
async def _async_setup(self) -> None:
"""Initiate stations."""
--
GitLab