Reference to the right DOMAINS

iMicknl · Jul 21, 2020 · f212bc6 · f212bc6
1 parent 67e09af
commit f212bc6
Show file tree

Hide file tree

Showing 8 changed files with 22 additions and 11 deletions.
diff --git a/custom_components/tahoma/binary_sensor.py b/custom_components/tahoma/binary_sensor.py
@@ -8,10 +8,11 @@
     DEVICE_CLASS_OCCUPANCY,
     DEVICE_CLASS_OPENING,
     DEVICE_CLASS_SMOKE,
+    DOMAIN as BINARY_SENSOR,
     BinarySensorEntity,
 )
 
-from .const import BINARY_SENSOR, DOMAIN
+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
@@ -1,8 +1,10 @@
 """Support for TaHoma climate devices."""
 
+from homeassistant.components.climate import DOMAIN as CLIMATE
+
 from .climate_aeh import AtlanticElectricalHeater
 from .climate_st import SomfyThermostat
-from .const import CLIMATE, DOMAIN
+from .const import DOMAIN
 
 AEH = "AtlanticElectricalHeater"
 ST = "SomfyThermostat"

diff --git a/custom_components/tahoma/cover.py b/custom_components/tahoma/cover.py
@@ -11,6 +11,7 @@
     DEVICE_CLASS_GATE,
     DEVICE_CLASS_SHUTTER,
     DEVICE_CLASS_WINDOW,
+    DOMAIN as COVER,
     SUPPORT_CLOSE,
     SUPPORT_CLOSE_TILT,
     SUPPORT_OPEN,
@@ -22,7 +23,7 @@
     CoverEntity,
 )
 
-from .const import COVER, DOMAIN
+from .const import DOMAIN
 from .switch import COMMAND_CYCLE
 from .tahoma_device import TahomaDevice
 
@@ -101,6 +102,7 @@ class TahomaCover(TahomaDevice, CoverEntity):
     def current_cover_position(self):
         """
         Return current position of cover.
+
         None is unknown, 0 is closed, 100 is fully open.
         """
 

diff --git a/custom_components/tahoma/light.py b/custom_components/tahoma/light.py
@@ -6,6 +6,7 @@
     ATTR_BRIGHTNESS,
     ATTR_EFFECT,
     ATTR_HS_COLOR,
+    DOMAIN as LIGHT,
     SUPPORT_BRIGHTNESS,
     SUPPORT_COLOR,
     SUPPORT_EFFECT,
@@ -14,7 +15,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, LIGHT
+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
@@ -2,10 +2,10 @@
 from datetime import timedelta
 import logging
 
-from homeassistant.components.lock import LockEntity
+from homeassistant.components.lock import DOMAIN as LOCK, LockEntity
 from homeassistant.const import STATE_LOCKED
 
-from .const import DOMAIN, LOCK
+from .const import DOMAIN
 from .tahoma_device import TahomaDevice
 
 _LOGGER = logging.getLogger(__name__)

diff --git a/custom_components/tahoma/scene.py b/custom_components/tahoma/scene.py
@@ -5,9 +5,9 @@
 from tahoma_api.client import TahomaClient
 from tahoma_api.models import Scenario
 
-from homeassistant.components.scene import Scene
+from homeassistant.components.scene import DOMAIN as SCENE, Scene
 
-from .const import DOMAIN, SCENE
+from .const import DOMAIN
 
 _LOGGER = logging.getLogger(__name__)
 

diff --git a/custom_components/tahoma/sensor.py b/custom_components/tahoma/sensor.py
@@ -9,6 +9,7 @@
     DEVICE_CLASS_ILLUMINANCE,
     DEVICE_CLASS_POWER,
     DEVICE_CLASS_TEMPERATURE,
+    DOMAIN as SENSOR,
     ENERGY_WATT_HOUR,
     POWER_WATT,
     TEMP_CELSIUS,
@@ -18,7 +19,7 @@
 )
 from homeassistant.helpers.entity import Entity
 
-from .const import DOMAIN, SENSOR
+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
@@ -2,10 +2,14 @@
 import logging
 from typing import Optional
 
-from homeassistant.components.switch import DEVICE_CLASS_SWITCH, SwitchEntity
+from homeassistant.components.switch import (
+    DEVICE_CLASS_SWITCH,
+    DOMAIN as SWITCH,
+    SwitchEntity,
+)
 from homeassistant.const import STATE_ON
 
-from .const import COMMAND_OFF, COMMAND_ON, CORE_ON_OFF_STATE, DOMAIN, SWITCH
+from .const import COMMAND_OFF, COMMAND_ON, CORE_ON_OFF_STATE, DOMAIN
 from .tahoma_device import TahomaDevice
 
 _LOGGER = logging.getLogger(__name__)