Use enums in tolo (#62020)

homeassistant/components/tolo/binary_sensor.py

@@ -1,12 +1,12 @@
 """TOLO Sauna binary sensors."""
 
 from homeassistant.components.binary_sensor import (
-    DEVICE_CLASS_OPENING,
+    BinarySensorDeviceClass,
     BinarySensorEntity,
 )
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import ENTITY_CATEGORY_DIAGNOSTIC
 from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity import EntityCategory
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
 from . import ToloSaunaCoordinatorEntity, ToloSaunaUpdateCoordinator
@@ -31,9 +31,9 @@ async def async_setup_entry(
 class ToloFlowInBinarySensor(ToloSaunaCoordinatorEntity, BinarySensorEntity):
     """Water In Valve Sensor."""
 
-    _attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC
+    _attr_entity_category = EntityCategory.DIAGNOSTIC
     _attr_name = "Water In Valve"
-    _attr_device_class = DEVICE_CLASS_OPENING
+    _attr_device_class = BinarySensorDeviceClass.OPENING
     _attr_icon = "mdi:water-plus-outline"
 
     def __init__(
@@ -53,9 +53,9 @@ class ToloFlowInBinarySensor(ToloSaunaCoordinatorEntity, BinarySensorEntity):
 class ToloFlowOutBinarySensor(ToloSaunaCoordinatorEntity, BinarySensorEntity):
     """Water Out Valve Sensor."""
 
-    _attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC
+    _attr_entity_category = EntityCategory.DIAGNOSTIC
     _attr_name = "Water Out Valve"
-    _attr_device_class = DEVICE_CLASS_OPENING
+    _attr_device_class = BinarySensorDeviceClass.OPENING
     _attr_icon = "mdi:water-minus-outline"
 
     def __init__(

homeassistant/components/tolo/button.py

@@ -4,8 +4,8 @@ from tololib.const import LampMode
 
 from homeassistant.components.button import ButtonEntity
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import ENTITY_CATEGORY_CONFIG
 from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity import EntityCategory
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
 from . import ToloSaunaCoordinatorEntity, ToloSaunaUpdateCoordinator
@@ -29,7 +29,7 @@ async def async_setup_entry(
 class ToloLampNextColorButton(ToloSaunaCoordinatorEntity, ButtonEntity):
     """Button for switching to the next lamp color."""
 
-    _attr_entity_category = ENTITY_CATEGORY_CONFIG
+    _attr_entity_category = EntityCategory.CONFIG
     _attr_icon = "mdi:palette"
     _attr_name = "Next Color"
 

homeassistant/components/tolo/select.py

@@ -6,8 +6,8 @@ from tololib.const import LampMode
 
 from homeassistant.components.select import SelectEntity
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import ENTITY_CATEGORY_CONFIG
 from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity import EntityCategory
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
 from . import ToloSaunaCoordinatorEntity, ToloSaunaUpdateCoordinator
@@ -28,7 +28,7 @@ class ToloLampModeSelect(ToloSaunaCoordinatorEntity, SelectEntity):
     """TOLO Sauna lamp mode select."""
 
     _attr_device_class = "tolo__lamp_mode"
-    _attr_entity_category = ENTITY_CATEGORY_CONFIG
+    _attr_entity_category = EntityCategory.CONFIG
     _attr_icon = "mdi:lightbulb-multiple-outline"
     _attr_name = "Lamp Mode"
     _attr_options = [lamp_mode.name.lower() for lamp_mode in LampMode]

homeassistant/components/tolo/sensor.py

@@ -1,14 +1,14 @@
 """TOLO Sauna (non-binary, general) sensors."""
 
-from homeassistant.components.sensor import STATE_CLASS_MEASUREMENT, SensorEntity
-from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import (
-    DEVICE_CLASS_TEMPERATURE,
-    ENTITY_CATEGORY_DIAGNOSTIC,
-    PERCENTAGE,
-    TEMP_CELSIUS,
+from homeassistant.components.sensor import (
+    SensorDeviceClass,
+    SensorEntity,
+    SensorStateClass,
 )
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import PERCENTAGE, TEMP_CELSIUS
 from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity import EntityCategory
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
 from . import ToloSaunaCoordinatorEntity, ToloSaunaUpdateCoordinator
@@ -33,10 +33,10 @@ async def async_setup_entry(
 class ToloWaterLevelSensor(ToloSaunaCoordinatorEntity, SensorEntity):
     """Sensor for tank water level."""
 
-    _attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC
+    _attr_entity_category = EntityCategory.DIAGNOSTIC
     _attr_name = "Water Level"
     _attr_icon = "mdi:waves-arrow-up"
-    _attr_state_class = STATE_CLASS_MEASUREMENT
+    _attr_state_class = SensorStateClass.MEASUREMENT
     _attr_native_unit_of_measurement = PERCENTAGE
 
     def __init__(
@@ -56,10 +56,10 @@ class ToloWaterLevelSensor(ToloSaunaCoordinatorEntity, SensorEntity):
 class ToloTankTemperatureSensor(ToloSaunaCoordinatorEntity, SensorEntity):
     """Sensor for tank temperature."""
 
-    _attr_entity_category = ENTITY_CATEGORY_DIAGNOSTIC
+    _attr_entity_category = EntityCategory.DIAGNOSTIC
     _attr_name = "Tank Temperature"
-    _attr_device_class = DEVICE_CLASS_TEMPERATURE
-    _attr_state_class = STATE_CLASS_MEASUREMENT
+    _attr_device_class = SensorDeviceClass.TEMPERATURE
+    _attr_state_class = SensorStateClass.MEASUREMENT
     _attr_native_unit_of_measurement = TEMP_CELSIUS
 
     def __init__(