diff --git a/homeassistant/components/wemo/__init__.py b/homeassistant/components/wemo/__init__.py index 9e479991d15..df2d8ed1f31 100644 --- a/homeassistant/components/wemo/__init__.py +++ b/homeassistant/components/wemo/__init__.py @@ -1,6 +1,7 @@ """Support for WeMo device discovery.""" import logging +import pywemo import requests import voluptuous as vol @@ -87,7 +88,6 @@ def setup(hass, config): async def async_setup_entry(hass, entry): """Set up a wemo config entry.""" - import pywemo config = hass.data[DOMAIN] diff --git a/homeassistant/components/wemo/binary_sensor.py b/homeassistant/components/wemo/binary_sensor.py index 4ef18f29021..bc300fde571 100644 --- a/homeassistant/components/wemo/binary_sensor.py +++ b/homeassistant/components/wemo/binary_sensor.py @@ -3,6 +3,7 @@ import asyncio import logging import async_timeout +from pywemo import discovery import requests from homeassistant.components.binary_sensor import BinarySensorDevice @@ -15,7 +16,6 @@ _LOGGER = logging.getLogger(__name__) def setup_platform(hass, config, add_entities, discovery_info=None): """Register discovered WeMo binary sensors.""" - from pywemo import discovery if discovery_info is not None: location = discovery_info["ssdp_description"] diff --git a/homeassistant/components/wemo/fan.py b/homeassistant/components/wemo/fan.py index dde5aa1cd89..91273fa033f 100644 --- a/homeassistant/components/wemo/fan.py +++ b/homeassistant/components/wemo/fan.py @@ -3,11 +3,12 @@ import asyncio import logging from datetime import timedelta -import requests import async_timeout +from pywemo import discovery +import requests import voluptuous as vol -import homeassistant.helpers.config_validation as cv +import homeassistant.helpers.config_validation as cv from homeassistant.components.fan import ( DOMAIN, SUPPORT_SET_SPEED, @@ -96,7 +97,6 @@ RESET_FILTER_LIFE_SCHEMA = vol.Schema({vol.Required(ATTR_ENTITY_ID): cv.entity_i def setup_platform(hass, config, add_entities, discovery_info=None): """Set up discovered WeMo humidifiers.""" - from pywemo import discovery if DATA_KEY not in hass.data: hass.data[DATA_KEY] = {} diff --git a/homeassistant/components/wemo/light.py b/homeassistant/components/wemo/light.py index be6aa6f47f7..dab96eb8c94 100644 --- a/homeassistant/components/wemo/light.py +++ b/homeassistant/components/wemo/light.py @@ -3,8 +3,9 @@ import asyncio import logging from datetime import timedelta -import requests import async_timeout +from pywemo import discovery +import requests from homeassistant import util from homeassistant.components.light import ( @@ -35,7 +36,6 @@ SUPPORT_WEMO = ( def setup_platform(hass, config, add_entities, discovery_info=None): """Set up discovered WeMo switches.""" - from pywemo import discovery if discovery_info is not None: location = discovery_info["ssdp_description"] diff --git a/homeassistant/components/wemo/switch.py b/homeassistant/components/wemo/switch.py index 1bc85506987..c1d07a06902 100644 --- a/homeassistant/components/wemo/switch.py +++ b/homeassistant/components/wemo/switch.py @@ -2,9 +2,10 @@ import asyncio import logging from datetime import datetime, timedelta -import requests import async_timeout +from pywemo import discovery +import requests from homeassistant.components.switch import SwitchDevice from homeassistant.exceptions import PlatformNotReady @@ -32,7 +33,6 @@ WEMO_STANDBY = 8 def setup_platform(hass, config, add_entities, discovery_info=None): """Set up discovered WeMo switches.""" - from pywemo import discovery if discovery_info is not None: location = discovery_info["ssdp_description"]