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

sensor.deutsche_bahn: add only_direct option (#11999)

Browse Source 
* sensor.deutsche_bahn: add only_direct option

This adds the `only_direct` option to the HA interface as provided by the schiene class.

* fix line length

as requested by @houndci-bot and @fabaff

* No line breaks needed
This commit is contained in:
akloeckner 2018-01-29 23:17:58 +01:00 committed by Fabian Affolter
parent ff0fd71608
commit 8b9dc71cde
1 changed files with 11 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -20,6 +20,8 @@ _LOGGER = logging.getLogger(__name__)
CONF_DESTINATION = 'to' CONF_DESTINATION = 'to'
CONF_START = 'from' CONF_START = 'from'
CONF_ONLY_DIRECT = 'only_direct'
DEFAULT_ONLY_DIRECT = False
ICON = 'mdi:train' ICON = 'mdi:train'
@ -28,6 +30,7 @@ SCAN_INTERVAL = timedelta(minutes=2)
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
vol.Required(CONF_DESTINATION): cv.string, vol.Required(CONF_DESTINATION): cv.string,
vol.Required(CONF_START): cv.string, vol.Required(CONF_START): cv.string,
vol.Optional(CONF_ONLY_DIRECT, default=DEFAULT_ONLY_DIRECT): cv.boolean,
}) })
@ -35,17 +38,18 @@ def setup_platform(hass, config, add_devices, discovery_info=None):
"""Set up the Deutsche Bahn Sensor.""" """Set up the Deutsche Bahn Sensor."""
start = config.get(CONF_START) start = config.get(CONF_START)
destination = config.get(CONF_DESTINATION) destination = config.get(CONF_DESTINATION)
only_direct = config.get(CONF_ONLY_DIRECT)
add_devices([DeutscheBahnSensor(start, destination)], True) add_devices([DeutscheBahnSensor(start, destination, only_direct)], True)
class DeutscheBahnSensor(Entity): class DeutscheBahnSensor(Entity):
"""Implementation of a Deutsche Bahn sensor.""" """Implementation of a Deutsche Bahn sensor."""
def __init__(self, start, goal): def __init__(self, start, goal, only_direct):
"""Initialize the sensor.""" """Initialize the sensor."""
self._name = '{} to {}'.format(start, goal) self._name = '{} to {}'.format(start, goal)
self.data = SchieneData(start, goal) self.data = SchieneData(start, goal, only_direct)
self._state = None self._state = None
@property @property
@ -82,19 +86,21 @@ class DeutscheBahnSensor(Entity):
class SchieneData(object): class SchieneData(object):
"""Pull data from the bahn.de web page.""" """Pull data from the bahn.de web page."""
def __init__(self, start, goal): def __init__(self, start, goal, only_direct):
"""Initialize the sensor.""" """Initialize the sensor."""
import schiene import schiene
self.start = start self.start = start
self.goal = goal self.goal = goal
self.only_direct = only_direct
self.schiene = schiene.Schiene() self.schiene = schiene.Schiene()
self.connections = [{}] self.connections = [{}]
def update(self): def update(self):
"""Update the connection data.""" """Update the connection data."""
self.connections = self.schiene.connections( self.connections = self.schiene.connections(
self.start, self.goal, dt_util.as_local(dt_util.utcnow())) self.start, self.goal, dt_util.as_local(dt_util.utcnow()),
self.only_direct)
for con in self.connections: for con in self.connections:
# Detail info is not useful. Having a more consistent interface # Detail info is not useful. Having a more consistent interface