From c125c4af4f4044d6bae4d0ca9c79125747a1db28 Mon Sep 17 00:00:00 2001 From: Hugo Dupras Date: Fri, 16 Dec 2016 06:40:33 +0100 Subject: [PATCH] Fix for GTFS sensor (#4828) * Fix for GTFS sensor Signed-off-by: Hugo D. (jabesq) * GTFS fix Signed-off-by: Hugo D. (jabesq) --- homeassistant/components/sensor/gtfs.py | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/homeassistant/components/sensor/gtfs.py b/homeassistant/components/sensor/gtfs.py index e76b8ed07ed..5769860284c 100644 --- a/homeassistant/components/sensor/gtfs.py +++ b/homeassistant/components/sensor/gtfs.py @@ -95,6 +95,9 @@ def get_next_departure(sched, start_station_id, end_station_id): for row in result: item = row + if item == {}: + return None + today = datetime.datetime.today().strftime('%Y-%m-%d') departure_time_string = '{} {}'.format(today, item[2]) arrival_time_string = '{} {}'.format(today, item[3]) @@ -221,6 +224,13 @@ class GTFSDepartureSensor(Entity): with self.lock: self._departure = get_next_departure(self._pygtfs, self.origin, self.destination) + if not self._departure: + self._state = 0 + self._attributes = {'Info': 'No more bus today'} + if self._name == '': + self._name = (self._custom_name or "GTFS Sensor") + return + self._state = self._departure['minutes_until_departure'] origin_station = self._departure['origin_station']