coordinator.py

Go to the documentation of this file.

"""DataUpdateCoordinator for LaCrosse View."""
 
from __future__ import annotations
 
from datetime import timedelta
import logging
from time import time
 
from lacrosse_view import HTTPError, LaCrosse, Location, LoginError, Sensor
 
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.exceptions import ConfigEntryAuthFailed, ConfigEntryNotReady
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
 
from .const import SCAN_INTERVAL
 
_LOGGER = logging.getLogger(__name__)
 
 
class LaCrosseUpdateCoordinator(DataUpdateCoordinator[list[Sensor]]):
    """DataUpdateCoordinator for LaCrosse View."""
 
    username: str
    password: str
    name: str
    id: str
    hass: HomeAssistant
 
    def __init__(
        self,
        hass: HomeAssistant,
        api: LaCrosse,
        entry: ConfigEntry,
    ) -> None:
        """Initialize DataUpdateCoordinator for LaCrosse View."""
        self.apiapi = api
        self.last_updatelast_update = time()
        self.usernameusername = entry.data["username"]
        self.passwordpassword = entry.data["password"]
        self.hasshasshass = hass
        self.namenamename = entry.data["name"]
        self.idid = entry.data["id"]
        super().__init__(
            hass,
            _LOGGER,
            name="LaCrosse View",
            update_interval=timedelta(seconds=SCAN_INTERVAL),
        )
 
    async def _async_update_data(self) -> list[Sensor]:
        """Get the data for LaCrosse View."""
        now = int(time())
 
        if self.last_updatelast_update < now - 59 * 60:  # Get new token once in a hour
            _LOGGER.debug("Refreshing token")
            self.last_updatelast_update = now
            try:
                await self.apiapi.login(self.usernameusername, self.passwordpassword)
            except LoginError as error:
                raise ConfigEntryAuthFailed from error
 
        try:
            # Fetch last hour of data
            sensors = await self.apiapi.get_sensors(
                location=Location(id=self.idid, name=self.namenamename),
                tz=self.hasshasshass.config.time_zone,
                start=str(now - 3600),
                end=str(now),
            )
        except HTTPError as error:
            raise ConfigEntryNotReady from error
 
        _LOGGER.debug("Got data: %s", sensors)
 
        # Verify that we have permission to read the sensors
        for sensor in sensors:
            if not sensor.permissions.get("read", False):
                raise ConfigEntryAuthFailed(
                    f"This account does not have permission to read {sensor.name}"
                )
 
        return sensors