From df0c8064b27bc3d36e51f119fd1bec0da4cedd11 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Mon, 23 Sep 2024 14:17:28 +0200
Subject: [PATCH] Move tolo base entity to separate module (#126530)

---
 homeassistant/components/tolo/__init__.py     | 25 +----------------
 .../components/tolo/binary_sensor.py          |  3 +-
 homeassistant/components/tolo/button.py       |  3 +-
 homeassistant/components/tolo/climate.py      |  3 +-
 homeassistant/components/tolo/entity.py       | 28 +++++++++++++++++++
 homeassistant/components/tolo/fan.py          |  3 +-
 homeassistant/components/tolo/light.py        |  3 +-
 homeassistant/components/tolo/number.py       |  3 +-
 homeassistant/components/tolo/select.py       |  3 +-
 homeassistant/components/tolo/sensor.py       |  3 +-
 homeassistant/components/tolo/switch.py       |  3 +-
 11 files changed, 47 insertions(+), 33 deletions(-)
 create mode 100644 homeassistant/components/tolo/entity.py

diff --git a/homeassistant/components/tolo/__init__.py b/homeassistant/components/tolo/__init__.py
index a90d23b0e22..58ba9f550a9 100644
--- a/homeassistant/components/tolo/__init__.py
+++ b/homeassistant/components/tolo/__init__.py
@@ -11,12 +11,7 @@ from tololib import ToloClient, ToloSettings, ToloStatus
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import CONF_HOST, Platform
 from homeassistant.core import HomeAssistant
-from homeassistant.helpers.device_registry import DeviceInfo
-from homeassistant.helpers.update_coordinator import (
-    CoordinatorEntity,
-    DataUpdateCoordinator,
-    UpdateFailed,
-)
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
 
 from .const import DEFAULT_RETRY_COUNT, DEFAULT_RETRY_TIMEOUT, DOMAIN
 
@@ -89,21 +84,3 @@ class ToloSaunaUpdateCoordinator(DataUpdateCoordinator[ToloSaunaData]):  # pylin
         except TimeoutError as error:
             raise UpdateFailed("communication timeout") from error
         return ToloSaunaData(status, settings)
-
-
-class ToloSaunaCoordinatorEntity(CoordinatorEntity[ToloSaunaUpdateCoordinator]):
-    """CoordinatorEntity for TOLO Sauna."""
-
-    _attr_has_entity_name = True
-
-    def __init__(
-        self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry
-    ) -> None:
-        """Initialize ToloSaunaCoordinatorEntity."""
-        super().__init__(coordinator)
-        self._attr_device_info = DeviceInfo(
-            name="TOLO Sauna",
-            identifiers={(DOMAIN, entry.entry_id)},
-            manufacturer="SteamTec",
-            model=self.coordinator.data.status.model.name.capitalize(),
-        )
diff --git a/homeassistant/components/tolo/binary_sensor.py b/homeassistant/components/tolo/binary_sensor.py
index f8cb442c92f..835bc913a86 100644
--- a/homeassistant/components/tolo/binary_sensor.py
+++ b/homeassistant/components/tolo/binary_sensor.py
@@ -9,8 +9,9 @@ from homeassistant.const import EntityCategory
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from . import ToloSaunaCoordinatorEntity, ToloSaunaUpdateCoordinator
+from . import ToloSaunaUpdateCoordinator
 from .const import DOMAIN
+from .entity import ToloSaunaCoordinatorEntity
 
 
 async def async_setup_entry(
diff --git a/homeassistant/components/tolo/button.py b/homeassistant/components/tolo/button.py
index 9a8ac67b9fe..7c32d7d7a29 100644
--- a/homeassistant/components/tolo/button.py
+++ b/homeassistant/components/tolo/button.py
@@ -8,8 +8,9 @@ from homeassistant.const import EntityCategory
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from . import ToloSaunaCoordinatorEntity, ToloSaunaUpdateCoordinator
+from . import ToloSaunaUpdateCoordinator
 from .const import DOMAIN
+from .entity import ToloSaunaCoordinatorEntity
 
 
 async def async_setup_entry(
diff --git a/homeassistant/components/tolo/climate.py b/homeassistant/components/tolo/climate.py
index 2994d97d54a..f6360e1d99b 100644
--- a/homeassistant/components/tolo/climate.py
+++ b/homeassistant/components/tolo/climate.py
@@ -25,8 +25,9 @@ from homeassistant.const import ATTR_TEMPERATURE, PRECISION_WHOLE, UnitOfTempera
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from . import ToloSaunaCoordinatorEntity, ToloSaunaUpdateCoordinator
+from . import ToloSaunaUpdateCoordinator
 from .const import DOMAIN
+from .entity import ToloSaunaCoordinatorEntity
 
 
 async def async_setup_entry(
diff --git a/homeassistant/components/tolo/entity.py b/homeassistant/components/tolo/entity.py
new file mode 100644
index 00000000000..68ddc382e7f
--- /dev/null
+++ b/homeassistant/components/tolo/entity.py
@@ -0,0 +1,28 @@
+"""Component to control TOLO Sauna/Steam Bath."""
+
+from __future__ import annotations
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.helpers.device_registry import DeviceInfo
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from . import ToloSaunaUpdateCoordinator
+from .const import DOMAIN
+
+
+class ToloSaunaCoordinatorEntity(CoordinatorEntity[ToloSaunaUpdateCoordinator]):
+    """CoordinatorEntity for TOLO Sauna."""
+
+    _attr_has_entity_name = True
+
+    def __init__(
+        self, coordinator: ToloSaunaUpdateCoordinator, entry: ConfigEntry
+    ) -> None:
+        """Initialize ToloSaunaCoordinatorEntity."""
+        super().__init__(coordinator)
+        self._attr_device_info = DeviceInfo(
+            name="TOLO Sauna",
+            identifiers={(DOMAIN, entry.entry_id)},
+            manufacturer="SteamTec",
+            model=self.coordinator.data.status.model.name.capitalize(),
+        )
diff --git a/homeassistant/components/tolo/fan.py b/homeassistant/components/tolo/fan.py
index 034bdb0b6a6..396dc0b0da4 100644
--- a/homeassistant/components/tolo/fan.py
+++ b/homeassistant/components/tolo/fan.py
@@ -9,8 +9,9 @@ from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from . import ToloSaunaCoordinatorEntity, ToloSaunaUpdateCoordinator
+from . import ToloSaunaUpdateCoordinator
 from .const import DOMAIN
+from .entity import ToloSaunaCoordinatorEntity
 
 
 async def async_setup_entry(
diff --git a/homeassistant/components/tolo/light.py b/homeassistant/components/tolo/light.py
index 809bb367072..5491aa90ea4 100644
--- a/homeassistant/components/tolo/light.py
+++ b/homeassistant/components/tolo/light.py
@@ -9,8 +9,9 @@ from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from . import ToloSaunaCoordinatorEntity, ToloSaunaUpdateCoordinator
+from . import ToloSaunaUpdateCoordinator
 from .const import DOMAIN
+from .entity import ToloSaunaCoordinatorEntity
 
 
 async def async_setup_entry(
diff --git a/homeassistant/components/tolo/number.py b/homeassistant/components/tolo/number.py
index 2d2c20715fa..acdd26fe9c0 100644
--- a/homeassistant/components/tolo/number.py
+++ b/homeassistant/components/tolo/number.py
@@ -20,8 +20,9 @@ from homeassistant.const import EntityCategory, UnitOfTime
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from . import ToloSaunaCoordinatorEntity, ToloSaunaUpdateCoordinator
+from . import ToloSaunaUpdateCoordinator
 from .const import DOMAIN
+from .entity import ToloSaunaCoordinatorEntity
 
 
 @dataclass(frozen=True, kw_only=True)
diff --git a/homeassistant/components/tolo/select.py b/homeassistant/components/tolo/select.py
index 96335cecc68..b41595d3a34 100644
--- a/homeassistant/components/tolo/select.py
+++ b/homeassistant/components/tolo/select.py
@@ -13,8 +13,9 @@ from homeassistant.const import EntityCategory
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from . import ToloSaunaCoordinatorEntity, ToloSaunaUpdateCoordinator
+from . import ToloSaunaUpdateCoordinator
 from .const import DOMAIN, AromaTherapySlot, LampMode
+from .entity import ToloSaunaCoordinatorEntity
 
 
 @dataclass(frozen=True, kw_only=True)
diff --git a/homeassistant/components/tolo/sensor.py b/homeassistant/components/tolo/sensor.py
index bee01cc283f..8ea6b68ae95 100644
--- a/homeassistant/components/tolo/sensor.py
+++ b/homeassistant/components/tolo/sensor.py
@@ -23,8 +23,9 @@ from homeassistant.const import (
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from . import ToloSaunaCoordinatorEntity, ToloSaunaUpdateCoordinator
+from . import ToloSaunaUpdateCoordinator
 from .const import DOMAIN
+from .entity import ToloSaunaCoordinatorEntity
 
 
 @dataclass(frozen=True, kw_only=True)
diff --git a/homeassistant/components/tolo/switch.py b/homeassistant/components/tolo/switch.py
index b90f548ee76..9799d106658 100644
--- a/homeassistant/components/tolo/switch.py
+++ b/homeassistant/components/tolo/switch.py
@@ -13,8 +13,9 @@ from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from . import ToloSaunaCoordinatorEntity, ToloSaunaUpdateCoordinator
+from . import ToloSaunaUpdateCoordinator
 from .const import DOMAIN
+from .entity import ToloSaunaCoordinatorEntity
 
 
 @dataclass(frozen=True, kw_only=True)