Initial commit

custom_components/elegoo_spaghetti_detection/entity.py

@@ -0,0 +1,36 @@
+"""Base entities for Elegoo spaghetti detection."""
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.helpers.entity import DeviceInfo, Entity
+
+from .const import DOMAIN
+from .runtime import SpaghettiDetectorRuntime
+
+
+class SpaghettiDetectorEntity(Entity):
+    """Base entity for a detector runtime."""
+
+    _attr_has_entity_name = True
+    _attr_should_poll = False
+
+    def __init__(
+        self,
+        entry: ConfigEntry,
+        runtime: SpaghettiDetectorRuntime,
+        key: str,
+    ) -> None:
+        self.entry = entry
+        self.runtime = runtime
+        self._attr_unique_id = f"{entry.entry_id}_{key}"
+        self._attr_device_info = DeviceInfo(
+            identifiers={(DOMAIN, entry.entry_id)},
+            manufacturer="Elegoo",
+            model="Spaghetti detection",
+            name=entry.title,
+        )
+
+    async def async_added_to_hass(self) -> None:
+        """Subscribe to runtime updates."""
+        self.async_on_remove(
+            self.runtime.async_add_listener(self.async_write_ha_state)
+        )