From 58942601b4f012abc361357f71f0efa63d2ab998 Mon Sep 17 00:00:00 2001
From: Robert Hillis <tkdrob4390@yahoo.com>
Date: Thu, 16 Dec 2021 07:09:35 -0500
Subject: [PATCH] Use enums in tolo (#62020)

---
 .../components/tolo/binary_sensor.py          | 12 +++++-----
 homeassistant/components/tolo/button.py       |  4 ++--
 homeassistant/components/tolo/select.py       |  4 ++--
 homeassistant/components/tolo/sensor.py       | 24 +++++++++----------
 4 files changed, 22 insertions(+), 22 deletions(-)

diff --git a/homeassistant/components/tolo/binary_sensor.py b/homeassistant/components/tolo/binary_sensor.py
index 777e2f16942..7b9bda34674 100644
--- a/homeassistant/components/tolo/binary_sensor.py
+++ b/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__(
diff --git a/homeassistant/components/tolo/button.py b/homeassistant/components/tolo/button.py
index 9dd7752b0c9..de3522d96b9 100644
--- a/homeassistant/components/tolo/button.py
+++ b/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"
 
diff --git a/homeassistant/components/tolo/select.py b/homeassistant/components/tolo/select.py
index 1dc1f4f6163..a0cdb0ed128 100644
--- a/homeassistant/components/tolo/select.py
+++ b/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]
diff --git a/homeassistant/components/tolo/sensor.py b/homeassistant/components/tolo/sensor.py
index 23533f68784..bcdb7db0165 100644
--- a/homeassistant/components/tolo/sensor.py
+++ b/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__(