mirror of
https://github.com/home-assistant/core.git
synced 2025-11-04 08:29:37 +00:00
78 lines
2.5 KiB
Python
78 lines
2.5 KiB
Python
"""Airgradient Update platform."""
|
|
|
|
from datetime import timedelta
|
|
import logging
|
|
|
|
from airgradient import AirGradientConnectionError
|
|
from propcache.api import cached_property
|
|
|
|
from homeassistant.components.update import UpdateDeviceClass, UpdateEntity
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
|
|
|
|
from . import AirGradientConfigEntry, AirGradientCoordinator
|
|
from .entity import AirGradientEntity
|
|
|
|
PARALLEL_UPDATES = 1
|
|
SCAN_INTERVAL = timedelta(hours=1)
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
config_entry: AirGradientConfigEntry,
|
|
async_add_entities: AddConfigEntryEntitiesCallback,
|
|
) -> None:
|
|
"""Set up Airgradient update platform."""
|
|
|
|
coordinator = config_entry.runtime_data
|
|
|
|
async_add_entities([AirGradientUpdate(coordinator)], True)
|
|
|
|
|
|
class AirGradientUpdate(AirGradientEntity, UpdateEntity):
|
|
"""Representation of Airgradient Update."""
|
|
|
|
_attr_device_class = UpdateDeviceClass.FIRMWARE
|
|
_server_unreachable_logged = False
|
|
|
|
def __init__(self, coordinator: AirGradientCoordinator) -> None:
|
|
"""Initialize the entity."""
|
|
super().__init__(coordinator)
|
|
self._attr_unique_id = f"{coordinator.serial_number}-update"
|
|
|
|
@cached_property
|
|
def should_poll(self) -> bool:
|
|
"""Return True because we need to poll the latest version."""
|
|
return True
|
|
|
|
@property
|
|
def installed_version(self) -> str:
|
|
"""Return the installed version of the entity."""
|
|
return self.coordinator.data.measures.firmware_version
|
|
|
|
@property
|
|
def available(self) -> bool:
|
|
"""Return if entity is available."""
|
|
return super().available and self._attr_available
|
|
|
|
async def async_update(self) -> None:
|
|
"""Update the entity."""
|
|
try:
|
|
self._attr_latest_version = (
|
|
await self.coordinator.client.get_latest_firmware_version(
|
|
self.coordinator.serial_number
|
|
)
|
|
)
|
|
except AirGradientConnectionError:
|
|
self._attr_latest_version = None
|
|
self._attr_available = False
|
|
if not self._server_unreachable_logged:
|
|
_LOGGER.error(
|
|
"Unable to connect to AirGradient server to check for updates"
|
|
)
|
|
self._server_unreachable_logged = True
|
|
else:
|
|
self._server_unreachable_logged = False
|
|
self._attr_available = True
|