update.py

Go to the documentation of this file.

"""Update platform for Tessie integration."""
 
from __future__ import annotations
 
from typing import Any
 
from tessie_api import schedule_software_update
 
from homeassistant.components.update import UpdateEntity, UpdateEntityFeature
from homeassistant.core import HomeAssistant
from homeassistant.helpers.entity_platform import AddEntitiesCallback
 
from . import TessieConfigEntry
from .const import TessieUpdateStatus
from .entity import TessieEntity
from .models import TessieVehicleData
 
PARALLEL_UPDATES = 0
 
 
async def async_setup_entry(
    hass: HomeAssistant,
    entry: TessieConfigEntry,
    async_add_entities: AddEntitiesCallback,
) -> None:
    """Set up the Tessie Update platform from a config entry."""
    data = entry.runtime_data
 
    async_add_entities(TessieUpdateEntity(vehicle) for vehicle in data.vehicles)
 
 
class TessieUpdateEntity(TessieEntity, UpdateEntity):
    """Tessie Updates entity."""
 
    _attr_supported_features = UpdateEntityFeature.PROGRESS
 
    def __init__(
        self,
        vehicle: TessieVehicleData,
    ) -> None:
        """Initialize the Update."""
        super().__init__(vehicle, "update")
 
    @property
    def supported_features(self) -> UpdateEntityFeature:
        """Flag supported features."""
        if self.getget("vehicle_state_software_update_status") in (
            TessieUpdateStatus.AVAILABLE,
            TessieUpdateStatus.SCHEDULED,
        ):
            return self._attr_supported_features_attr_supported_features | UpdateEntityFeature.INSTALL
        return self._attr_supported_features_attr_supported_features
 
    @property
    def installed_version(self) -> str:
        """Return the current app version."""
        # Discard build from version number
        return self.coordinator.data["vehicle_state_car_version"].split(" ")[0]
 
    @property
    def latest_version(self) -> str | None:
        """Return the latest version."""
        if self.getget("vehicle_state_software_update_status") in (
            TessieUpdateStatus.AVAILABLE,
            TessieUpdateStatus.SCHEDULED,
            TessieUpdateStatus.INSTALLING,
            TessieUpdateStatus.DOWNLOADING,
            TessieUpdateStatus.WIFI_WAIT,
        ):
            return self.getget("vehicle_state_software_update_version")
        return self.installed_versioninstalled_versioninstalled_versioninstalled_version
 
    @property
    def in_progress(self) -> bool:
        """Update installation progress."""
        return (
            self.getget("vehicle_state_software_update_status")
            == TessieUpdateStatus.INSTALLING
        )
 
    @property
    def update_percentage(self) -> int | None:
        """Update installation progress."""
        if (
            self.getget("vehicle_state_software_update_status")
            == TessieUpdateStatus.INSTALLING
        ):
            return self.getget("vehicle_state_software_update_install_perc")
        return None
 
    async def async_install(
        self, version: str | None, backup: bool, **kwargs: Any
    ) -> None:
        """Install an update."""
        await self.runrun(schedule_software_update, in_seconds=0)
        self.setset(
            ("vehicle_state_software_update_status", TessieUpdateStatus.INSTALLING)
        )