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

Use faster contains check in update (#106436)

Browse Source
This commit is contained in:
J. Nick Koston 2023-12-26 21:45:51 -10:00 committed by GitHub
parent 6dbfd70e30
commit 7a2a99db2e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
homeassistant/components/update/__init__.py
View File

@ -140,7 +140,7 @@ async def async_install(entity: UpdateEntity, service_call: ServiceCall) -> None
# If version is specified, but not supported by the entity. # If version is specified, but not supported by the entity.
if ( if (
version is not None version is not None
and not entity.supported_features & UpdateEntityFeature.SPECIFIC_VERSION and UpdateEntityFeature.SPECIFIC_VERSION not in entity.supported_features
): ):
raise HomeAssistantError( raise HomeAssistantError(
f"Installing a specific version is not supported for {entity.entity_id}" f"Installing a specific version is not supported for {entity.entity_id}"
@ -149,7 +149,7 @@ async def async_install(entity: UpdateEntity, service_call: ServiceCall) -> None
# If backup is requested, but not supported by the entity. # If backup is requested, but not supported by the entity.
if ( if (
backup := service_call.data[ATTR_BACKUP] backup := service_call.data[ATTR_BACKUP]
) and not entity.supported_features & UpdateEntityFeature.BACKUP: ) and UpdateEntityFeature.BACKUP not in entity.supported_features:
raise HomeAssistantError(f"Backup is not supported for {entity.entity_id}") raise HomeAssistantError(f"Backup is not supported for {entity.entity_id}")
# Update is already in progress. # Update is already in progress.
@ -263,7 +263,7 @@ class UpdateEntity(
return self._attr_entity_category return self._attr_entity_category
if hasattr(self, "entity_description"): if hasattr(self, "entity_description"):
return self.entity_description.entity_category return self.entity_description.entity_category
if self.supported_features & UpdateEntityFeature.INSTALL: if UpdateEntityFeature.INSTALL in self.supported_features:
return EntityCategory.CONFIG return EntityCategory.CONFIG
return EntityCategory.DIAGNOSTIC return EntityCategory.DIAGNOSTIC
@ -408,7 +408,7 @@ class UpdateEntity(
# If entity supports progress, return the in_progress value. # If entity supports progress, return the in_progress value.
# Otherwise, we use the internal progress value. # Otherwise, we use the internal progress value.
if self.supported_features & UpdateEntityFeature.PROGRESS: if UpdateEntityFeature.PROGRESS in self.supported_features:
in_progress = self.in_progress in_progress = self.in_progress
else: else:
in_progress = self.__in_progress in_progress = self.__in_progress
@ -444,7 +444,7 @@ class UpdateEntity(
Handles setting the in_progress state in case the entity doesn't Handles setting the in_progress state in case the entity doesn't
support it natively. support it natively.
""" """
if not self.supported_features & UpdateEntityFeature.PROGRESS: if UpdateEntityFeature.PROGRESS not in self.supported_features:
self.__in_progress = True self.__in_progress = True
self.async_write_ha_state() self.async_write_ha_state()
@ -490,7 +490,7 @@ async def websocket_release_notes(
) )
return return
if not entity.supported_features & UpdateEntityFeature.RELEASE_NOTES: if UpdateEntityFeature.RELEASE_NOTES not in entity.supported_features:
connection.send_error( connection.send_error(
msg["id"], msg["id"],
websocket_api.const.ERR_NOT_SUPPORTED, websocket_api.const.ERR_NOT_SUPPORTED,