home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-07 06:35:10 +02:00
Code Issues Packages Projects Releases Wiki Activity

Enforce UpdateEntityFeature (#82465)

Browse Source
This commit is contained in:
epenet
2022-11-22 07:16:53 +01:00
committed by GitHub
parent ff5f1aee53
commit ab26ea06b9
3 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
homeassistant/components/mqtt/update.py
View File

@@ -253,9 +253,9 @@ class MqttUpdate(MqttEntity, UpdateEntity, RestoreEntity):
get_mqtt_data(self.hass).state_write_requests.write_state_request(self) get_mqtt_data(self.hass).state_write_requests.write_state_request(self)
@property @property
def supported_features(self) -> UpdateEntityFeature | int: def supported_features(self) -> UpdateEntityFeature:
"""Return the list of supported features.""" """Return the list of supported features."""
support = 0 support = UpdateEntityFeature(0)
if self._config.get(CONF_COMMAND_TOPIC) is not None: if self._config.get(CONF_COMMAND_TOPIC) is not None:
support |= UpdateEntityFeature.INSTALL support |= UpdateEntityFeature.INSTALL

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

@@ -190,7 +190,7 @@ class UpdateEntity(RestoreEntity):
_attr_release_summary: str | None = None _attr_release_summary: str | None = None
_attr_release_url: str | None = None _attr_release_url: str | None = None
_attr_state: None = None _attr_state: None = None
_attr_supported_features: UpdateEntityFeature | int = 0 _attr_supported_features: UpdateEntityFeature = UpdateEntityFeature(0)
_attr_title: str | None = None _attr_title: str | None = None
__skipped_version: str | None = None __skipped_version: str | None = None
__in_progress: bool = False __in_progress: bool = False
@@ -270,7 +270,7 @@ class UpdateEntity(RestoreEntity):
return self._attr_release_url return self._attr_release_url
@property @property
def supported_features(self) -> UpdateEntityFeature | int: def supported_features(self) -> UpdateEntityFeature:
"""Flag supported features.""" """Flag supported features."""
return self._attr_supported_features return self._attr_supported_features

2
pylint/plugins/hass_enforce_type_hints.py
View File

@@ -2345,7 +2345,7 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
), ),
TypeHintMatch( TypeHintMatch(
function_name="supported_features", function_name="supported_features",
return_type=["UpdateEntityFeature", "int"], return_type="UpdateEntityFeature",
), ),
TypeHintMatch( TypeHintMatch(
function_name="title", function_name="title",