Disable operation mode feature if it is not available

klejejs · Aug 9, 2024 · d30e5dd · d30e5dd
1 parent 41b13be
commit d30e5dd
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/custom_components/thermia/water_heater.py b/custom_components/thermia/water_heater.py
@@ -132,10 +132,12 @@ def operation_list(self):
     @property
     def supported_features(self):
         """Return the list of supported features."""
-        return (
-            WaterHeaterEntityFeature.TARGET_TEMPERATURE
-            | WaterHeaterEntityFeature.OPERATION_MODE
-        )
+        features = WaterHeaterEntityFeature.TARGET_TEMPERATURE
+
+        if self.current_operation is not None:
+            features |= WaterHeaterEntityFeature.OPERATION_MODE
+
+        return features
 
     async def async_set_temperature(self, **kwargs):
         """Set new target temperature."""