Add state_class: measurement for battery level sensors (+ per-entity overrides) to enable statistics (#55)

* Added state_class override functionality

* Update version

Author: derolli1976

custom_components/enpal_webparser/const.py

@@ -59,6 +59,7 @@
     "energy_battery_charge_level": "battery",
     "energy_battery_charge_level_unit_1": "battery",
     "energy_battery_charge_level_unit_2": "battery",
+    "energy_battery_charge_level_absolute": "battery",
 }
 
 # --- Wallbox Mode Mapping ---
@@ -214,3 +215,10 @@
     "voltage_wallbox_connector_1_phase_b": "mdi:transmission-tower",
     "voltage_wallbox_connector_1_phase_c": "mdi:transmission-tower",
 }
+
+STATE_CLASS_OVERRIDES = {
+    "energy_battery_charge_level": "measurement",
+    "energy_battery_charge_level_unit_1": "measurement",
+    "energy_battery_charge_level_unit_2": "measurement",
+    "energy_battery_charge_level_absolute": "measurement",
+}

custom_components/enpal_webparser/entity_factory.py

@@ -24,7 +24,7 @@
 from homeassistant.helpers.entity import DeviceInfo
 
 from .utils import make_id
-from .const import ICON_MAP
+from .const import ICON_MAP, STATE_CLASS_OVERRIDES
 
 
 class EnpalBaseSensor(CoordinatorEntity, SensorEntity):
@@ -48,6 +48,11 @@ def __init__(self, sensor: dict, coordinator: DataUpdateCoordinator):
             self._attr_device_class = getattr(SensorDeviceClass, device_class.upper())
         else:
             self._attr_device_class = device_class
+        
+        # Overrides for state class if specified
+        sensor_id = self._attr_unique_id
+        if sensor_id in STATE_CLASS_OVERRIDES:
+            self._attr_state_class = STATE_CLASS_OVERRIDES[sensor_id]
 
     @property
     def native_value(self):
@@ -93,4 +98,9 @@ def build_sensor_entity(sensor: dict, coordinator: DataUpdateCoordinator) -> Sen
 class EnpalEnergySensor(EnpalBaseSensor):
     def __init__(self, sensor: dict, coordinator: DataUpdateCoordinator):
         super().__init__(sensor, coordinator)
+        # Default: Energy counter is total_increasing
         self._attr_state_class = "total_increasing"
+        # Allow explicit overrides
+        sensor_id = self._attr_unique_id
+        if sensor_id in STATE_CLASS_OVERRIDES:
+            self._attr_state_class = STATE_CLASS_OVERRIDES[sensor_id]

custom_components/enpal_webparser/manifest.json

@@ -8,5 +8,5 @@
   "iot_class": "local_polling",
   "issue_tracker": "https://github.com/derolli1976/enpal/issues",
   "requirements": [],
-  "version": "2.1.5"
+  "version": "2.1.6"
 }