temporal fpl1 folder

2022-07-21 20:43:14 -04:00
parent 5e0cf5f659
commit e410106391
16 changed files with 2 additions and 0 deletions
--- a/custom_components/fpl1/sensor.py
+++ b/custom_components/fpl1/sensor.py
@@ -0,0 +1,82 @@
+"""Sensor platform for integration_blueprint."""
+
+from .sensor_KWHSensor import (
+    ProjectedKWHSensor,
+    DailyAverageKWHSensor,
+    BillToDateKWHSensor,
+    NetReceivedKWHSensor,
+    NetDeliveredKWHSensor,
+)
+from .sensor_DatesSensor import (
+    CurrentBillDateSensor,
+    NextBillDateSensor,
+    ServiceDaysSensor,
+    AsOfDaysSensor,
+    RemainingDaysSensor,
+)
+from .sensor_ProjectedBillSensor import (
+    FplProjectedBillSensor,
+    ProjectedBudgetBillSensor,
+    ProjectedActualBillSensor,
+    DeferedAmountSensor,
+)
+from .sensor_AverageDailySensor import (
+    DailyAverageSensor,
+    BudgetDailyAverageSensor,
+    ActualDailyAverageSensor,
+)
+from .sensor_DailyUsageSensor import (
+    FplDailyUsageKWHSensor,
+    FplDailyUsageSensor,
+    FplDailyDeliveredKWHSensor,
+    FplDailyReceivedKWHSensor,
+)
+from .const import DOMAIN
+
+# from .TestSensor import TestSensor
+
+
+async def async_setup_entry(hass, entry, async_add_devices):
+    """Setup sensor platform."""
+    accounts = entry.data.get("accounts")
+
+    coordinator = hass.data[DOMAIN][entry.entry_id]
+    fpl_accounts = []
+
+    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))
+
+    async_add_devices(fpl_accounts)