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

---
 .../components/tailscale/__init__.py          | 46 ----------------
 .../components/tailscale/binary_sensor.py     |  2 +-
 homeassistant/components/tailscale/entity.py  | 52 +++++++++++++++++++
 homeassistant/components/tailscale/sensor.py  |  2 +-
 4 files changed, 54 insertions(+), 48 deletions(-)
 create mode 100644 homeassistant/components/tailscale/entity.py

diff --git a/homeassistant/components/tailscale/__init__.py b/homeassistant/components/tailscale/__init__.py
index 5498687332f..549bf07e181 100644
--- a/homeassistant/components/tailscale/__init__.py
+++ b/homeassistant/components/tailscale/__init__.py
@@ -2,17 +2,9 @@
 
 from __future__ import annotations
 
-from tailscale import Device as TailscaleDevice
-
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.const import Platform
 from homeassistant.core import HomeAssistant
-from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
-from homeassistant.helpers.entity import EntityDescription
-from homeassistant.helpers.update_coordinator import (
-    CoordinatorEntity,
-    DataUpdateCoordinator,
-)
 
 from .const import DOMAIN
 from .coordinator import TailscaleDataUpdateCoordinator
@@ -37,41 +29,3 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     if unload_ok:
         del hass.data[DOMAIN][entry.entry_id]
     return unload_ok
-
-
-class TailscaleEntity(CoordinatorEntity):
-    """Defines a Tailscale base entity."""
-
-    _attr_has_entity_name = True
-
-    def __init__(
-        self,
-        *,
-        coordinator: DataUpdateCoordinator,
-        device: TailscaleDevice,
-        description: EntityDescription,
-    ) -> None:
-        """Initialize a Tailscale sensor."""
-        super().__init__(coordinator=coordinator)
-        self.entity_description = description
-        self.device_id = device.device_id
-        self._attr_unique_id = f"{device.device_id}_{description.key}"
-
-    @property
-    def device_info(self) -> DeviceInfo:
-        """Return the device info."""
-        device: TailscaleDevice = self.coordinator.data[self.device_id]
-
-        configuration_url = "https://login.tailscale.com/admin/machines/"
-        if device.addresses:
-            configuration_url += device.addresses[0]
-
-        return DeviceInfo(
-            configuration_url=configuration_url,
-            entry_type=DeviceEntryType.SERVICE,
-            identifiers={(DOMAIN, device.device_id)},
-            manufacturer="Tailscale Inc.",
-            model=device.os,
-            name=device.name.split(".")[0],
-            sw_version=device.client_version,
-        )
diff --git a/homeassistant/components/tailscale/binary_sensor.py b/homeassistant/components/tailscale/binary_sensor.py
index 7803a7eb472..981f871de09 100644
--- a/homeassistant/components/tailscale/binary_sensor.py
+++ b/homeassistant/components/tailscale/binary_sensor.py
@@ -17,8 +17,8 @@ from homeassistant.const import EntityCategory
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from . import TailscaleEntity
 from .const import DOMAIN
+from .entity import TailscaleEntity
 
 
 @dataclass(frozen=True, kw_only=True)
diff --git a/homeassistant/components/tailscale/entity.py b/homeassistant/components/tailscale/entity.py
new file mode 100644
index 00000000000..a14b873a00f
--- /dev/null
+++ b/homeassistant/components/tailscale/entity.py
@@ -0,0 +1,52 @@
+"""The Tailscale integration."""
+
+from __future__ import annotations
+
+from tailscale import Device as TailscaleDevice
+
+from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
+from homeassistant.helpers.entity import EntityDescription
+from homeassistant.helpers.update_coordinator import (
+    CoordinatorEntity,
+    DataUpdateCoordinator,
+)
+
+from .const import DOMAIN
+
+
+class TailscaleEntity(CoordinatorEntity):
+    """Defines a Tailscale base entity."""
+
+    _attr_has_entity_name = True
+
+    def __init__(
+        self,
+        *,
+        coordinator: DataUpdateCoordinator,
+        device: TailscaleDevice,
+        description: EntityDescription,
+    ) -> None:
+        """Initialize a Tailscale sensor."""
+        super().__init__(coordinator=coordinator)
+        self.entity_description = description
+        self.device_id = device.device_id
+        self._attr_unique_id = f"{device.device_id}_{description.key}"
+
+    @property
+    def device_info(self) -> DeviceInfo:
+        """Return the device info."""
+        device: TailscaleDevice = self.coordinator.data[self.device_id]
+
+        configuration_url = "https://login.tailscale.com/admin/machines/"
+        if device.addresses:
+            configuration_url += device.addresses[0]
+
+        return DeviceInfo(
+            configuration_url=configuration_url,
+            entry_type=DeviceEntryType.SERVICE,
+            identifiers={(DOMAIN, device.device_id)},
+            manufacturer="Tailscale Inc.",
+            model=device.os,
+            name=device.name.split(".")[0],
+            sw_version=device.client_version,
+        )
diff --git a/homeassistant/components/tailscale/sensor.py b/homeassistant/components/tailscale/sensor.py
index 99b91d17442..fa4c966a7d7 100644
--- a/homeassistant/components/tailscale/sensor.py
+++ b/homeassistant/components/tailscale/sensor.py
@@ -18,8 +18,8 @@ from homeassistant.const import EntityCategory
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
-from . import TailscaleEntity
 from .const import DOMAIN
+from .entity import TailscaleEntity
 
 
 @dataclass(frozen=True, kw_only=True)