diff --git a/homeassistant/components/trafikverket_train/__init__.py b/homeassistant/components/trafikverket_train/__init__.py index b2945f7b62611219bdaa9bbd999bc33e31f57157..d09077dd01afe3944ea1a1e235044b89c2e172ce 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 49d4e1ded74f0f500e154ef76b60c6f127072d50..c4e1a41837140a2e4f3ac61dc6959b838d43219d 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."""