From 173a11893a04a6e0d03bf45f60cf8889eea76c57 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Tue, 26 Apr 2022 18:30:20 +0200 Subject: [PATCH] Add short-hand attributes to vacuum (#69417) --- homeassistant/components/vacuum/__init__.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/homeassistant/components/vacuum/__init__.py b/homeassistant/components/vacuum/__init__.py index 7820f09704c..74636e82e69 100644 --- a/homeassistant/components/vacuum/__init__.py +++ b/homeassistant/components/vacuum/__init__.py @@ -176,30 +176,36 @@ class _BaseVacuum(Entity): Contains common properties and functions for all vacuum devices. """ + _attr_battery_icon: str + _attr_battery_level: int | None = None + _attr_fan_speed: str | None = None + _attr_fan_speed_list: list[str] + _attr_supported_features: int + @property def supported_features(self) -> int: """Flag vacuum cleaner features that are supported.""" - raise NotImplementedError() + return self._attr_supported_features @property def battery_level(self) -> int | None: """Return the battery level of the vacuum cleaner.""" - return None + return self._attr_battery_level @property def battery_icon(self) -> str: """Return the battery icon for the vacuum cleaner.""" - raise NotImplementedError() + return self._attr_battery_icon @property def fan_speed(self) -> str | None: """Return the fan speed of the vacuum cleaner.""" - return None + return self._attr_fan_speed @property def fan_speed_list(self) -> list[str]: """Get the list of available fan speed steps of the vacuum cleaner.""" - raise NotImplementedError() + return self._attr_fan_speed_list @property def capability_attributes(self) -> Mapping[str, Any] | None: