jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 19:27:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix Trafivkerket Train coordinator exceptions (#97919)

Browse Source 
Fix tvt coordinator exceptions
This commit is contained in:
G Johansson 2023-08-06 19:07:18 +02:00 committed by GitHub
parent 163bbe2c5d
commit 6bc5b8989d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -6,6 +6,12 @@ from datetime import date, datetime, time, timedelta
import logging import logging
from pytrafikverket import TrafikverketTrain from pytrafikverket import TrafikverketTrain
from pytrafikverket.exceptions import (
InvalidAuthentication,
MultipleTrainAnnouncementFound,
NoTrainAnnouncementFound,
UnknownError,
)
from pytrafikverket.trafikverket_train import StationInfo, TrainStop from pytrafikverket.trafikverket_train import StationInfo, TrainStop
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
@ -119,9 +125,13 @@ class TVDataUpdateCoordinator(DataUpdateCoordinator[TrainData]):
state = await self._train_api.async_get_next_train_stop( state = await self._train_api.async_get_next_train_stop(
self.from_station, self.to_station, when self.from_station, self.to_station, when
) )
except ValueError as error: except InvalidAuthentication as error:
if "Invalid authentication" in error.args[0]: raise ConfigEntryAuthFailed from error
raise ConfigEntryAuthFailed from error except (
NoTrainAnnouncementFound,
MultipleTrainAnnouncementFound,
UnknownError,
) as error:
raise UpdateFailed( raise UpdateFailed(
f"Train departure {when} encountered a problem: {error}" f"Train departure {when} encountered a problem: {error}"
) from error ) from error