Add data detection for heap sensor (#845)

ecowitt2mqtt/const.py

@@ -82,6 +82,7 @@
 DATA_POINT_FEELSLIKE: Final = "feelslike"
 DATA_POINT_FROST_POINT: Final = "frostpoint"
 DATA_POINT_FROST_RISK: Final = "frostrisk"
+DATA_POINT_HEAP: Final = "heap"
 DATA_POINT_HEATINDEX: Final = "heatindex"
 DATA_POINT_HOURLY_RAIN: Final = "hourlyrain"
 DATA_POINT_HRAIN_PIEZO: Final = "hrain_piezo"
@@ -237,6 +238,9 @@
 # Lightning units:
 STRIKES: Final = "strikes"
 
+# Memory units:
+MEMORY_BYTES: Final = "bytes"
+
 # Percentage units
 PERCENTAGE: Final = "%"
 

ecowitt2mqtt/data.py

@@ -31,6 +31,7 @@
     DATA_POINT_GLOB_WETNESS,
     DATA_POINT_GLOB_WIND,
     DATA_POINT_GLOB_WINDDIR,
+    DATA_POINT_HEAP,
     DATA_POINT_HEATINDEX,
     DATA_POINT_HUMI_CO2,
     DATA_POINT_HUMIDEX,
@@ -71,6 +72,7 @@
     SimpleCalculator,
 )
 from ecowitt2mqtt.helpers.calculator.battery import BatteryCalculator
+from ecowitt2mqtt.helpers.calculator.heap import HeapCalculator
 from ecowitt2mqtt.helpers.calculator.humidity import (
     AbsoluteHumidityCalculator,
     RelativeHumidityCalculator,
@@ -147,6 +149,7 @@
     DATA_POINT_GLOB_WETNESS: RelativeHumidityCalculator,
     DATA_POINT_GLOB_WIND: WindSpeedCalculator,
     DATA_POINT_GLOB_WINDDIR: WindDirCalculator,
+    DATA_POINT_HEAP: HeapCalculator,
     DATA_POINT_HEATINDEX: HeatIndexCalculator,
     DATA_POINT_HUMIDEX: HumidexCalculator,
     DATA_POINT_HUMIDEX_PERCEPTION: HumidexPerceptionCalculator,

ecowitt2mqtt/helpers/calculator/heap.py

@@ -0,0 +1,18 @@
+"""Define time calculators."""
+from __future__ import annotations
+
+from ecowitt2mqtt.const import MEMORY_BYTES
+from ecowitt2mqtt.helpers.calculator import SimpleCalculator
+
+
+class HeapCalculator(SimpleCalculator):
+    """Define a data update interval calculator."""
+
+    @property
+    def output_unit(self) -> str:
+        """Get the output unit of measurement for this calculation.
+
+        Returns:
+            A unit string.
+        """
+        return MEMORY_BYTES

ecowitt2mqtt/helpers/publisher/mqtt/hass.py

@@ -40,6 +40,7 @@
     DATA_POINT_GLOB_WETNESS,
     DATA_POINT_GLOB_WIND,
     DATA_POINT_GLOB_WINDDIR,
+    DATA_POINT_HEAP,
     DATA_POINT_HEATINDEX,
     DATA_POINT_HOURLY_RAIN,
     DATA_POINT_HRAIN_PIEZO,
@@ -300,6 +301,11 @@ class HassDiscoveryInfo:
     DATA_POINT_GLOB_WINDDIR: EntityDescription(
         icon="mdi:compass",
     ),
+    DATA_POINT_HEAP: EntityDescription(
+        icon="mdi:memory",
+        entity_category=EntityCategory.DIAGNOSTIC,
+        state_class=StateClass.MEASUREMENT,
+    ),
     DATA_POINT_HEATINDEX: EntityDescription(
         device_class=DeviceClass.TEMPERATURE,
         state_class=StateClass.MEASUREMENT,

tests/data/test_data_processing.py

@@ -17,6 +17,7 @@
     ELECTRIC_POTENTIAL_VOLT,
     ILLUMINANCE_WATTS_PER_SQUARE_METER,
     LENGTH_MILES,
+    MEMORY_BYTES,
     PERCENTAGE,
     PRECIPITATION_INCHES,
     PRECIPITATION_INCHES_PER_HOUR,
@@ -645,6 +646,13 @@ def test_missing_distance(device_data: dict[str, Any], ecowitt: Ecowitt) -> None
             attributes={},
             data_type=DataPointType.NON_BOOLEAN,
         ),
+        "heap": CalculatedDataPoint(
+            data_point_key="heap",
+            value=18604.0,
+            unit=MEMORY_BYTES,
+            attributes={},
+            data_type=DataPointType.NON_BOOLEAN,
+        ),
     }
 
 
@@ -2914,6 +2922,13 @@ def test_precision(device_data: dict[str, Any], ecowitt: Ecowitt) -> None:
                     attributes={},
                     data_type=DataPointType.NON_BOOLEAN,
                 ),
+                "heap": CalculatedDataPoint(
+                    data_point_key="heap",
+                    value=18604.0,
+                    unit=MEMORY_BYTES,
+                    attributes={},
+                    data_type=DataPointType.NON_BOOLEAN,
+                ),
             },
         ),
         (

tests/fixtures/payload_gw2000a_1.json

@@ -40,5 +40,6 @@
   "leafwetness_ch1": "14",
   "leaf_batt1": "1.78",
   "freq": "868M",
-  "model": "GW2000A"
+  "model": "GW2000A",
+  "heap": "18604"
 }