added bill to date sensor and organize sensor initialization file

custom_components/fpl/sensor.py

@@ -15,6 +15,7 @@ from .sensor_DatesSensor import (
     RemainingDaysSensor,
 )
 from .sensor_ProjectedBillSensor import (
+    BillToDateSensor,
     FplProjectedBillSensor,
     ProjectedBudgetBillSensor,
     ProjectedActualBillSensor,
@@ -32,64 +33,69 @@ from .sensor_DailyUsageSensor import (
     FplDailyReceivedKWHSensor,
 )
 
-from .sensor_test import TestSensor
+from .const import CONF_ACCOUNTS, CONF_TERRITORY, DOMAIN, FPL_MAINREGION, FPL_NORTHWEST
 
-from .const import DOMAIN, DEBUG
+ALL_REGIONS = [FPL_MAINREGION, FPL_NORTHWEST]
+ONLY_MAINREGION = [FPL_MAINREGION]
 
-# from .TestSensor import TestSensor
+sensors = {}
+
+
+def registerSensor(sensor, regions):
+    """register all available sensors"""
+    for region in regions:
+        if region in sensors:
+            sensors[region].append(sensor)
+        else:
+            sensors[region] = [sensor]
+
+
+# bill sensors
+registerSensor(FplProjectedBillSensor, ALL_REGIONS)
+registerSensor(BillToDateSensor, ALL_REGIONS)
+
+# budget billing
+registerSensor(ProjectedBudgetBillSensor, ONLY_MAINREGION)
+registerSensor(ProjectedActualBillSensor, ONLY_MAINREGION)
+registerSensor(DeferedAmountSensor, ONLY_MAINREGION)
+
+
+# usage sensors
+registerSensor(DailyAverageSensor, ONLY_MAINREGION)
+registerSensor(BudgetDailyAverageSensor, ONLY_MAINREGION)
+registerSensor(ActualDailyAverageSensor, ONLY_MAINREGION)
+
+registerSensor(FplDailyUsageSensor, ONLY_MAINREGION)
+registerSensor(FplDailyUsageKWHSensor, ONLY_MAINREGION)
+
+# date sensors
+registerSensor(CurrentBillDateSensor, ALL_REGIONS)
+registerSensor(NextBillDateSensor, ONLY_MAINREGION)
+registerSensor(ServiceDaysSensor, ALL_REGIONS)
+registerSensor(AsOfDaysSensor, ALL_REGIONS)
+registerSensor(RemainingDaysSensor, ALL_REGIONS)
+
+# KWH sensors
+registerSensor(ProjectedKWHSensor, ALL_REGIONS)
+registerSensor(DailyAverageKWHSensor, ONLY_MAINREGION)
+registerSensor(BillToDateKWHSensor, ALL_REGIONS)
+registerSensor(NetReceivedKWHSensor, ONLY_MAINREGION)
+registerSensor(NetDeliveredKWHSensor, ONLY_MAINREGION)
+registerSensor(FplDailyReceivedKWHSensor, ONLY_MAINREGION)
+registerSensor(FplDailyDeliveredKWHSensor, ONLY_MAINREGION)
 
 
 async def async_setup_entry(hass, entry, async_add_devices):
     """Setup sensor platform."""
 
-    accounts = entry.data.get("accounts")
-    territory = entry.data.get("territory")
-
-    print(f"setting sensor for {territory}")
+    accounts = entry.data.get(CONF_ACCOUNTS)
+    territory = entry.data.get(CONF_TERRITORY)
 
     coordinator = hass.data[DOMAIN][entry.entry_id]
     fpl_accounts = []
 
-    if DEBUG:
-        for account in accounts:
-            fpl_accounts.append(TestSensor(coordinator, entry, account))
-
-            fpl_accounts.append(FplProjectedBillSensor(coordinator, entry, account))
-    else:
-        for account in accounts:
-            # Test Sensor
-            # fpl_accounts.append(TestSensor(coordinator, entry, account))
-
-            # bill sensors
-            fpl_accounts.append(FplProjectedBillSensor(coordinator, entry, account))
-            fpl_accounts.append(ProjectedBudgetBillSensor(coordinator, entry, account))
-            fpl_accounts.append(ProjectedActualBillSensor(coordinator, entry, account))
-            fpl_accounts.append(DeferedAmountSensor(coordinator, entry, account))
-
-            # usage sensors
-            fpl_accounts.append(DailyAverageSensor(coordinator, entry, account))
-            fpl_accounts.append(BudgetDailyAverageSensor(coordinator, entry, account))
-            fpl_accounts.append(ActualDailyAverageSensor(coordinator, entry, account))
-
-            fpl_accounts.append(FplDailyUsageSensor(coordinator, entry, account))
-            fpl_accounts.append(FplDailyUsageKWHSensor(coordinator, entry, account))
-
-            # date sensors
-            fpl_accounts.append(CurrentBillDateSensor(coordinator, entry, account))
-            fpl_accounts.append(NextBillDateSensor(coordinator, entry, account))
-            fpl_accounts.append(ServiceDaysSensor(coordinator, entry, account))
-            fpl_accounts.append(AsOfDaysSensor(coordinator, entry, account))
-            fpl_accounts.append(RemainingDaysSensor(coordinator, entry, account))
-
-            # KWH sensors
-            fpl_accounts.append(ProjectedKWHSensor(coordinator, entry, account))
-            fpl_accounts.append(DailyAverageKWHSensor(coordinator, entry, account))
-            fpl_accounts.append(BillToDateKWHSensor(coordinator, entry, account))
-
-            fpl_accounts.append(NetReceivedKWHSensor(coordinator, entry, account))
-            fpl_accounts.append(NetDeliveredKWHSensor(coordinator, entry, account))
-
-            fpl_accounts.append(FplDailyReceivedKWHSensor(coordinator, entry, account))
-            fpl_accounts.append(FplDailyDeliveredKWHSensor(coordinator, entry, account))
+    for account in accounts:
+        for sensor in sensors[territory]:
+            fpl_accounts.append(sensor(coordinator, entry, account))
 
     async_add_devices(fpl_accounts)