"""Blink Coordinator.""" from __future__ import annotations from datetime import timedelta import logging from typing import Any from aiohttp import ClientError from blinkpy.auth import BlinkTwoFARequiredError, UnauthorizedError from blinkpy.blinkpy import Blink from homeassistant.config_entries import ConfigEntry from homeassistant.core import HomeAssistant from homeassistant.exceptions import ( ConfigEntryAuthFailed, ConfigEntryError, ConfigEntryNotReady, ) from homeassistant.helpers.update_coordinator import DataUpdateCoordinator from .const import DOMAIN _LOGGER = logging.getLogger(__name__) SCAN_INTERVAL = 300 type BlinkConfigEntry = ConfigEntry[BlinkUpdateCoordinator] class BlinkUpdateCoordinator(DataUpdateCoordinator[dict[str, Any]]): """BlinkUpdateCoordinator - In charge of downloading the data for a site.""" config_entry: BlinkConfigEntry def __init__( self, hass: HomeAssistant, config_entry: BlinkConfigEntry, api: Blink ) -> None: """Initialize the data service.""" self.api = api super().__init__( hass, _LOGGER, config_entry=config_entry, name=DOMAIN, update_interval=timedelta(seconds=SCAN_INTERVAL), ) async def _async_setup(self): """Set up the coordinator.""" try: await self.api.start() except (ClientError, TimeoutError) as ex: raise ConfigEntryNotReady("Can not connect to host") from ex except (BlinkTwoFARequiredError, UnauthorizedError) as ex: raise ConfigEntryAuthFailed("Required Blink re-authentication") from ex except Exception as ex: raise ConfigEntryError("Unknown error connecting to Blink") from ex if not self.api.available: raise ConfigEntryNotReady async def _async_update_data(self) -> dict[str, Any]: """Async update wrapper.""" try: return await self.api.refresh(force=True) except UnauthorizedError as ex: raise ConfigEntryAuthFailed("Blink API authorization failed") from ex