"""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()