Use UpdateEntityFeature in update platform (#3418)

fixes #3417
hacs · Dec 28, 2023 · f7fa3c2 · f7fa3c2
1 parent 98a1f52
commit f7fa3c2
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/custom_components/hacs/update.py b/custom_components/hacs/update.py
@@ -3,7 +3,7 @@
 
 from typing import Any
 
-from homeassistant.components.update import UpdateEntity
+from homeassistant.components.update import UpdateEntity, UpdateEntityFeature
 from homeassistant.core import HomeAssistantError, callback
 from homeassistant.helpers.dispatcher import async_dispatcher_connect
 
@@ -27,7 +27,12 @@ async def async_setup_entry(hass, _config_entry, async_add_devices):
 class HacsRepositoryUpdateEntity(HacsRepositoryEntity, UpdateEntity):
     """Update entities for repositories downloaded with HACS."""
 
-    _attr_supported_features = 1 | 2 | 4 | 16
+    _attr_supported_features = (
+        UpdateEntityFeature.INSTALL
+        | UpdateEntityFeature.SPECIFIC_VERSION
+        | UpdateEntityFeature.PROGRESS
+        | UpdateEntityFeature.RELEASE_NOTES
+    )
 
     @property
     def name(self) -> str | None: