home-assistant/core
1
0
Fork 1
mirror of https://github.com/home-assistant/core.git synced 2025-08-03 12:45:28 +02:00
Code Issues Packages Projects Releases Wiki Activity

Return fan_speed_list based on SUPPORT_FAN_SPEED. (#35347)

Browse Source
This commit is contained in:
ochlocracy
2020-05-12 10:24:13 -04:00
committed by GitHub
parent 53771bd576
commit 15e8f4614c
1 changed files with 3 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
homeassistant/components/vacuum/__init__.py
View File

@@ -249,7 +249,7 @@ class VacuumEntity(_BaseVacuum, ToggleEntity):
@property
def capability_attributes(self):
"""Return capability attributes."""
if self.fan_speed is not None:
if self.supported_features & SUPPORT_FAN_SPEED:
return {ATTR_FAN_SPEED_LIST: self.fan_speed_list}
@property
@@ -316,8 +316,7 @@ class VacuumDevice(VacuumEntity):
"""Print deprecation warning."""
super().__init_subclass__(**kwargs)
_LOGGER.warning(
"VacuumDevice is deprecated, modify %s to extend VacuumEntity",
cls.__name__,
"VacuumDevice is deprecated, modify %s to extend VacuumEntity", cls.__name__
)
@@ -341,7 +340,7 @@ class StateVacuumEntity(_BaseVacuum):
@property
def capability_attributes(self):
"""Return capability attributes."""
if self.fan_speed is not None:
if self.supported_features & SUPPORT_FAN_SPEED:
return {ATTR_FAN_SPEED_LIST: self.fan_speed_list}
@property