Merge pull request #179 from CharlesGillanders/additonal-sensors

Add tree count and co2 reduction sensors

Author: Poshy163

custom_components/alphaess/coordinator.py

@@ -160,6 +160,8 @@ async def _async_update_data(self):
                     inverterdata["Total Load"] = await safe_get(_sumdata, "eload")
                     inverterdata["Total Income"] = await safe_get(_sumdata, "totalIncome")
                     inverterdata["Total Generation"] = await safe_get(_sumdata, "epvtotal")
+                    inverterdata["Trees Planted"] = await safe_get(_sumdata, "treeNum")
+                    inverterdata["Co2 Reduction"] = await safe_get(_sumdata, "carbonNum")
 
                     self_data = {
                         "Self Consumption": await safe_get(_sumdata, "eselfConsumption"),

custom_components/alphaess/enums.py

@@ -70,3 +70,5 @@ class AlphaESSNames(str, Enum):
     evcurrentsetting = "Household current setup"
     startcharging = "Start Charging"
     stopcharging = "Stop Charging"
+    treePlanted = "Trees Planted"
+    carbonReduction = "Co2 Reduction"

custom_components/alphaess/sensorlist.py

@@ -4,7 +4,7 @@
     SensorDeviceClass,
     SensorStateClass,
 )
-from homeassistant.const import UnitOfEnergy, PERCENTAGE, UnitOfPower, CURRENCY_DOLLAR, EntityCategory
+from homeassistant.const import UnitOfEnergy, PERCENTAGE, UnitOfPower, CURRENCY_DOLLAR, EntityCategory, UnitOfMass
 
 from .entity import AlphaESSSensorDescription, AlphaESSButtonDescription, AlphaESSNumberDescription
 from .enums import AlphaESSNames
@@ -364,8 +364,22 @@
         device_class=SensorDeviceClass.POWER,
         state_class=SensorStateClass.MEASUREMENT,
         icon="mdi:flash",
+    ),
+    AlphaESSSensorDescription(
+        key=AlphaESSNames.carbonReduction,
+        name="Co2 Reduction",
+        native_unit_of_measurement=UnitOfMass.KILOGRAMS,
+        device_class=SensorDeviceClass.WEIGHT,
+        state_class=SensorStateClass.MEASUREMENT,
+        icon="mdi:molecule-co2",
+    ),
+    AlphaESSSensorDescription(
+        key=AlphaESSNames.treePlanted,
+        name="Trees Planted",
+        native_unit_of_measurement=None,
+        state_class=SensorStateClass.MEASUREMENT,
+        icon="mdi:tree",
     )
-
 ]
 
 LIMITED_SENSOR_DESCRIPTIONS: List[AlphaESSSensorDescription] = [
@@ -646,6 +660,21 @@
         device_class=SensorDeviceClass.POWER,
         state_class=SensorStateClass.MEASUREMENT,
         icon="mdi:flash",
+    ),
+    AlphaESSSensorDescription(
+        key=AlphaESSNames.carbonReduction,
+        name="Carbon Reduction",
+        native_unit_of_measurement=UnitOfMass.KILOGRAMS,
+        device_class=SensorDeviceClass.WEIGHT,
+        state_class=SensorStateClass.MEASUREMENT,
+        icon="mdi:molecule-co2",
+    ),
+    AlphaESSSensorDescription(
+        key=AlphaESSNames.treePlanted,
+        name="Trees Planted",
+        native_unit_of_measurement=None,
+        state_class=SensorStateClass.MEASUREMENT,
+        icon="mdi:tree",
     )
 ]