From 4bec14b0f6f73bf8e2e963cec21e3faa0d3b25c6 Mon Sep 17 00:00:00 2001 From: springstan <46536646+springstan@users.noreply.github.com> Date: Fri, 29 Nov 2019 22:16:14 +0100 Subject: [PATCH] Move imports to top for swiss_public_transport (#29228) * Move imports to top for swiss_public_transport * Remove import of exceptions since only one exception is used --- homeassistant/components/swiss_public_transport/sensor.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/homeassistant/components/swiss_public_transport/sensor.py b/homeassistant/components/swiss_public_transport/sensor.py index 3cf8babf554..be967247dc7 100644 --- a/homeassistant/components/swiss_public_transport/sensor.py +++ b/homeassistant/components/swiss_public_transport/sensor.py @@ -2,6 +2,8 @@ from datetime import timedelta import logging +from opendata_transport import OpendataTransport +from opendata_transport.exceptions import OpendataTransportError import voluptuous as vol from homeassistant.components.sensor import PLATFORM_SCHEMA @@ -45,7 +47,6 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( async def async_setup_platform(hass, config, async_add_entities, discovery_info=None): """Set up the Swiss public transport sensor.""" - from opendata_transport import OpendataTransport, exceptions name = config.get(CONF_NAME) start = config.get(CONF_START) @@ -56,7 +57,7 @@ async def async_setup_platform(hass, config, async_add_entities, discovery_info= try: await opendata.async_get_data() - except exceptions.OpendataTransportError: + except OpendataTransportError: _LOGGER.error( "Check at http://transport.opendata.ch/examples/stationboard.html " "if your station names are valid" @@ -122,7 +123,6 @@ class SwissPublicTransportSensor(Entity): async def async_update(self): """Get the latest data from opendata.ch and update the states.""" - from opendata_transport.exceptions import OpendataTransportError try: if self._remaining_time.total_seconds() < 0: