feat(sensor): Added current rating sensor

BottlecapDave · Jul 23, 2022 · 0893adf · 0893adf
1 parent 8dfdf34
commit 0893adf
Show file tree

Hide file tree

Showing 3 changed files with 86 additions and 743 deletions.
diff --git a/custom_components/carbon_intensity/__init__.py b/custom_components/carbon_intensity/__init__.py
@@ -31,9 +31,9 @@ async def async_setup_entry(hass, entry):
     setup_dependencies(hass, entry.data)
 
     # Forward our entry to setup our default sensors
-    # hass.async_create_task(
-    #   hass.config_entries.async_forward_entry_setup(entry, "sensor")
-    # )
+    hass.async_create_task(
+      hass.config_entries.async_forward_entry_setup(entry, "sensor")
+    )
   elif CONFIG_TARGET_NAME in entry.data:
     # Forward our entry to setup our target rate sensors
     hass.async_create_task(

diff --git a/custom_components/carbon_intensity/binary_sensor.py b/custom_components/carbon_intensity/binary_sensor.py
@@ -52,7 +52,9 @@ async def async_setup_target_sensors(hass, entry, async_add_entities):
 
   coordinator = hass.data[DOMAIN][DATA_RATES_COORDINATOR]
 
-  async_add_entities([CarbonIntensityTargetRate(coordinator, config)], True)
+  entities = [CarbonIntensityTargetRate(coordinator, config)]
+
+  async_add_entities(entities, True)
 
 class CarbonIntensityTargetRate(CoordinatorEntity, BinarySensorEntity):
   """Sensor for calculating when a target should be turned on or off."""