Merge pull request #100 from RustyDust/sru_work

Add sensor for `battery_care`, fix oversight in relation to new mode 11

Author: RustyDust

custom_components/sonnenbatterie/coordinator.py

@@ -9,7 +9,7 @@
 from homeassistant.helpers.update_coordinator import DataUpdateCoordinator
 from sonnenbatterie import AsyncSonnenBatterie
 
-from custom_components.sonnenbatterie import LOGGER, DOMAIN, ATTR_SONNEN_DEBUG, CONF_SERIAL_NUMBER
+from custom_components.sonnenbatterie import LOGGER, DOMAIN, ATTR_SONNEN_DEBUG
 
 
 class SonnenbatterieCoordinator(DataUpdateCoordinator):
@@ -87,6 +87,7 @@ def populate_battery_info(self):
     async def _async_update_data(self):
         """Populate self.latestdata"""
         if time() - self._last_login > 60:
+            # noinspection PyBroadException
             try:
                 await self.sbconn.logout()
             except:
@@ -123,6 +124,9 @@ async def _async_update_data(self):
             result = await self.sbconn.get_commissioning_settings()
             self.latestData["commissioning_settings"] = result
 
+            result = await self.sbconn.sb2.get_status()
+            self.latestData["v2_status"] = result
+
             self._last_error = None
 
         except Exception as e:

custom_components/sonnenbatterie/manifest.json

@@ -8,5 +8,5 @@
     "iot_class": "local_polling",
     "issue_tracker": "https://github.com/weltmeyer/ha_sonnenbatterie/issues",
     "requirements": ["requests","sonnenbatterie>=0.6.0"],
-    "version": "2025.03.01"
+    "version": "2025.03.02"
 }

custom_components/sonnenbatterie/sensor_list.py

@@ -288,7 +288,7 @@ def generate_powermeter_sensors(_coordinator):
         key="state_operating_mode",
         legacy_key="operating_mode",
         icon="mdi:state-machine",
-        options=["1", "2", "6", "10"],
+        options=["1", "2", "6", "10", "11"],
         device_class=SensorDeviceClass.ENUM,
         value_fn=lambda coordinator: coordinator.latestData.get("status", {}).get(
             "OperatingMode"
@@ -546,5 +546,14 @@ def generate_powermeter_sensors(_coordinator):
         value_fn=lambda coordinator: coordinator.latestData.get("commissioning_settings", {})
                                      .get('data', {}).get('attributes',{}).get('tou_max_power_limit', 'unknown'),
         entity_registry_enabled_default=True,
+    ),
+    SonnenbatterieSensorEntityDescription(
+        key="battery_care",
+        icon="mdi:wrench-clock",
+        state_class=SensorStateClass.MEASUREMENT,
+        entity_category=EntityCategory.DIAGNOSTIC,
+        value_fn=lambda coordinator: (coordinator.latestData.get("v2_status", {})
+        .get("dischargeNotAllowed", "false") == "true"),
+        entity_registry_enabled_default=True,
     )
 )

custom_components/sonnenbatterie/translations/de.json

@@ -185,6 +185,9 @@
             },
             "tou_max_power": {
                 "name": "Maximaler ToU Netzbezug"
+            },
+            "battery_care": {
+                "name": "Batteriepflege aktiv"
             }
         }
     },

custom_components/sonnenbatterie/translations/en.json

@@ -186,6 +186,9 @@
             },
             "tou_max_power": {
                 "name": "ToU max power consumption"
+            },
+            "battery_care": {
+                "name": "Battery care active"
             }
         }
     },