Add update platform to myuplink (#109786)

* Add update platform to myuplink

* Address comments from review
This commit is contained in:
Åke Strandberg 2024-02-06 23:14:31 +01:00 committed by GitHub
parent 59e9010b65
commit 9250dd0355
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 74 additions and 1 deletions

View File

@ -16,7 +16,7 @@ from .api import AsyncConfigEntryAuth
from .const import DOMAIN
from .coordinator import MyUplinkDataCoordinator
PLATFORMS: list[Platform] = [Platform.SENSOR]
PLATFORMS: list[Platform] = [Platform.SENSOR, Platform.UPDATE]
async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> bool:

View File

@ -0,0 +1,73 @@
"""Update entity for myUplink."""
from typing import cast
from homeassistant.components.update import (
UpdateDeviceClass,
UpdateEntity,
UpdateEntityDescription,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import MyUplinkDataCoordinator
from .const import DOMAIN
from .entity import MyUplinkEntity
UPDATE_DESCRIPTION = UpdateEntityDescription(
key="update",
device_class=UpdateDeviceClass.FIRMWARE,
)
async def async_setup_entry(
hass: HomeAssistant,
config_entry: ConfigEntry,
async_add_entities: AddEntitiesCallback,
) -> None:
"""Set up update entity."""
entities: list[UpdateEntity] = []
coordinator: MyUplinkDataCoordinator = hass.data[DOMAIN][config_entry.entry_id]
# Setup update entities
for device_id in coordinator.data.devices:
entities.append(
MyUplinkDeviceUpdate(
coordinator=coordinator,
device_id=device_id,
entity_description=UPDATE_DESCRIPTION,
unique_id_suffix="upd",
)
)
async_add_entities(entities)
class MyUplinkDeviceUpdate(MyUplinkEntity, UpdateEntity):
"""Representation of a myUplink device update entity."""
def __init__(
self,
coordinator: MyUplinkDataCoordinator,
device_id: str,
entity_description: UpdateEntityDescription,
unique_id_suffix: str,
) -> None:
"""Initialize the update entity."""
super().__init__(
coordinator=coordinator,
device_id=device_id,
unique_id_suffix=unique_id_suffix,
)
self.entity_description = entity_description
@property
def installed_version(self) -> str | None:
"""Return installed_version."""
return cast(str, self.coordinator.data.devices[self.device_id].firmwareCurrent)
@property
def latest_version(self) -> str | None:
"""Return latest_version."""
return cast(str, self.coordinator.data.devices[self.device_id].firmwareDesired)