From bf410ec53b48159fdd04d8c67a2972e7bb7939ae Mon Sep 17 00:00:00 2001
From: Mick Vleeshouwer <mick@imick.nl>
Date: Tue, 21 Jul 2020 13:05:30 +0200
Subject: [PATCH] Remove unused imports

---
 custom_components/tahoma/binary_sensor.py | 2 +-
 custom_components/tahoma/climate.py       | 2 +-
 custom_components/tahoma/const.py         | 2 +-
 custom_components/tahoma/cover.py         | 2 +-
 custom_components/tahoma/light.py         | 2 +-
 custom_components/tahoma/lock.py          | 5 +----
 custom_components/tahoma/sensor.py        | 2 +-
 custom_components/tahoma/switch.py        | 2 +-
 8 files changed, 8 insertions(+), 11 deletions(-)

diff --git a/custom_components/tahoma/binary_sensor.py b/custom_components/tahoma/binary_sensor.py
index 52b004685..63c6a0f5e 100644
--- a/custom_components/tahoma/binary_sensor.py
+++ b/custom_components/tahoma/binary_sensor.py
@@ -11,7 +11,7 @@
     BinarySensorEntity,
 )
 
-from .const import DOMAIN, TAHOMA_TYPES
+from .const import DOMAIN
 from .tahoma_device import TahomaDevice
 
 SCAN_INTERVAL = timedelta(seconds=120)
diff --git a/custom_components/tahoma/climate.py b/custom_components/tahoma/climate.py
index 6c6fad282..937daf4f7 100644
--- a/custom_components/tahoma/climate.py
+++ b/custom_components/tahoma/climate.py
@@ -2,7 +2,7 @@
 
 from .climate_aeh import AtlanticElectricalHeater
 from .climate_st import SomfyThermostat
-from .const import DOMAIN, TAHOMA_TYPES
+from .const import DOMAIN
 
 AEH = "AtlanticElectricalHeater"
 ST = "SomfyThermostat"
diff --git a/custom_components/tahoma/const.py b/custom_components/tahoma/const.py
index 1858f3962..7cb23ede1 100644
--- a/custom_components/tahoma/const.py
+++ b/custom_components/tahoma/const.py
@@ -2,7 +2,7 @@
 
 DOMAIN = "tahoma"
 
-# Used to map the Somfy uiClass to the Home Assistant platform
+# Used to map the Somfy widget and uiClass to the Home Assistant platform
 TAHOMA_TYPES = {
     "Light": "light",
     "ExteriorScreen": "cover",
diff --git a/custom_components/tahoma/cover.py b/custom_components/tahoma/cover.py
index b639b44e2..dc8ebb400 100644
--- a/custom_components/tahoma/cover.py
+++ b/custom_components/tahoma/cover.py
@@ -22,7 +22,7 @@
     CoverEntity,
 )
 
-from .const import DOMAIN, TAHOMA_TYPES
+from .const import DOMAIN
 from .switch import COMMAND_CYCLE
 from .tahoma_device import TahomaDevice
 
diff --git a/custom_components/tahoma/light.py b/custom_components/tahoma/light.py
index 33a5a599b..e32da10f8 100644
--- a/custom_components/tahoma/light.py
+++ b/custom_components/tahoma/light.py
@@ -14,7 +14,7 @@
 from homeassistant.const import STATE_ON
 import homeassistant.util.color as color_util
 
-from .const import COMMAND_OFF, COMMAND_ON, CORE_ON_OFF_STATE, DOMAIN, TAHOMA_TYPES
+from .const import COMMAND_OFF, COMMAND_ON, CORE_ON_OFF_STATE, DOMAIN
 from .tahoma_device import TahomaDevice
 
 _LOGGER = logging.getLogger(__name__)
diff --git a/custom_components/tahoma/lock.py b/custom_components/tahoma/lock.py
index c1c2b6c47..3ba0777b2 100644
--- a/custom_components/tahoma/lock.py
+++ b/custom_components/tahoma/lock.py
@@ -5,7 +5,6 @@
 from homeassistant.components.lock import LockEntity
 from homeassistant.const import STATE_LOCKED
 
-from .const import DOMAIN, TAHOMA_TYPES
 from .tahoma_device import TahomaDevice
 
 _LOGGER = logging.getLogger(__name__)
@@ -25,9 +24,7 @@ async def async_setup_entry(hass, entry, async_add_entities):
     controller = data.get("controller")
 
     entities = [
-        TahomaLock(device, controller)
-        for device in data.get("devices")
-        if TAHOMA_TYPES[device.ui_class] == "lock"
+        TahomaLock(device, controller) for device in data.get("entities").get("lock")
     ]
 
     async_add_entities(entities)
diff --git a/custom_components/tahoma/sensor.py b/custom_components/tahoma/sensor.py
index b3a92be95..81fbcb9dc 100644
--- a/custom_components/tahoma/sensor.py
+++ b/custom_components/tahoma/sensor.py
@@ -18,7 +18,7 @@
 )
 from homeassistant.helpers.entity import Entity
 
-from .const import DOMAIN, TAHOMA_TYPES
+from .const import DOMAIN
 from .tahoma_device import TahomaDevice
 
 _LOGGER = logging.getLogger(__name__)
diff --git a/custom_components/tahoma/switch.py b/custom_components/tahoma/switch.py
index 2cfed7fcd..44a2a1f27 100644
--- a/custom_components/tahoma/switch.py
+++ b/custom_components/tahoma/switch.py
@@ -5,7 +5,7 @@
 from homeassistant.components.switch import DEVICE_CLASS_SWITCH, SwitchEntity
 from homeassistant.const import STATE_ON
 
-from .const import COMMAND_OFF, COMMAND_ON, CORE_ON_OFF_STATE, DOMAIN, TAHOMA_TYPES
+from .const import COMMAND_OFF, COMMAND_ON, CORE_ON_OFF_STATE, DOMAIN
 from .tahoma_device import TahomaDevice
 
 _LOGGER = logging.getLogger(__name__)