jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-04-23 16:57:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

Explicitly set config entry in Trafikverket Train coordinator (#134304)

Browse Source
This commit is contained in:
G Johansson 2024-12-30 23:18:22 +01:00 committed by GitHub
parent a9949a0aab
commit 6266a4153d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 7 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
homeassistant/components/trafikverket_train/__init__.py
View File

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

13
homeassistant/components/trafikverket_train/coordinator.py
View File

@ -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."""