55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
|
"""Data update coordinator for Version entities."""
|
||
|
|
from __future__ import annotations
|
||
|
|
|
||
|
|
from typing import Any
|
||
|
|
|
||
|
|
from awesomeversion import AwesomeVersion
|
||
|
|
from pyhaversion import HaVersion
|
||
|
|
from pyhaversion.exceptions import HaVersionException
|
||
|
|
|
||
|
|
from homeassistant.config_entries import ConfigEntry
|
||
|
|
from homeassistant.core import HomeAssistant
|
||
|
|
from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
|
||
|
|
|
||
|
|
from .const import DOMAIN, LOGGER, UPDATE_COORDINATOR_UPDATE_INTERVAL
|
||
|
|
|
||
|
|
|
||
|
|
class VersionDataUpdateCoordinator(DataUpdateCoordinator):
|
||
|
|
"""Data update coordinator for Version entities."""
|
||
|
|
|
||
|
|
config_entry: ConfigEntry
|
||
|
|
|
||
|
|
def __init__(
|
||
|
|
self,
|
||
|
|
hass: HomeAssistant,
|
||
|
|
api: HaVersion,
|
||
|
|
) -> None:
|
||
|
|
"""Initialize the coordinator."""
|
||
|
|
super().__init__(
|
||
|
|
hass=hass,
|
||
|
|
logger=LOGGER,
|
||
|
|
name=DOMAIN,
|
||
|
|
update_method=self._async_update_version_data,
|
||
|
|
update_interval=UPDATE_COORDINATOR_UPDATE_INTERVAL,
|
||
|
|
)
|
||
|
|
self._api = api
|
||
|
|
self._version: AwesomeVersion | None = None
|
||
|
|
self._version_data: dict[str, Any] | None = None
|
||
|
|
|
||
|
|
@property
|
||
|
|
def version(self) -> str | None:
|
||
|
|
"""Return the latest version."""
|
||
|
|
return str(self._version) if self._version else None
|
||
|
|
|
||
|
|
@property
|
||
|
|
def version_data(self) -> dict[str, Any]:
|
||
|
|
"""Return the version data."""
|
||
|
|
return self._version_data or {}
|
||
|
|
|
||
|
|
async def _async_update_version_data(self) -> None:
|
||
|
|
"""Update version data."""
|
||
|
|
try:
|
||
|
|
self._version, self._version_data = await self._api.get_version()
|
||
|
|
except HaVersionException as exception:
|
||
|
|
raise UpdateFailed(exception) from exception
|