Fix: Add sensors entities for relays of the optional extension module

ylabonte · Apr 3, 2024 · a2e0700 · a2e0700
1 parent 39a56cf
commit a2e0700
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/custom_components/proconip_pool_controller/sensor.py b/custom_components/proconip_pool_controller/sensor.py
@@ -50,7 +50,8 @@ async def async_setup_entry(
                 coordinator=coordinator, canister_no=i + 1, instance_id=entry.entry_id
             )
         )
-    for i in range(8):
+    number_of_relays = 16 if coordinator.data.is_relay_extension_enabled() else 8
+    for i in range(number_of_relays):
         sensor_entities.append(
             ProconipRelayStateSensor(
                 coordinator=coordinator, relay_no=i + 1, instance_id=entry.entry_id