Preset modes patch (#35)

* patch error and bugs on some presets for climate

* translation and icon

* update tests to change async_update_entity by import

Author: Delmael

custom_components/smartbox/climate.py

@@ -159,6 +159,12 @@ def preset_mode(self) -> str:  # noqa: PLR0911
         if self._node.node_type == SmartboxNodeType.HTR_MOD:
             _check_status_key("mode", self._node.node_type, self._status)
             mode = self._status["mode"]
+            if mode == "auto":
+                return PRESET_SCHEDULE
+            if mode == "presence":
+                return PRESET_ACTIVITY
+            if mode == "self_learn":
+                return PRESET_SELF_LEARN
             if mode == "manual":
                 _check_status_key("selected_temp", self._node.node_type, self._status)
                 selected_temp = self._status["selected_temp"]
@@ -173,12 +179,6 @@ def preset_mode(self) -> str:  # noqa: PLR0911
                     f"{self._node.node_type} and {mode} - please report to {GITHUB_ISSUES_URL}."
                 )
                 raise ValueError(msg)
-            if mode == "auto":
-                return PRESET_SCHEDULE
-            if mode == "presence":
-                return PRESET_ACTIVITY
-            if mode == "self_learn":
-                return PRESET_SELF_LEARN
             msg = f"Unknown smartbox node mode {mode}"
             raise ValueError(msg)
         return PRESET_HOME
@@ -196,8 +196,8 @@ def preset_modes(self) -> list[str]:
                     PRESET_COMFORT,
                     PRESET_ECO,
                     PRESET_FROST,
-                    PRESET_SCHEDULE,
                     PRESET_SELF_LEARN,
+                    PRESET_SCHEDULE,
                 ]
             )
         else:

custom_components/smartbox/icons.json

@@ -0,0 +1,17 @@
+{
+  "entity": {
+    "climate": {
+      "thermostat": {
+        "state_attributes": {
+          "preset_mode": {
+            "state": {
+              "frost": "mdi:snowflake",
+              "schedule": "mdi:calendar-clock-outline",
+              "self_learn": "mdi:brain"
+            }
+          }
+        }
+      }
+    }
+  }
+}

custom_components/smartbox/manifest.json

@@ -10,5 +10,5 @@
   "issue_tracker": "https://github.com/ajtudela/hass-smartbox/issues",
   "quality_scale": "custom",
   "requirements": ["smartbox>=2.2.0,<2.3.0"],
-  "version": "2.1.4"
+  "version": "2.1.5"
 }

custom_components/smartbox/models.py

@@ -442,6 +442,8 @@ async def get_samples(self, start_time: int, end_time: int) -> SamplesDict:
     @property
     def total_energy(self) -> float | None:
         """Get the energy used."""
+        if not self._samples:
+            return None
         return self._samples[-1]["counter"]
 
     @property

custom_components/smartbox/sensor.py

@@ -203,7 +203,7 @@ def native_value(self) -> float:
             self._status["power"]
             if (
                 self._node.node_type == SmartboxNodeType.PMO
-                or self._node.is_heating(self._status)
+                or ("power" in self._status and self._node.is_heating(self._status))
             )
             else 0
         )

custom_components/smartbox/translations/en.json

@@ -54,6 +54,19 @@
     }
   },
   "entity": {
+    "climate": {
+      "thermostat": {
+        "state_attributes": {
+          "preset_mode": {
+            "state": {
+              "frost": "Frost",
+              "schedule": "Schedule",
+              "self_learn": "Self-learning"
+            }
+          }
+        }
+      }
+    },
     "sensor": {
       "charge_level": {
         "name": "Charge Level"

custom_components/smartbox/translations/es.json

@@ -57,6 +57,19 @@
     }
   },
   "entity": {
+    "climate": {
+      "thermostat": {
+        "state_attributes": {
+          "preset_mode": {
+            "state": {
+              "frost": "Anticongelante",
+              "schedule": "Planificación",
+              "self_learn": "Autoaprendizaje"
+            }
+          }
+        }
+      }
+    },
     "sensor": {
       "charge_level": {
         "name": "Nivel de carga"

custom_components/smartbox/translations/fr.json

@@ -54,6 +54,19 @@
     }
   },
   "entity": {
+    "climate": {
+      "thermostat": {
+        "state_attributes": {
+          "preset_mode": {
+            "state": {
+              "frost": "Anti-gel",
+              "schedule": "Planning",
+              "self_learn": "Auto apprentissage"
+            }
+          }
+        }
+      }
+    },
     "sensor": {
       "charge_level": {
         "name": "Niveau de charge"

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "hass-smartbox"
-version = "2.1.4"
+version = "2.1.5"
 description = "Home Assistant integration for Smartbox."
 readme = "README.md"
 requires-python = ">=3.13"