Use UpdateEntityStateAttribute enum in Shelly (#175995)

Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
epenet
2026-07-08 17:23:10 +02:00
committed by GitHub
co-authored by Claude Opus 4.8
parent 642b079482
commit bd575bbbce
+5 -4
View File
@@ -10,12 +10,11 @@ from aioshelly.exceptions import DeviceConnectionError, InvalidAuthError, RpcCal
from awesomeversion import AwesomeVersion, AwesomeVersionStrategy
from homeassistant.components.update import (
ATTR_INSTALLED_VERSION,
ATTR_LATEST_VERSION,
UpdateDeviceClass,
UpdateEntity,
UpdateEntityDescription,
UpdateEntityFeature,
UpdateEntityStateAttribute,
)
from homeassistant.const import EntityCategory
from homeassistant.core import HomeAssistant, callback
@@ -391,7 +390,9 @@ class RpcSleepingUpdateEntity(
if self.last_state is None:
return None
return self.last_state.attributes.get(ATTR_INSTALLED_VERSION)
return self.last_state.attributes.get(
UpdateEntityStateAttribute.INSTALLED_VERSION
)
@property
@override
@@ -407,7 +408,7 @@ class RpcSleepingUpdateEntity(
if self.last_state is None:
return None
return self.last_state.attributes.get(ATTR_LATEST_VERSION)
return self.last_state.attributes.get(UpdateEntityStateAttribute.LATEST_VERSION)
@property
@override