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

Add the two next trains (#2390)

Browse Source
This commit is contained in:
Fabian Affolter 2016-06-30 02:44:35 +02:00 committed by Paulus Schoutsen
parent 5cce02ab62
commit 8dd7ebb08e
1 changed files with 34 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

65
homeassistant/components/sensor/deutsche_bahn.py
View File

@ -1,38 +1,43 @@
""" """
Support for information about the German trans system. Support for information about the German train system.
For more details about this platform, please refer to the documentation at For more details about this platform, please refer to the documentation at
https://home-assistant.io/components/sensor.deutsche_bahn/ https://home-assistant.io/components/sensor.deutsche_bahn/
""" """
import logging import logging
from datetime import timedelta, datetime from datetime import timedelta
import voluptuous as vol
from homeassistant.const import (CONF_PLATFORM)
import homeassistant.helpers.config_validation as cv
from homeassistant.util import Throttle from homeassistant.util import Throttle
from homeassistant.helpers.entity import Entity from homeassistant.helpers.entity import Entity
_LOGGER = logging.getLogger(__name__)
REQUIREMENTS = ['schiene==0.17'] REQUIREMENTS = ['schiene==0.17']
CONF_START = 'from'
CONF_DESTINATION = 'to'
ICON = 'mdi:train' ICON = 'mdi:train'
_LOGGER = logging.getLogger(__name__)
PLATFORM_SCHEMA = vol.Schema({
vol.Required(CONF_PLATFORM): 'deutsche_bahn',
vol.Required(CONF_START): cv.string,
vol.Required(CONF_DESTINATION): cv.string,
})
# Return cached results if last scan was less then this time ago. # Return cached results if last scan was less then this time ago.
MIN_TIME_BETWEEN_UPDATES = timedelta(seconds=120) MIN_TIME_BETWEEN_UPDATES = timedelta(seconds=120)
def setup_platform(hass, config, add_devices_callback, discovery_info=None): def setup_platform(hass, config, add_devices, discovery_info=None):
"""Setup the Deutsche Bahn Sensor.""" """Setup the Deutsche Bahn Sensor."""
start = config.get('from') start = config.get(CONF_START)
goal = config.get('to') destination = config.get(CONF_DESTINATION)
if start is None: add_devices([DeutscheBahnSensor(start, destination)])
_LOGGER.error('Missing required variable: "from"')
return False
if goal is None:
_LOGGER.error('Missing required variable: "to"')
return False
dev = []
dev.append(DeutscheBahnSensor(start, goal))
add_devices_callback(dev)
# pylint: disable=too-few-public-methods # pylint: disable=too-few-public-methods
@ -63,16 +68,17 @@ class DeutscheBahnSensor(Entity):
@property @property
def state_attributes(self): def state_attributes(self):
"""Return the state attributes.""" """Return the state attributes."""
return self.data.connections[0] connections = self.data.connections[0]
connections['next'] = self.data.connections[1]['departure']
connections['next_on'] = self.data.connections[2]['departure']
return connections
def update(self): def update(self):
"""Get the latest delay from bahn.de and updates the state.""" """Get the latest delay from bahn.de and updates the state."""
self.data.update() self.data.update()
self._state = self.data.connections[0].get('departure', 'Unknown') self._state = self.data.connections[0].get('departure', 'Unknown')
if self.data.connections[0]['delay'] != 0: if self.data.connections[0]['delay'] != 0:
self._state += " + {}".format( self._state += " + {}".format(self.data.connections[0]['delay'])
self.data.connections[0]['delay']
)
# pylint: disable=too-few-public-methods # pylint: disable=too-few-public-methods
@ -90,18 +96,15 @@ class SchieneData(object):
@Throttle(MIN_TIME_BETWEEN_UPDATES) @Throttle(MIN_TIME_BETWEEN_UPDATES)
def update(self): def update(self):
"""Update the connection data.""" """Update the connection data."""
self.connections = self.schiene.connections(self.start, self.connections = self.schiene.connections(self.start, self.goal)
self.goal,
datetime.now())
for con in self.connections: for con in self.connections:
# Details info is not useful. # Detail info is not useful. Having a more consistent interface
# Having a more consistent interface simplifies # simplifies usage of template sensors.
# usage of Template sensors later on
if 'details' in con: if 'details' in con:
con.pop('details') con.pop('details')
delay = con.get('delay', delay = con.get('delay', {'delay_departure': 0,
{'delay_departure': 0, 'delay_arrival': 0})
'delay_arrival': 0}) # IMHO only delay_departure is useful
# IMHO only delay_departure is usefull
con['delay'] = delay['delay_departure'] con['delay'] = delay['delay_departure']
con['ontime'] = con.get('ontime', False) con['ontime'] = con.get('ontime', False)