diff --git a/homeassistant/components/isy994/binary_sensor.py b/homeassistant/components/isy994/binary_sensor.py index 74695f2b84a..da12ebd3ef8 100644 --- a/homeassistant/components/isy994/binary_sensor.py +++ b/homeassistant/components/isy994/binary_sensor.py @@ -15,12 +15,11 @@ from pyisy.helpers import NodeProperty from pyisy.nodes import Group, Node from homeassistant.components.binary_sensor import ( - DOMAIN as BINARY_SENSOR, BinarySensorDeviceClass, BinarySensorEntity, ) from homeassistant.config_entries import ConfigEntry -from homeassistant.const import STATE_ON +from homeassistant.const import STATE_ON, Platform from homeassistant.core import CALLBACK_TYPE, HomeAssistant, callback from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.event import async_track_point_in_utc_time @@ -76,7 +75,7 @@ async def async_setup_entry( entity: ISYInsteonBinarySensorEntity | ISYBinarySensorEntity | ISYBinarySensorHeartbeat | ISYBinarySensorProgramEntity hass_isy_data = hass.data[ISY994_DOMAIN][entry.entry_id] - for node in hass_isy_data[ISY994_NODES][BINARY_SENSOR]: + for node in hass_isy_data[ISY994_NODES][Platform.BINARY_SENSOR]: assert isinstance(node, Node) device_class, device_type = _detect_device_type_and_class(node) if node.protocol == PROTO_INSTEON: @@ -189,10 +188,10 @@ async def async_setup_entry( entity = ISYBinarySensorEntity(node, device_class) entities.append(entity) - for name, status, _ in hass_isy_data[ISY994_PROGRAMS][BINARY_SENSOR]: + for name, status, _ in hass_isy_data[ISY994_PROGRAMS][Platform.BINARY_SENSOR]: entities.append(ISYBinarySensorProgramEntity(name, status)) - await migrate_old_unique_ids(hass, BINARY_SENSOR, entities) + await migrate_old_unique_ids(hass, Platform.BINARY_SENSOR, entities) async_add_entities(entities) diff --git a/homeassistant/components/isy994/climate.py b/homeassistant/components/isy994/climate.py index ea6327fa1da..44d64c05be7 100644 --- a/homeassistant/components/isy994/climate.py +++ b/homeassistant/components/isy994/climate.py @@ -19,7 +19,6 @@ from pyisy.nodes import Node from homeassistant.components.climate import ( ATTR_TARGET_TEMP_HIGH, ATTR_TARGET_TEMP_LOW, - DOMAIN as CLIMATE, FAN_AUTO, FAN_OFF, FAN_ON, @@ -29,7 +28,12 @@ from homeassistant.components.climate import ( HVACMode, ) from homeassistant.config_entries import ConfigEntry -from homeassistant.const import ATTR_TEMPERATURE, PRECISION_TENTHS, UnitOfTemperature +from homeassistant.const import ( + ATTR_TEMPERATURE, + PRECISION_TENTHS, + Platform, + UnitOfTemperature, +) from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback @@ -60,10 +64,10 @@ async def async_setup_entry( entities = [] hass_isy_data = hass.data[ISY994_DOMAIN][entry.entry_id] - for node in hass_isy_data[ISY994_NODES][CLIMATE]: + for node in hass_isy_data[ISY994_NODES][Platform.CLIMATE]: entities.append(ISYThermostatEntity(node)) - await migrate_old_unique_ids(hass, CLIMATE, entities) + await migrate_old_unique_ids(hass, Platform.CLIMATE, entities) async_add_entities(entities) diff --git a/homeassistant/components/isy994/cover.py b/homeassistant/components/isy994/cover.py index ac85e8c7670..e7d3349958c 100644 --- a/homeassistant/components/isy994/cover.py +++ b/homeassistant/components/isy994/cover.py @@ -7,11 +7,11 @@ from pyisy.constants import ISY_VALUE_UNKNOWN from homeassistant.components.cover import ( ATTR_POSITION, - DOMAIN as COVER, CoverEntity, CoverEntityFeature, ) from homeassistant.config_entries import ConfigEntry +from homeassistant.const import Platform from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback @@ -33,13 +33,13 @@ async def async_setup_entry( """Set up the ISY cover platform.""" hass_isy_data = hass.data[ISY994_DOMAIN][entry.entry_id] entities: list[ISYCoverEntity | ISYCoverProgramEntity] = [] - for node in hass_isy_data[ISY994_NODES][COVER]: + for node in hass_isy_data[ISY994_NODES][Platform.COVER]: entities.append(ISYCoverEntity(node)) - for name, status, actions in hass_isy_data[ISY994_PROGRAMS][COVER]: + for name, status, actions in hass_isy_data[ISY994_PROGRAMS][Platform.COVER]: entities.append(ISYCoverProgramEntity(name, status, actions)) - await migrate_old_unique_ids(hass, COVER, entities) + await migrate_old_unique_ids(hass, Platform.COVER, entities) async_add_entities(entities) diff --git a/homeassistant/components/isy994/fan.py b/homeassistant/components/isy994/fan.py index 51b89da19a6..5fd14927322 100644 --- a/homeassistant/components/isy994/fan.py +++ b/homeassistant/components/isy994/fan.py @@ -6,8 +6,9 @@ from typing import Any from pyisy.constants import ISY_VALUE_UNKNOWN, PROTO_INSTEON -from homeassistant.components.fan import DOMAIN as FAN, FanEntity, FanEntityFeature +from homeassistant.components.fan import FanEntity, FanEntityFeature from homeassistant.config_entries import ConfigEntry +from homeassistant.const import Platform from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.util.percentage import ( @@ -30,13 +31,13 @@ async def async_setup_entry( hass_isy_data = hass.data[ISY994_DOMAIN][entry.entry_id] entities: list[ISYFanEntity | ISYFanProgramEntity] = [] - for node in hass_isy_data[ISY994_NODES][FAN]: + for node in hass_isy_data[ISY994_NODES][Platform.FAN]: entities.append(ISYFanEntity(node)) - for name, status, actions in hass_isy_data[ISY994_PROGRAMS][FAN]: + for name, status, actions in hass_isy_data[ISY994_PROGRAMS][Platform.FAN]: entities.append(ISYFanProgramEntity(name, status, actions)) - await migrate_old_unique_ids(hass, FAN, entities) + await migrate_old_unique_ids(hass, Platform.FAN, entities) async_add_entities(entities) diff --git a/homeassistant/components/isy994/light.py b/homeassistant/components/isy994/light.py index 0b315996a80..92cfb452969 100644 --- a/homeassistant/components/isy994/light.py +++ b/homeassistant/components/isy994/light.py @@ -7,8 +7,9 @@ from pyisy.constants import ISY_VALUE_UNKNOWN from pyisy.helpers import NodeProperty from pyisy.nodes import Node -from homeassistant.components.light import DOMAIN as LIGHT, ColorMode, LightEntity +from homeassistant.components.light import ColorMode, LightEntity from homeassistant.config_entries import ConfigEntry +from homeassistant.const import Platform from homeassistant.core import HomeAssistant, callback from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.restore_state import RestoreEntity @@ -36,10 +37,10 @@ async def async_setup_entry( restore_light_state = isy_options.get(CONF_RESTORE_LIGHT_STATE, False) entities = [] - for node in hass_isy_data[ISY994_NODES][LIGHT]: + for node in hass_isy_data[ISY994_NODES][Platform.LIGHT]: entities.append(ISYLightEntity(node, restore_light_state)) - await migrate_old_unique_ids(hass, LIGHT, entities) + await migrate_old_unique_ids(hass, Platform.LIGHT, entities) async_add_entities(entities) async_setup_light_services(hass) diff --git a/homeassistant/components/isy994/lock.py b/homeassistant/components/isy994/lock.py index 99e2997f2f6..1101eebd632 100644 --- a/homeassistant/components/isy994/lock.py +++ b/homeassistant/components/isy994/lock.py @@ -5,8 +5,9 @@ from typing import Any from pyisy.constants import ISY_VALUE_UNKNOWN -from homeassistant.components.lock import DOMAIN as LOCK, LockEntity +from homeassistant.components.lock import LockEntity from homeassistant.config_entries import ConfigEntry +from homeassistant.const import Platform from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback @@ -23,13 +24,13 @@ async def async_setup_entry( """Set up the ISY lock platform.""" hass_isy_data = hass.data[ISY994_DOMAIN][entry.entry_id] entities: list[ISYLockEntity | ISYLockProgramEntity] = [] - for node in hass_isy_data[ISY994_NODES][LOCK]: + for node in hass_isy_data[ISY994_NODES][Platform.LOCK]: entities.append(ISYLockEntity(node)) - for name, status, actions in hass_isy_data[ISY994_PROGRAMS][LOCK]: + for name, status, actions in hass_isy_data[ISY994_PROGRAMS][Platform.LOCK]: entities.append(ISYLockProgramEntity(name, status, actions)) - await migrate_old_unique_ids(hass, LOCK, entities) + await migrate_old_unique_ids(hass, Platform.LOCK, entities) async_add_entities(entities) diff --git a/homeassistant/components/isy994/sensor.py b/homeassistant/components/isy994/sensor.py index 434e760ceec..4a80229eef7 100644 --- a/homeassistant/components/isy994/sensor.py +++ b/homeassistant/components/isy994/sensor.py @@ -21,13 +21,12 @@ from pyisy.helpers import NodeProperty from pyisy.nodes import Node from homeassistant.components.sensor import ( - DOMAIN as SENSOR, SensorDeviceClass, SensorEntity, SensorStateClass, ) from homeassistant.config_entries import ConfigEntry -from homeassistant.const import UnitOfTemperature +from homeassistant.const import Platform, UnitOfTemperature from homeassistant.core import HomeAssistant from homeassistant.helpers.entity import EntityCategory from homeassistant.helpers.entity_platform import AddEntitiesCallback @@ -113,7 +112,7 @@ async def async_setup_entry( hass_isy_data = hass.data[ISY994_DOMAIN][entry.entry_id] entities: list[ISYSensorEntity | ISYSensorVariableEntity] = [] - for node in hass_isy_data[ISY994_NODES][SENSOR]: + for node in hass_isy_data[ISY994_NODES][Platform.SENSOR]: _LOGGER.debug("Loading %s", node.name) entities.append(ISYSensorEntity(node)) @@ -135,7 +134,7 @@ async def async_setup_entry( for vname, vobj in hass_isy_data[ISY994_VARIABLES]: entities.append(ISYSensorVariableEntity(vname, vobj)) - await migrate_old_unique_ids(hass, SENSOR, entities) + await migrate_old_unique_ids(hass, Platform.SENSOR, entities) async_add_entities(entities) diff --git a/homeassistant/components/isy994/switch.py b/homeassistant/components/isy994/switch.py index 594dbb34810..a637f4f632a 100644 --- a/homeassistant/components/isy994/switch.py +++ b/homeassistant/components/isy994/switch.py @@ -5,8 +5,9 @@ from typing import Any from pyisy.constants import ISY_VALUE_UNKNOWN, PROTO_GROUP -from homeassistant.components.switch import DOMAIN as SWITCH, SwitchEntity +from homeassistant.components.switch import SwitchEntity from homeassistant.config_entries import ConfigEntry +from homeassistant.const import Platform from homeassistant.core import HomeAssistant from homeassistant.helpers.entity_platform import AddEntitiesCallback @@ -21,13 +22,13 @@ async def async_setup_entry( """Set up the ISY switch platform.""" hass_isy_data = hass.data[ISY994_DOMAIN][entry.entry_id] entities: list[ISYSwitchProgramEntity | ISYSwitchEntity] = [] - for node in hass_isy_data[ISY994_NODES][SWITCH]: + for node in hass_isy_data[ISY994_NODES][Platform.SWITCH]: entities.append(ISYSwitchEntity(node)) - for name, status, actions in hass_isy_data[ISY994_PROGRAMS][SWITCH]: + for name, status, actions in hass_isy_data[ISY994_PROGRAMS][Platform.SWITCH]: entities.append(ISYSwitchProgramEntity(name, status, actions)) - await migrate_old_unique_ids(hass, SWITCH, entities) + await migrate_old_unique_ids(hass, Platform.SWITCH, entities) async_add_entities(entities)