From 48909539bea94ca9866d587e141100f9e632e7d2 Mon Sep 17 00:00:00 2001 From: Robin Date: Wed, 23 Aug 2017 08:05:06 +0100 Subject: [PATCH] Fix issue 8894 with uk_transport component if no next_buses or next_trains (#9046) * Fix bug if no next_buses or trains Fixes https://github.com/home-assistant/home-assistant/issues/8894 * Requested fixes --- .../components/sensor/uk_transport.py | 19 ++++++++++++------- 1 file changed, 12 insertions(+), 7 deletions(-) diff --git a/homeassistant/components/sensor/uk_transport.py b/homeassistant/components/sensor/uk_transport.py index b9ce98ec257..bcac4b47279 100644 --- a/homeassistant/components/sensor/uk_transport.py +++ b/homeassistant/components/sensor/uk_transport.py @@ -180,9 +180,12 @@ class UkTransportLiveBusTimeSensor(UkTransportSensor): 'estimated': departure['best_departure_estimate'] }) - self._state = min(map( - _delta_mins, [bus['scheduled'] for bus in self._next_buses] - )) + if self._next_buses: + self._state = min( + _delta_mins(bus['scheduled']) + for bus in self._next_buses) + else: + self._state = None @property def device_state_attributes(self): @@ -242,10 +245,12 @@ class UkTransportLiveTrainTimeSensor(UkTransportSensor): 'operator_name': departure['operator_name'] }) - self._state = min(map( - _delta_mins, - [train['scheduled'] for train in self._next_trains] - )) + if self._next_trains: + self._state = min( + _delta_mins(train['scheduled']) + for train in self._next_trains) + else: + self._state = None @property def device_state_attributes(self):