Refactor fix_44

iMicknl · Jun 22, 2020 · 3b3c534 · 3b3c534
1 parent 962cff2
commit 3b3c534
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 9 deletions.
diff --git a/custom_components/tahoma/const.py b/custom_components/tahoma/const.py
@@ -69,12 +69,6 @@
 }
 
 
-TAHOMA_HORIZONTAL_DEVICES = {
-    "PositionableHorizontalAwning": DEVICE_CLASS_AWNING,
-    "PositionableHorizontalAwningUno": DEVICE_CLASS_AWNING,
-    "UpDownHorizontalAwning": DEVICE_CLASS_AWNING,
-}
-
 # Used to map the Somfy widget or uiClass to the Home Assistant device classes
 TAHOMA_BINARY_SENSOR_DEVICE_CLASSES = {
     "SmokeSensor": DEVICE_CLASS_SMOKE,

diff --git a/custom_components/tahoma/cover.py b/custom_components/tahoma/cover.py
@@ -44,7 +44,6 @@
     IO_PRIORITY_LOCK_LEVEL_STATE,
     IO_PRIORITY_LOCK_ORIGINATOR_STATE,
     TAHOMA_COVER_DEVICE_CLASSES,
-    TAHOMA_HORIZONTAL_DEVICES,
     TAHOMA_TYPES,
 )
 from .tahoma_device import TahomaDevice
@@ -127,7 +126,7 @@ def update(self):
 
         if getattr(self, "_position", False):
             # HorizontalAwning devices need a reversed position that can not be obtained via the API
-            if self.tahoma_device.widget in TAHOMA_HORIZONTAL_DEVICES:
+            if "Horizontal" in self.tahoma_device.widget:
                 self._position = 100 - self._position
 
             # TODO Check if this offset is really necessary
@@ -186,7 +185,7 @@ def set_cover_position(self, **kwargs):
         position = 100 - kwargs.get(ATTR_POSITION, 0)
 
         # HorizontalAwning devices need a reversed position that can not be obtained via the API
-        if self.tahoma_device.widget in TAHOMA_HORIZONTAL_DEVICES:
+        if "Horizontal" in self.tahoma_device.widget:
             position = kwargs.get(ATTR_POSITION, 0)
 
         if COMMAND_SET_POSITION in self.tahoma_device.command_definitions: