Remove current tea mode from kettle

custom_components/yandex_station/water_heater.py

@@ -30,9 +30,6 @@ async def async_setup_entry(hass, entry, async_add_entities):
 class YandexKettle(WaterHeaterEntity, YandexEntity):
     _attr_temperature_unit = UnitOfTemperature.CELSIUS
 
-    on_value: bool = None
-    mode_value: str = None
-
     def internal_init(self, capabilities: dict, properties: dict):
         self._attr_operation_list = ["on", "off"] if "on" in capabilities else []
 
@@ -47,22 +44,11 @@ def internal_init(self, capabilities: dict, properties: dict):
 
     def internal_update(self, capabilities: dict, properties: dict):
         if "on" in capabilities:
-            self.on_value = capabilities["on"]
-
-        if "tea_mode" in capabilities:
-            self.mode_value = capabilities["tea_mode"]
-
-        if self.on_value is False:
-            self._attr_current_operation = "off"
-        elif self.mode_value:
-            self._attr_current_operation = self.mode_value
-        else:
-            self._attr_current_operation = "on"
-
-        if "temperature" in properties:
-            self._attr_current_temperature = properties["temperature"]
+            self._attr_current_operation = "on" if capabilities["on"] else "off"
         if "temperature" in capabilities:
             self._attr_target_temperature = capabilities["temperature"]
+        if "temperature" in properties:
+            self._attr_current_temperature = properties["temperature"]
 
     async def async_set_operation_mode(self, operation_mode):
         if operation_mode == "on":