Initial commit

custom_components/oasis_mini/entity.py

@@ -0,0 +1,47 @@
+"""Oasis Mini entity."""
+
+from __future__ import annotations
+
+import logging
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.helpers.entity import DeviceInfo, EntityDescription
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from .const import DOMAIN
+from .coordinator import OasisMiniCoordinator
+from .pyoasismini import OasisMini
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class OasisMiniEntity(CoordinatorEntity[OasisMiniCoordinator]):
+    """Base class for Oasis Mini entities."""
+
+    _attr_has_entity_name = True
+
+    def __init__(
+        self,
+        coordinator: OasisMiniCoordinator,
+        entry: ConfigEntry,
+        description: EntityDescription,
+    ) -> None:
+        """Construct a Oasis Mini entity."""
+        super().__init__(coordinator)
+        self.entity_description = description
+        serial_number = coordinator.device.serial_number
+        self._attr_unique_id = f"{serial_number}-{description.key}"
+
+        self._attr_device_info = DeviceInfo(
+            identifiers={(DOMAIN, serial_number)},
+            name=entry.title,
+            manufacturer="Kinetic Oasis",
+            model="Oasis Mini",
+            serial_number=serial_number,
+            sw_version=coordinator.device.software_version,
+        )
+
+    @property
+    def device(self) -> OasisMini:
+        """Return the device."""
+        return self.coordinator.device