Add price descriptor attribute to price sensors

home-assistant · joostlek · Jul 15, 2025 · Mar 11, 2022 · Mar 11, 2022 · Mar 11, 2022
commit 3c9994eca56dbaab61415c5dbadfaebddfaa13c7
diff --git a/homeassistant/components/amberelectric/manifest.json b/homeassistant/components/amberelectric/manifest.json
@@ -7,7 +7,7 @@
         "@madpilot"
     ],
     "requirements": [
-        "amberelectric==1.0.3"
+        "amberelectric==1.0.4"
     ],
     "iot_class": "cloud_polling",
     "loggers": ["amberelectric"]

diff --git a/homeassistant/components/amberelectric/sensor.py b/homeassistant/components/amberelectric/sensor.py
@@ -106,6 +106,7 @@ def extra_state_attributes(self) -> Mapping[str, Any] | None:
         data["renewables"] = round(interval.renewables)
         data["estimate"] = interval.estimate
         data["spike_status"] = interval.spike_status.value
+        data["descriptor"] = interval.descriptor.value
         data["channel_type"] = interval.channel_type.value
 
         if interval.range is not None:
@@ -160,6 +161,7 @@ def extra_state_attributes(self) -> Mapping[str, Any] | None:
             datum["end_time"] = interval.end_time.isoformat()
             datum["renewables"] = round(interval.renewables)
             datum["spike_status"] = interval.spike_status.value
+            datum["descriptor"] = interval.descriptor.value
 
             if interval.range is not None:
                 datum["range_min"] = format_cents_to_dollars(interval.range.min)

diff --git a/tests/components/amberelectric/helpers.py b/tests/components/amberelectric/helpers.py
@@ -6,7 +6,7 @@
 from amberelectric.model.channel import ChannelType
 from amberelectric.model.current_interval import CurrentInterval
 from amberelectric.model.forecast_interval import ForecastInterval
-from amberelectric.model.interval import SpikeStatus
+from amberelectric.model.interval import Descriptor, SpikeStatus
 from dateutil import parser
 
 
@@ -26,6 +26,7 @@ def generate_actual_interval(
         renewables=50,
         channel_type=channel_type.value,
         spike_status=SpikeStatus.NO_SPIKE.value,
+        descriptor=Descriptor.LOW.value,
     )
 
 
@@ -45,6 +46,7 @@ def generate_current_interval(
         renewables=50.6,
         channel_type=channel_type.value,
         spike_status=SpikeStatus.NO_SPIKE.value,
+        descriptor=Descriptor.LOW.value,
         estimate=True,
     )
 
@@ -65,6 +67,7 @@ def generate_forecast_interval(
         renewables=50,
         channel_type=channel_type.value,
         spike_status=SpikeStatus.NO_SPIKE.value,
+        descriptor=Descriptor.LOW.value,
         estimate=True,
     )
 

diff --git a/tests/components/amberelectric/test_sensor.py b/tests/components/amberelectric/test_sensor.py
@@ -116,6 +116,7 @@ async def test_general_price_sensor(hass: HomeAssistant, setup_general: Mock) ->
     assert attributes["renewables"] == 51
     assert attributes["estimate"] is True
     assert attributes["spike_status"] == "none"
+    assert attributes["descriptor"] == "low"
     assert attributes["channel_type"] == "general"
     assert attributes["attribution"] == "Data provided by Amber Electric"
     assert attributes.get("range_min") is None
@@ -155,6 +156,7 @@ async def test_general_and_controlled_load_price_sensor(
     assert attributes["renewables"] == 51
     assert attributes["estimate"] is True
     assert attributes["spike_status"] == "none"
+    assert attributes["descriptor"] == "low"
     assert attributes["channel_type"] == "controlledLoad"
     assert attributes["attribution"] == "Data provided by Amber Electric"
 
@@ -178,6 +180,7 @@ async def test_general_and_feed_in_price_sensor(
     assert attributes["renewables"] == 51
     assert attributes["estimate"] is True
     assert attributes["spike_status"] == "none"
+    assert attributes["descriptor"] == "low"
     assert attributes["channel_type"] == "feedIn"
     assert attributes["attribution"] == "Data provided by Amber Electric"
 
@@ -204,6 +207,7 @@ async def test_general_forecast_sensor(
     assert first_forecast["end_time"] == "2021-09-21T09:00:00+10:00"
     assert first_forecast["renewables"] == 50
     assert first_forecast["spike_status"] == "none"
+    assert first_forecast["descriptor"] == "low"
 
     assert first_forecast.get("range_min") is None
     assert first_forecast.get("range_max") is None
@@ -246,6 +250,7 @@ async def test_controlled_load_forecast_sensor(
     assert first_forecast["end_time"] == "2021-09-21T09:00:00+10:00"
     assert first_forecast["renewables"] == 50
     assert first_forecast["spike_status"] == "none"
+    assert first_forecast["descriptor"] == "low"
 
 
 async def test_feed_in_forecast_sensor(
@@ -270,6 +275,7 @@ async def test_feed_in_forecast_sensor(
     assert first_forecast["end_time"] == "2021-09-21T09:00:00+10:00"
     assert first_forecast["renewables"] == 50
     assert first_forecast["spike_status"] == "none"
+    assert first_forecast["descriptor"] == "low"
 
 
 def test_renewable_sensor(hass: HomeAssistant, setup_general) -> None: