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
This commit is contained in:
Robin 2017-08-23 08:05:06 +01:00 committed by Paulus Schoutsen
parent 2355216f61
commit 48909539be

View file

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