jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-30 08:47:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Restore latest version of ZHA Update entity

Browse Source 
Required for https://github.com/zigpy/zha/pull/460
This commit is contained in:
abmantis 2025-06-09 18:01:57 +01:00
parent 0cce4d1b81
commit 0040e9e5af
1 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
homeassistant/components/zha/update.py
View File

@ -10,13 +10,14 @@ from zha.exceptions import ZHAException
from zigpy.application import ControllerApplication from zigpy.application import ControllerApplication
from homeassistant.components.update import ( from homeassistant.components.update import (
ATTR_LATEST_VERSION,
UpdateDeviceClass, UpdateDeviceClass,
UpdateEntity, UpdateEntity,
UpdateEntityFeature, UpdateEntityFeature,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import Platform from homeassistant.const import Platform
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant, State, callback
from homeassistant.exceptions import HomeAssistantError from homeassistant.exceptions import HomeAssistantError
from homeassistant.helpers.dispatcher import async_dispatcher_connect from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback from homeassistant.helpers.entity_platform import AddConfigEntryEntitiesCallback
@ -195,3 +196,10 @@ class ZHAFirmwareUpdateEntity(
"""Update the entity.""" """Update the entity."""
await CoordinatorEntity.async_update(self) await CoordinatorEntity.async_update(self)
await super().async_update() await super().async_update()
@callback
def restore_external_state_attributes(self, state: State) -> None:
"""Restore entity state."""
self.entity_data.entity.restore_external_state_attributes(
latest_version=state.attributes.get(ATTR_LATEST_VERSION),
)