feat: add battery capacity sensor

Author: mkuthan

custom_components/solis_cloud_control/__init__.py

@@ -9,7 +9,7 @@
 from .api import SolisCloudControlApiClient
 from .const import CONF_INVERTER_SN
 
-PLATFORMS: list[Platform] = [Platform.SELECT, Platform.TEXT, Platform.NUMBER, Platform.SWITCH]
+PLATFORMS: list[Platform] = [Platform.SELECT, Platform.TEXT, Platform.NUMBER, Platform.SWITCH, Platform.SENSOR]
 
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:

custom_components/solis_cloud_control/const.py

@@ -14,6 +14,7 @@
 CID_BATTERY_RESERVE_SOC = 157
 CID_BATTERY_OVER_DISCHARGE_SOC = 158
 CID_BATTERY_FORCE_CHARGE_SOC = 160
+CID_BATTERY_CAPACITY = 172
 CID_STORAGE_MODE = 636
 CID_CHARGE_SLOT1_SWITCH = 5916
 CID_CHARGE_SLOT1_TIME = 5946

custom_components/solis_cloud_control/coordinator.py

@@ -7,6 +7,7 @@
 
 from custom_components.solis_cloud_control.api import SolisCloudControlApiClient, SolisCloudControlApiError
 from custom_components.solis_cloud_control.const import (
+    CID_BATTERY_CAPACITY,
     CID_BATTERY_FORCE_CHARGE_SOC,
     CID_BATTERY_OVER_DISCHARGE_SOC,
     CID_BATTERY_RESERVE_SOC,
@@ -32,6 +33,7 @@
     CID_BATTERY_RESERVE_SOC,
     CID_BATTERY_FORCE_CHARGE_SOC,
     CID_BATTERY_OVER_DISCHARGE_SOC,
+    CID_BATTERY_CAPACITY,
     CID_CHARGE_SLOT1_CURRENT,
     CID_CHARGE_SLOT1_SOC,
     CID_CHARGE_SLOT1_SWITCH,

custom_components/solis_cloud_control/sensor.py

@@ -0,0 +1,43 @@
+from homeassistant.components.sensor import SensorDeviceClass, SensorEntity, SensorEntityDescription
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import PERCENTAGE
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers.entity_platform import AddEntitiesCallback
+
+from .const import CID_BATTERY_CAPACITY
+from .coordinator import SolisCloudControlCoordinator
+from .entity import SolisCloudControlEntity
+
+
+async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback) -> None:  # noqa: ARG001
+    coordinator: SolisCloudControlCoordinator = entry.runtime_data
+    async_add_entities(
+        [
+            BatteryCapacitySensor(
+                coordinator=coordinator,
+                entity_description=SensorEntityDescription(
+                    key="battery_capacity",
+                    name="Battery Capacity",
+                    icon="mdi:battery",
+                ),
+                cid=CID_BATTERY_CAPACITY,
+            ),
+        ]
+    )
+
+
+class BatteryCapacitySensor(SolisCloudControlEntity, SensorEntity):
+    def __init__(
+        self, coordinator: SolisCloudControlCoordinator, entity_description: SensorEntityDescription, cid: int
+    ) -> None:
+        super().__init__(coordinator, entity_description, cid)
+        self._attr_device_class = SensorDeviceClass.BATTERY
+        self._attr_native_unit_of_measurement = PERCENTAGE
+
+    @property
+    def native_value(self) -> float | None:
+        if not self.coordinator.data:
+            return None
+
+        value_str = self.coordinator.data.get(self.cid)
+        return float(value_str) if value_str is not None else None

custom_components/solis_cloud_control/switch.py

@@ -21,7 +21,7 @@ async def async_setup_entry(
                 entity_description=SwitchEntityDescription(
                     key="slot1_charge_switch",
                     name="Slot1 Charge",
-                    icon="mdi:battery-charging",
+                    icon="mdi:battery-plus-outline",
                 ),
                 cid=CID_CHARGE_SLOT1_SWITCH,
             ),
@@ -30,7 +30,7 @@ async def async_setup_entry(
                 entity_description=SwitchEntityDescription(
                     key="slot1_discharge_switch",
                     name="Slot1 Discharge",
-                    icon="mdi:battery-minus",
+                    icon="mdi:battery-minus-outline",
                 ),
                 cid=CID_DISCHARGE_SLOT1_SWITCH,
             ),