From 052c3fcb6a3d2ba1b22b81daa3cadf9bc765eebf Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 27 Apr 2022 09:57:18 +0200 Subject: [PATCH] Use shorthand attributes in xiaomi_miio vacuum (#70848) --- .../components/xiaomi_miio/vacuum.py | 21 +++++++------------ 1 file changed, 7 insertions(+), 14 deletions(-) diff --git a/homeassistant/components/xiaomi_miio/vacuum.py b/homeassistant/components/xiaomi_miio/vacuum.py index a144794ff93..6d398ff40b9 100644 --- a/homeassistant/components/xiaomi_miio/vacuum.py +++ b/homeassistant/components/xiaomi_miio/vacuum.py @@ -227,12 +227,12 @@ class MiroboVacuum( return self._state @property - def battery_level(self): + def battery_level(self) -> int: """Return the battery level of the vacuum cleaner.""" return self.coordinator.data.status.battery @property - def fan_speed(self): + def fan_speed(self) -> str: """Return the fan speed of the vacuum cleaner.""" speed = self.coordinator.data.status.fanspeed if speed in self.coordinator.data.fan_speeds_reverse: @@ -240,16 +240,14 @@ class MiroboVacuum( _LOGGER.debug("Unable to find reverse for %s", speed) - return speed + return str(speed) @property - def fan_speed_list(self): + def fan_speed_list(self) -> list[str]: """Get the list of available fan speed steps of the vacuum cleaner.""" - return ( - list(self.coordinator.data.fan_speeds) - if self.coordinator.data.fan_speeds - else [] - ) + if speed_list := self.coordinator.data.fan_speeds: + return list(speed_list) + return [] @property def timers(self): @@ -276,11 +274,6 @@ class MiroboVacuum( attrs[ATTR_TIMERS] = self.timers return attrs - @property - def supported_features(self): - """Flag vacuum cleaner robot features that are supported.""" - return self._attr_supported_features - async def _try_command(self, mask_error, func, *args, **kwargs): """Call a vacuum command handling error messages.""" try: