From 15e8f4614c1d518c3f07628cf25fbb0957c3a7ab Mon Sep 17 00:00:00 2001 From: ochlocracy <5885236+ochlocracy@users.noreply.github.com> Date: Tue, 12 May 2020 10:24:13 -0400 Subject: [PATCH] Return fan_speed_list based on SUPPORT_FAN_SPEED. (#35347) --- homeassistant/components/vacuum/__init__.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/homeassistant/components/vacuum/__init__.py b/homeassistant/components/vacuum/__init__.py index 7ef2e3889a9..111fa64f988 100644 --- a/homeassistant/components/vacuum/__init__.py +++ b/homeassistant/components/vacuum/__init__.py @@ -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