Introduce base entity in streamlabs water (#107095)

homeassistant/components/streamlabswater/entity.py

@@ -0,0 +1,31 @@
+"""Base entity for Streamlabs integration."""
+from homeassistant.core import DOMAIN
+from homeassistant.helpers.device_registry import DeviceInfo
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from .coordinator import StreamlabsCoordinator, StreamlabsData
+
+
+class StreamlabsWaterEntity(CoordinatorEntity[StreamlabsCoordinator]):
+    """Defines a base Streamlabs entity."""
+
+    _attr_has_entity_name = True
+
+    def __init__(
+        self,
+        coordinator: StreamlabsCoordinator,
+        location_id: str,
+        key: str,
+    ) -> None:
+        """Initialize the Streamlabs entity."""
+        super().__init__(coordinator)
+        self._location_id = location_id
+        self._attr_unique_id = f"{location_id}-{key}"
+        self._attr_device_info = DeviceInfo(
+            identifiers={(DOMAIN, location_id)}, name=self.location_data.name
+        )
+
+    @property
+    def location_data(self) -> StreamlabsData:
+        """Returns the data object."""
+        return self.coordinator.data[self._location_id]