"""The Renson integration.""" from __future__ import annotations import asyncio from dataclasses import dataclass from datetime import timedelta import logging from typing import Any from renson_endura_delta.renson import RensonVentilation from homeassistant.config_entries import ConfigEntry from homeassistant.const import CONF_HOST, Platform from homeassistant.core import HomeAssistant from homeassistant.exceptions import ConfigEntryNotReady from homeassistant.helpers.update_coordinator import DataUpdateCoordinator from .const import DOMAIN _LOGGER = logging.getLogger(__name__) PLATFORMS = [ Platform.BINARY_SENSOR, Platform.FAN, Platform.NUMBER, Platform.SENSOR, ] @dataclass class RensonData: """Renson data class.""" api: RensonVentilation coordinator: RensonCoordinator async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: """Set up Renson from a config entry.""" api = RensonVentilation(entry.data[CONF_HOST]) coordinator = RensonCoordinator("Renson", hass, api) if not await hass.async_add_executor_job(api.connect): raise ConfigEntryNotReady("Cannot connect to Renson device") await coordinator.async_config_entry_first_refresh() hass.data.setdefault(DOMAIN, {})[entry.entry_id] = RensonData( api, coordinator, ) await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS) return True async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool: """Unload a config entry.""" if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS): hass.data[DOMAIN].pop(entry.entry_id) return unload_ok class RensonCoordinator(DataUpdateCoordinator): """Data update coordinator for Renson.""" def __init__( self, name: str, hass: HomeAssistant, api: RensonVentilation, update_interval=timedelta(seconds=30), ) -> None: """Initialize my coordinator.""" super().__init__( hass, _LOGGER, # Name of the data. For logging purposes. name=name, # Polling interval. Will only be polled if there are subscribers. update_interval=update_interval, ) self.api = api async def _async_update_data(self) -> dict[str, Any]: """Fetch data from API endpoint.""" async with asyncio.timeout(30): return await self.hass.async_add_executor_job(self.api.get_all_data)