python

custom_components/ttlock/entity.py

@@ -0,0 +1,39 @@
+"""Base entity class for TTLock integration."""
+from __future__ import annotations
+
+from abc import ABC, abstractmethod
+
+from homeassistant.core import callback
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from .coordinator import LockUpdateCoordinator
+
+
+class BaseLockEntity(CoordinatorEntity[LockUpdateCoordinator], ABC):
+    """Abstract base class for lock entity."""
+
+    coordinator: LockUpdateCoordinator
+
+    def __init__(
+        self,
+        coordinator: LockUpdateCoordinator,
+    ) -> None:
+        """Initialize."""
+        super().__init__(coordinator)
+        self._attr_device_info = coordinator.device_info
+        self._attr_unique_id = (
+            f"{coordinator.unique_id}-{self.__class__.__name__.lower()}"
+        )
+        self._update_from_coordinator()
+
+        # self.entity_description = description
+
+    @abstractmethod
+    def _update_from_coordinator(self) -> None:
+        pass
+
+    @callback
+    def _handle_coordinator_update(self) -> None:
+        """Handle updated data from the coordinator."""
+        self._update_from_coordinator()
+        self.async_write_ha_state()