Ensure all battery cell voltages are reported (#36)

cdpuk · Mar 7, 2023 · 205f782 · 205f782
1 parent 209936f
commit 205f782
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/custom_components/givenergy_local/sensor.py b/custom_components/givenergy_local/sensor.py
@@ -590,4 +590,7 @@ class BatteryCellsVoltageSensor(BatteryBasicSensor):
     @property
     def extra_state_attributes(self) -> Mapping[str, Any] | None:
         """Expose individual cell voltages."""
-        return self.data.dict(include={f"v_battery_cell_{i:02d}" for i in range(1, 16)})  # type: ignore[no-any-return]
+        num_cells = self.data.battery_num_cells
+        return self.data.dict(  # type: ignore[no-any-return]
+            include={f"v_battery_cell_{i:02d}" for i in range(1, num_cells + 1)}
+        )