Move deprecated battery level from vacuum to sensor #754

Author: AlexxIT

custom_components/yandex_station/sensor.py

@@ -48,6 +48,7 @@
     "devices.types.remote.ir",  # temperature, humidity
     "devices.types.smart_speaker.yandex.station.pickle",  # co2_level, temp., hum.
     "devices.types.smart_speaker.yandex.station.plum",  # battery
+    "devices.types.vacuum_cleaner",  # battery_level
 )
 INCLUDE_PROPERTIES = ("devices.properties.float", "devices.properties.event")
 

custom_components/yandex_station/vacuum.py

@@ -67,19 +67,13 @@ def internal_init(self, capabilities: dict, properties: dict):
             self._attr_supported_features |= VacuumEntityFeature.FAN_SPEED
             self._attr_fan_speed_list = [p["value"] for p in item["modes"]]
 
-        if "battery_level" in properties:
-            self._attr_supported_features |= VacuumEntityFeature.BATTERY
-
     def internal_update(self, capabilities: dict, properties: dict):
         if "pause" in capabilities:
             self.pause_value = capabilities["pause"]
 
         if "on" in capabilities:
             self.on_value = capabilities["on"]
 
-        if "battery_level" in properties:
-            self._attr_battery_level = properties["battery_level"]
-
         if "work_speed" in capabilities:
             self._attr_fan_speed = capabilities["work_speed"]