Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Improvement to allow parsing of station ID in vasttrafik integration. Addresses #34851 (#43136)

pull/43304/head
Thomas 2020-11-17 07:34:20 +01:00 committed by GitHub
parent 95504b7408
commit 4e6035d057
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
homeassistant/components/vasttrafik/sensor.py
View File

@ -80,14 +80,23 @@ class VasttrafikDepartureSensor(Entity):
"""Initialize the sensor.""" """Initialize the sensor."""
self._planner = planner self._planner = planner
self._name = name or departure self._name = name or departure
self._departure = planner.location_name(departure)[0] self._departure = self.get_station_id(departure)
self._heading = planner.location_name(heading)[0] if heading else None self._heading = self.get_station_id(heading) if heading else None
self._lines = lines if lines else None self._lines = lines if lines else None
self._delay = timedelta(minutes=delay) self._delay = timedelta(minutes=delay)
self._departureboard = None self._departureboard = None
self._state = None self._state = None
self._attributes = None self._attributes = None
def get_station_id(self, location):
"""Get the station ID."""
if location.isdecimal():
station_info = {"station_name": location, "station_id": location}
else:
station_id = self._planner.location_name(location)[0]["id"]
station_info = {"station_name": location, "station_id": station_id}
return station_info
@property @property
def name(self): def name(self):
"""Return the name of the sensor.""" """Return the name of the sensor."""
@ -113,8 +122,8 @@ class VasttrafikDepartureSensor(Entity):
"""Get the departure board.""" """Get the departure board."""
try: try:
self._departureboard = self._planner.departureboard( self._departureboard = self._planner.departureboard(
self._departure["id"], self._departure["station_id"],
direction=self._heading["id"] if self._heading else None, direction=self._heading["station_id"] if self._heading else None,
date=now() + self._delay, date=now() + self._delay,
) )
except vasttrafik.Error: except vasttrafik.Error:
@ -123,9 +132,9 @@ class VasttrafikDepartureSensor(Entity):
if not self._departureboard: if not self._departureboard:
_LOGGER.debug( _LOGGER.debug(
"No departures from %s heading %s", "No departures from departure station %s " "to destination station %s",
self._departure["name"], self._departure["station_name"],
self._heading["name"] if self._heading else "ANY", self._heading["station_name"] if self._heading else "ANY",
) )
self._state = None self._state = None
self._attributes = {} self._attributes = {}