From 6b959f42f61a8b005b3340adbae47f7a90101eae Mon Sep 17 00:00:00 2001
From: Vincent Wolsink <ik@vincentwolsink.nl>
Date: Fri, 18 Jul 2025 00:06:51 +0200
Subject: [PATCH] Introduce base entity for supporting multiple platforms in
 Huum (#148957)

---
 homeassistant/components/huum/climate.py | 13 ++-----------
 homeassistant/components/huum/entity.py  | 24 ++++++++++++++++++++++++
 2 files changed, 26 insertions(+), 11 deletions(-)
 create mode 100644 homeassistant/components/huum/entity.py

diff --git a/homeassistant/components/huum/climate.py b/homeassistant/components/huum/climate.py
index c82fd2c91a5..6a50137f0a7 100644
--- a/homeassistant/components/huum/climate.py
+++ b/homeassistant/components/huum/climate.py
@@ -16,12 +16,10 @@ from homeassistant.components.climate import (
 from homeassistant.const import ATTR_TEMPERATURE, PRECISION_WHOLE, UnitOfTemperature
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import HomeAssistantError
-from homeassistant.helpers.device_registry import DeviceInfo
 from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
-from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
-from .const import DOMAIN
 from .coordinator import HuumConfigEntry, HuumDataUpdateCoordinator
+from .entity import HuumBaseEntity
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -35,7 +33,7 @@ async def async_setup_entry(
     async_add_entities([HuumDevice(entry.runtime_data)])
 
 
-class HuumDevice(CoordinatorEntity[HuumDataUpdateCoordinator], ClimateEntity):
+class HuumDevice(HuumBaseEntity, ClimateEntity):
     """Representation of a heater."""
 
     _attr_hvac_modes = [HVACMode.HEAT, HVACMode.OFF]
@@ -46,7 +44,6 @@ class HuumDevice(CoordinatorEntity[HuumDataUpdateCoordinator], ClimateEntity):
     )
     _attr_target_temperature_step = PRECISION_WHOLE
     _attr_temperature_unit = UnitOfTemperature.CELSIUS
-    _attr_has_entity_name = True
     _attr_name = None
 
     def __init__(self, coordinator: HuumDataUpdateCoordinator) -> None:
@@ -54,12 +51,6 @@ class HuumDevice(CoordinatorEntity[HuumDataUpdateCoordinator], ClimateEntity):
         super().__init__(coordinator)
 
         self._attr_unique_id = coordinator.config_entry.entry_id
-        self._attr_device_info = DeviceInfo(
-            identifiers={(DOMAIN, coordinator.config_entry.entry_id)},
-            name="Huum sauna",
-            manufacturer="Huum",
-            model="UKU WiFi",
-        )
 
     @property
     def min_temp(self) -> int:
diff --git a/homeassistant/components/huum/entity.py b/homeassistant/components/huum/entity.py
new file mode 100644
index 00000000000..cd30119f6fe
--- /dev/null
+++ b/homeassistant/components/huum/entity.py
@@ -0,0 +1,24 @@
+"""Define Huum Base entity."""
+
+from homeassistant.helpers.device_registry import DeviceInfo
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from .const import DOMAIN
+from .coordinator import HuumDataUpdateCoordinator
+
+
+class HuumBaseEntity(CoordinatorEntity[HuumDataUpdateCoordinator]):
+    """Huum base Entity."""
+
+    _attr_has_entity_name = True
+
+    def __init__(self, coordinator: HuumDataUpdateCoordinator) -> None:
+        """Initialize the entity."""
+        super().__init__(coordinator)
+
+        self._attr_device_info = DeviceInfo(
+            identifiers={(DOMAIN, coordinator.config_entry.entry_id)},
+            name="Huum sauna",
+            manufacturer="Huum",
+            model="UKU WiFi",
+        )