Adding config option for tracking unknown sources

dvd-dev · Jan 15, 2022 · f2d2c7e · f2d2c7e
1 parent 8addaeb
commit f2d2c7e
Show file tree

Hide file tree

Showing 4 changed files with 23 additions and 0 deletions.
diff --git a/custom_components/hilo/__init__.py b/custom_components/hilo/__init__.py
@@ -51,13 +51,15 @@
     CONF_HQ_PLAN_NAME,
     CONF_LOG_TRACES,
     CONF_TARIFF,
+    CONF_TRACK_UNKNOWN_SOURCES,
     CONF_UNTARIFICATED_DEVICES,
     DEFAULT_APPRECIATION_PHASE,
     DEFAULT_CHALLENGE_LOCK,
     DEFAULT_GENERATE_ENERGY_METERS,
     DEFAULT_HQ_PLAN_NAME,
     DEFAULT_LOG_TRACES,
     DEFAULT_SCAN_INTERVAL,
+    DEFAULT_TRACK_UNKNOWN_SOURCES,
     DEFAULT_UNTARIFICATED_DEVICES,
     DOMAIN,
     HILO_ENERGY_TOTAL,
@@ -212,6 +214,9 @@ def __init__(self, hass: HomeAssistant, entry: ConfigEntry, api: API) -> None:
         self.challenge_lock = entry.options.get(
             CONF_CHALLENGE_LOCK, DEFAULT_CHALLENGE_LOCK
         )
+        self.track_unknown_sources = entry.options.get(
+            CONF_TRACK_UNKNOWN_SOURCES, DEFAULT_TRACK_UNKNOWN_SOURCES
+        )
         self.untarificated_devices = entry.options.get(
             CONF_UNTARIFICATED_DEVICES, DEFAULT_UNTARIFICATED_DEVICES
         )

diff --git a/custom_components/hilo/config_flow.py b/custom_components/hilo/config_flow.py
@@ -25,13 +25,15 @@
     CONF_GENERATE_ENERGY_METERS,
     CONF_HQ_PLAN_NAME,
     CONF_LOG_TRACES,
+    CONF_TRACK_UNKNOWN_SOURCES,
     CONF_UNTARIFICATED_DEVICES,
     DEFAULT_APPRECIATION_PHASE,
     DEFAULT_CHALLENGE_LOCK,
     DEFAULT_GENERATE_ENERGY_METERS,
     DEFAULT_HQ_PLAN_NAME,
     DEFAULT_LOG_TRACES,
     DEFAULT_SCAN_INTERVAL,
+    DEFAULT_TRACK_UNKNOWN_SOURCES,
     DEFAULT_UNTARIFICATED_DEVICES,
     DOMAIN,
     LOG,
@@ -61,6 +63,10 @@
             CONF_CHALLENGE_LOCK,
             default=DEFAULT_CHALLENGE_LOCK,
         ): cv.boolean,
+        vol.Optional(
+            CONF_TRACK_UNKNOWN_SOURCES,
+            default=DEFAULT_TRACK_UNKNOWN_SOURCES,
+        ): cv.boolean,
         vol.Optional(
             CONF_APPRECIATION_PHASE,
             default=DEFAULT_APPRECIATION_PHASE,
@@ -189,6 +195,14 @@ async def async_step_init(
                             )
                         },
                     ): cv.boolean,
+                    vol.Optional(
+                        CONF_TRACK_UNKNOWN_SOURCES,
+                        description={
+                            "suggested_value": self.config_entry.options.get(
+                                CONF_TRACK_UNKNOWN_SOURCES
+                            )
+                        },
+                    ): cv.boolean,
                     vol.Optional(
                         CONF_HQ_PLAN_NAME,
                         description={

diff --git a/custom_components/hilo/const.py b/custom_components/hilo/const.py
@@ -23,6 +23,9 @@
 CONF_CHALLENGE_LOCK = "challenge_lock"
 DEFAULT_CHALLENGE_LOCK = False
 
+CONF_TRACK_UNKNOWN_SOURCES = "track_unknown_sources"
+DEFAULT_TRACK_UNKNOWN_SOURCES = False
+
 CONF_ENERGY_METER_PERIOD = "energy_meter_period"
 DEFAULT_ENERGY_METER_PERIOD = DAILY
 

diff --git a/custom_components/hilo/translations/en.json b/custom_components/hilo/translations/en.json
@@ -29,6 +29,7 @@
           "scan_interval": "Scan interval (min: 15s)",
           "log_traces": "Also log request data and websocket messages (requires debug log level on both the integration and pyhilo)",
           "challenge_lock": "Lock climate entities during Hilo challenges, preventing any changes when a challenge is in progress.",
+          "track_unknown_sources": "Track unknown power sources in a separate energy sensor. This is a round approximation calculated when we get a reading from the Smart Energy Meter.",
           "appreciation_phase": "Add an appreciation phase of X hours before the preheat phase."
         }
       }