mirror of
https://github.com/home-assistant/core
synced 2024-08-02 23:40:32 +02:00
48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
"""Coordinator for radiotherm."""
|
|
from __future__ import annotations
|
|
|
|
from datetime import timedelta
|
|
import logging
|
|
from socket import timeout
|
|
from urllib.error import URLError
|
|
|
|
from radiotherm.validate import RadiothermTstatError
|
|
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
|
|
|
|
from .data import RadioThermInitData, RadioThermUpdate, async_get_data
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
UPDATE_INTERVAL = timedelta(seconds=15)
|
|
|
|
|
|
class RadioThermUpdateCoordinator(DataUpdateCoordinator[RadioThermUpdate]):
|
|
"""DataUpdateCoordinator to gather data for radio thermostats."""
|
|
|
|
def __init__(self, hass: HomeAssistant, init_data: RadioThermInitData) -> None:
|
|
"""Initialize DataUpdateCoordinator."""
|
|
self.init_data = init_data
|
|
self._description = f"{init_data.name} ({init_data.host})"
|
|
super().__init__(
|
|
hass,
|
|
_LOGGER,
|
|
name=f"radiotherm {self.init_data.name}",
|
|
update_interval=UPDATE_INTERVAL,
|
|
)
|
|
|
|
async def _async_update_data(self) -> RadioThermUpdate:
|
|
"""Update data from the thermostat."""
|
|
try:
|
|
return await async_get_data(self.hass, self.init_data.tstat)
|
|
except RadiothermTstatError as ex:
|
|
msg = f"{self._description} was busy (invalid value returned): {ex}"
|
|
raise UpdateFailed(msg) from ex
|
|
except (OSError, URLError) as ex:
|
|
msg = f"{self._description} connection error: {ex}"
|
|
raise UpdateFailed(msg) from ex
|
|
except timeout as ex:
|
|
msg = f"{self._description}) timed out waiting for a response: {ex}"
|
|
raise UpdateFailed(msg) from ex
|