First commit

custom_components/daily/entity.py

@@ -0,0 +1,40 @@
+"""Daily Sensor class."""
+from homeassistant.helpers.restore_state import RestoreEntity
+import logging
+
+_LOGGER = logging.getLogger(__name__)
+
+
+class DailySensorEntity(RestoreEntity):
+    """Daily Sensor Entity."""
+
+    def __init__(self, coordinator, config_entry):
+        """Initialize dailysensorentity."""
+        self.coordinator = coordinator
+        self.config_entry = config_entry
+        self.entity_id = f"sensor.{coordinator.name}"
+
+    @property
+    def should_poll(self):
+        """No need to poll. Coordinator notifies entity of updates."""
+        return False
+
+    @property
+    def available(self):
+        """Return if entity is available."""
+        return self.coordinator.last_update_success
+
+    @property
+    def unique_id(self):
+        """Return a unique ID to use for this entity."""
+        return self.config_entry.entry_id
+
+    async def async_added_to_hass(self):
+        """Connect to dispatcher listening for entity data notifications."""
+        self.async_on_remove(
+            self.coordinator.async_add_listener(self.async_write_ha_state)
+        )
+
+    async def async_update(self):
+        """Update Coordinator entity."""
+        await self.coordinator.async_request_refresh()