Mitigate NMBS key errors (#50026)

on Liveboard and connections as documented in issue #48824
This commit is contained in:
Marius 2021-05-03 17:06:46 +02:00 committed by GitHub
parent efb1bb08a4
commit 13dee0f028
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -152,7 +152,7 @@ class NMBSLiveBoard(SensorEntity):
"""Set the state equal to the next departure."""
liveboard = self._api_client.get_liveboard(self._station)
if liveboard is None or not liveboard["departures"]:
if liveboard is None or not liveboard.get("departures"):
return
next_departure = liveboard["departures"]["departure"][0]
@ -269,7 +269,7 @@ class NMBSSensor(SensorEntity):
self._station_from, self._station_to
)
if connections is None or not connections["connection"]:
if connections is None or not connections.get("connection"):
return
if int(connections["connection"][0]["departure"]["left"]) > 0: