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

Move imports in vasttrafik component (#27480)

Browse Source
This commit is contained in:
Quentame 2019-10-12 21:54:41 +02:00 committed by Paulus Schoutsen
parent 54d63c63c3
commit 0a2ec30ce3
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2,6 +2,7 @@
from datetime import timedelta
import logging
import vasttrafik
import voluptuous as vol
import homeassistant.helpers.config_validation as cv
@ -54,7 +55,6 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
def setup_platform(hass, config, add_entities, discovery_info=None):
"""Set up the departure sensor."""
import vasttrafik
planner = vasttrafik.JournyPlanner(config.get(CONF_KEY), config.get(CONF_SECRET))
sensors = []
@ -62,7 +62,6 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
for departure in config.get(CONF_DEPARTURES):
sensors.append(
VasttrafikDepartureSensor(
vasttrafik,
planner,
departure.get(CONF_NAME),
departure.get(CONF_FROM),
@ -77,9 +76,8 @@ def setup_platform(hass, config, add_entities, discovery_info=None):
class VasttrafikDepartureSensor(Entity):
"""Implementation of a Vasttrafik Departure Sensor."""
def __init__(self, vasttrafik, planner, name, departure, heading, lines, delay):
def __init__(self, planner, name, departure, heading, lines, delay):
"""Initialize the sensor."""
self._vasttrafik = vasttrafik
self._planner = planner
self._name = name or departure
self._departure = planner.location_name(departure)[0]
@ -119,7 +117,7 @@ class VasttrafikDepartureSensor(Entity):
direction=self._heading["id"] if self._heading else None,
date=now() + self._delay,
)
except self._vasttrafik.Error:
except vasttrafik.Error:
_LOGGER.debug("Unable to read departure board, updating token")
self._planner.update_token()