From bed3fcfd434bd276857413c8f5b35a6ebb6501c6 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Mon, 23 Sep 2024 08:47:23 +0200
Subject: [PATCH] Move cert_expiry base entity to separate module (#126478)

---
 .../components/cert_expiry/entity.py          | 23 +++++++++++++++++++
 .../components/cert_expiry/sensor.py          | 20 +++-------------
 2 files changed, 26 insertions(+), 17 deletions(-)
 create mode 100644 homeassistant/components/cert_expiry/entity.py

diff --git a/homeassistant/components/cert_expiry/entity.py b/homeassistant/components/cert_expiry/entity.py
new file mode 100644
index 00000000000..f412f16fba8
--- /dev/null
+++ b/homeassistant/components/cert_expiry/entity.py
@@ -0,0 +1,23 @@
+"""Counter for the days until an HTTPS (TLS) certificate will expire."""
+
+from __future__ import annotations
+
+from typing import Any
+
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from .coordinator import CertExpiryDataUpdateCoordinator
+
+
+class CertExpiryEntity(CoordinatorEntity[CertExpiryDataUpdateCoordinator]):
+    """Defines a base Cert Expiry entity."""
+
+    _attr_has_entity_name = True
+
+    @property
+    def extra_state_attributes(self) -> dict[str, Any]:
+        """Return additional sensor state attributes."""
+        return {
+            "is_valid": self.coordinator.is_cert_valid,
+            "error": str(self.coordinator.cert_error),
+        }
diff --git a/homeassistant/components/cert_expiry/sensor.py b/homeassistant/components/cert_expiry/sensor.py
index f52ff8a40d8..a6f163b51be 100644
--- a/homeassistant/components/cert_expiry/sensor.py
+++ b/homeassistant/components/cert_expiry/sensor.py
@@ -3,7 +3,6 @@
 from __future__ import annotations
 
 from datetime import datetime, timedelta
-from typing import Any
 
 import voluptuous as vol
 
@@ -20,10 +19,11 @@ from homeassistant.helpers.device_registry import DeviceEntryType, DeviceInfo
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.event import async_call_later
 from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
-from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
-from . import CertExpiryConfigEntry, CertExpiryDataUpdateCoordinator
+from . import CertExpiryConfigEntry
 from .const import DEFAULT_PORT, DOMAIN
+from .coordinator import CertExpiryDataUpdateCoordinator
+from .entity import CertExpiryEntity
 
 SCAN_INTERVAL = timedelta(hours=12)
 
@@ -73,20 +73,6 @@ async def async_setup_entry(
     async_add_entities(sensors, True)
 
 
-class CertExpiryEntity(CoordinatorEntity[CertExpiryDataUpdateCoordinator]):
-    """Defines a base Cert Expiry entity."""
-
-    _attr_has_entity_name = True
-
-    @property
-    def extra_state_attributes(self) -> dict[str, Any]:
-        """Return additional sensor state attributes."""
-        return {
-            "is_valid": self.coordinator.is_cert_valid,
-            "error": str(self.coordinator.cert_error),
-        }
-
-
 class SSLCertificateTimestamp(CertExpiryEntity, SensorEntity):
     """Implementation of the Cert Expiry timestamp sensor."""