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

Set GTFS icon by route type (#20876)

Browse Source
This commit is contained in:
René-Marc Simard 2019-02-08 22:38:39 -05:00 committed by Andrew Sayre
parent d16d14b648
commit 9db9a81793
1 changed files with 14 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
homeassistant/components/sensor/gtfs.py
View File

@ -29,6 +29,16 @@ DEFAULT_NAME = 'GTFS Sensor'
DEFAULT_PATH = 'gtfs' DEFAULT_PATH = 'gtfs'
ICON = 'mdi:train' ICON = 'mdi:train'
ICONS = {
0: 'mdi:tram',
1: 'mdi:subway',
2: 'mdi:train',
3: 'mdi:bus',
4: 'mdi:ferry',
5: 'mdi:train-variant',
6: 'mdi:gondola',
7: 'mdi:stairs',
}
TIME_FORMAT = '%Y-%m-%d %H:%M:%S' TIME_FORMAT = '%Y-%m-%d %H:%M:%S'
@ -193,6 +203,7 @@ class GTFSDepartureSensor(Entity):
self.destination = destination self.destination = destination
self._offset = offset self._offset = offset
self._custom_name = name self._custom_name = name
self._icon = ICON
self._name = '' self._name = ''
self._unit_of_measurement = 'min' self._unit_of_measurement = 'min'
self._state = 0 self._state = 0
@ -223,7 +234,7 @@ class GTFSDepartureSensor(Entity):
@property @property
def icon(self): def icon(self):
"""Icon to use in the frontend, if any.""" """Icon to use in the frontend, if any."""
return ICON return self._icon
def update(self): def update(self):
"""Get the latest data from GTFS and update the states.""" """Get the latest data from GTFS and update the states."""
@ -257,6 +268,8 @@ class GTFSDepartureSensor(Entity):
self._attributes = {} self._attributes = {}
self._attributes['offset'] = self._offset.seconds / 60 self._attributes['offset'] = self._offset.seconds / 60
self._icon = ICONS.get(route.route_type, ICON)
def dict_for_table(resource): def dict_for_table(resource):
"""Return a dict for the SQLAlchemy resource given.""" """Return a dict for the SQLAlchemy resource given."""
return dict((col, getattr(resource, col)) return dict((col, getattr(resource, col))