Categorize some of the sensors as diagnostic (#15)

slovdahl · Jun 27, 2023 · 36d1a59 · 36d1a59
1 parent 8bb57c2
commit 36d1a59
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/custom_components/spothinta/sensor.py b/custom_components/spothinta/sensor.py
@@ -13,7 +13,7 @@
     SensorStateClass,
 )
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import CURRENCY_EURO, UnitOfEnergy
+from homeassistant.const import CURRENCY_EURO, EntityCategory, UnitOfEnergy
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers.device_registry import DeviceEntryType
 from homeassistant.helpers.entity import DeviceInfo
@@ -75,55 +75,63 @@ class SpotHintaSensorEntityDescription(
         key="max_price_today",
         name="Highest price - Today",
         service_type="energy",
+        entity_category=EntityCategory.DIAGNOSTIC,
         native_unit_of_measurement=f"{CURRENCY_EURO}/{UnitOfEnergy.KILO_WATT_HOUR}",
         value_fn=lambda data: data.energy_today.highest_price_today,
     ),
     SpotHintaSensorEntityDescription(
         key="max_price_tomorrow",
         name="Highest price - Tomorrow",
         service_type="energy",
+        entity_category=EntityCategory.DIAGNOSTIC,
         native_unit_of_measurement=f"{CURRENCY_EURO}/{UnitOfEnergy.KILO_WATT_HOUR}",
         value_fn=lambda data: data.energy_today.highest_price_tomorrow,
     ),
     SpotHintaSensorEntityDescription(
         key="min_price_today",
         name="Lowest price - Today",
         service_type="energy",
+        entity_category=EntityCategory.DIAGNOSTIC,
         native_unit_of_measurement=f"{CURRENCY_EURO}/{UnitOfEnergy.KILO_WATT_HOUR}",
         value_fn=lambda data: data.energy_today.lowest_price_today,
     ),
     SpotHintaSensorEntityDescription(
         key="min_price_tomorrow",
         name="Lowest price - Tomorrow",
         service_type="energy",
+        entity_category=EntityCategory.DIAGNOSTIC,
         native_unit_of_measurement=f"{CURRENCY_EURO}/{UnitOfEnergy.KILO_WATT_HOUR}",
         value_fn=lambda data: data.energy_today.lowest_price_tomorrow,
     ),
     SpotHintaSensorEntityDescription(
         key="highest_price_time_today",
         name="Time of highest price - Today",
         service_type="energy",
+        entity_category=EntityCategory.DIAGNOSTIC,
         device_class=SensorDeviceClass.TIMESTAMP,
         value_fn=lambda data: data.energy_today.highest_price_time_today,
     ),
     SpotHintaSensorEntityDescription(
         key="highest_price_time_tomorrow",
         name="Time of highest price - Tomorrow",
         service_type="energy",
+        entity_category=EntityCategory.DIAGNOSTIC,
         device_class=SensorDeviceClass.TIMESTAMP,
         value_fn=lambda data: data.energy_today.highest_price_time_tomorrow,
     ),
     SpotHintaSensorEntityDescription(
         key="lowest_price_time_today",
         name="Time of lowest price - Today",
         service_type="energy",
+        entity_category=EntityCategory.DIAGNOSTIC,
         device_class=SensorDeviceClass.TIMESTAMP,
         value_fn=lambda data: data.energy_today.lowest_price_time_today,
     ),
     SpotHintaSensorEntityDescription(
         key="lowest_price_time_tomorrow",
         name="Time of lowest price - Tomorrow",
         service_type="energy",
+        entity_category=EntityCategory.DIAGNOSTIC,
         device_class=SensorDeviceClass.TIMESTAMP,
         value_fn=lambda data: data.energy_today.lowest_price_time_tomorrow,
     ),