Add Tasmota firmware update availability support

homeassistant/components/tasmota/const.py

@@ -19,6 +19,7 @@ PLATFORMS = [
     Platform.LIGHT,
     Platform.SENSOR,
     Platform.SWITCH,
+    Platform.UPDATE,
 ]
 
 TASMOTA_EVENT = "tasmota_event"

homeassistant/components/tasmota/coordinator.py

@@ -0,0 +1,38 @@
+"""Data update coordinators for Tasmota."""
+
+from datetime import timedelta
+import logging
+
+from aiogithubapi import GitHubAPI, GitHubRatelimitException, GitHubReleaseModel
+from aiogithubapi.client import GitHubConnectionException
+
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.aiohttp_client import async_get_clientsession
+from homeassistant.helpers.update_coordinator import DataUpdateCoordinator, UpdateFailed
+
+
+class TasmotaLatestReleaseUpdateCoordinator(DataUpdateCoordinator[GitHubReleaseModel]):
+    """Data update coordinator for Tasmota latest release info."""
+
+    def __init__(self, hass: HomeAssistant, config_entry: ConfigEntry) -> None:
+        """Initialize the coordinator."""
+        self.client = GitHubAPI(session=async_get_clientsession(hass))
+        super().__init__(
+            hass,
+            logger=logging.getLogger(__name__),
+            config_entry=config_entry,
+            name="Tasmota latest release",
+            update_interval=timedelta(days=1),
+        )
+
+    async def _async_update_data(self) -> GitHubReleaseModel:
+        """Get new data."""
+        try:
+            response = await self.client.repos.releases.latest("arendst/Tasmota")
+            if response.data is None:
+                raise UpdateFailed("No data received")
+        except (GitHubConnectionException, GitHubRatelimitException) as ex:
+            raise UpdateFailed(ex) from ex
+        else:
+            return response.data

homeassistant/components/tasmota/manifest.json

@@ -8,5 +8,5 @@
   "iot_class": "local_push",
   "loggers": ["hatasmota"],
   "mqtt": ["tasmota/discovery/#"],
-  "requirements": ["HATasmota==0.10.1"]
+  "requirements": ["HATasmota==0.10.1", "aiogithubapi==24.6.0"]
 }

homeassistant/components/tasmota/update.py

@@ -0,0 +1,79 @@
+"""Update entity for Tasmota."""
+
+import re
+
+from homeassistant.components.update import (
+    UpdateDeviceClass,
+    UpdateEntity,
+    UpdateEntityFeature,
+)
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers import device_registry as dr
+from homeassistant.helpers.device_registry import DeviceEntry
+from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
+
+from .coordinator import TasmotaLatestReleaseUpdateCoordinator
+
+
+async def async_setup_entry(
+    hass: HomeAssistant,
+    config_entry: ConfigEntry,
+    async_add_entities: AddConfigEntryEntitiesCallback,
+) -> None:
+    """Set up Tasmota update entities."""
+    coordinator = TasmotaLatestReleaseUpdateCoordinator(hass, config_entry)
+    await coordinator.async_config_entry_first_refresh()
+
+    device_registry = dr.async_get(hass)
+    devices = device_registry.devices.get_devices_for_config_entry_id(
+        config_entry.entry_id
+    )
+    async_add_entities(TasmotaUpdateEntity(coordinator, device) for device in devices)
+
+
+class TasmotaUpdateEntity(UpdateEntity):
+    """Representation of a Tasmota update entity."""
+
+    _attr_device_class = UpdateDeviceClass.FIRMWARE
+    _attr_name = "Firmware"
+    _attr_title = "Tasmota firmware"
+    _attr_supported_features = UpdateEntityFeature.RELEASE_NOTES
+
+    def __init__(
+        self,
+        coordinator: TasmotaLatestReleaseUpdateCoordinator,
+        device_entry: DeviceEntry,
+    ) -> None:
+        """Initialize the Tasmota update entity."""
+        self.coordinator = coordinator
+        self.device_entry = device_entry
+        self._attr_unique_id = f"{device_entry.id}_update"
+
+    @property
+    def installed_version(self) -> str | None:
+        """Return the installed version."""
+        return self.device_entry.sw_version  # type:ignore[union-attr]
+
+    @property
+    def latest_version(self) -> str:
+        """Return the latest version."""
+        return self.coordinator.data.tag_name.removeprefix("v")
+
+    @property
+    def release_url(self) -> str:
+        """Return the release URL."""
+        return self.coordinator.data.html_url
+
+    @property
+    def release_summary(self) -> str:
+        """Return the release summary."""
+        return self.coordinator.data.name
+
+    def release_notes(self) -> str | None:
+        """Return the release notes."""
+        if not self.coordinator.data.body:
+            return None
+        return re.sub(
+            r"^<picture>.*?</picture>", "", self.coordinator.data.body, flags=re.DOTALL
+        )

requirements_all.txt

@@ -265,6 +265,7 @@ aioflo==2021.11.0
 aioftp==0.21.3
 
 # homeassistant.components.github
+# homeassistant.components.tasmota
 aiogithubapi==24.6.0
 
 # homeassistant.components.guardian

requirements_test_all.txt

@@ -253,6 +253,7 @@ aiofiles==24.1.0
 aioflo==2021.11.0
 
 # homeassistant.components.github
+# homeassistant.components.tasmota
 aiogithubapi==24.6.0
 
 # homeassistant.components.guardian