initial commit v0.2.1

custom_components/heating_mpc/sensor.py

@@ -0,0 +1,90 @@
+"""Diagnostics and predicted temperatures exposed as HA sensors."""
+from __future__ import annotations
+
+from homeassistant.components.sensor import SensorEntity, SensorDeviceClass
+from homeassistant.const import UnitOfTemperature
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from .const import DOMAIN
+
+
+async def async_setup_entry(hass, entry, async_add_entities) -> None:
+    coordinator = hass.data[DOMAIN][entry.entry_id]
+    entities = [MPCStatus(coordinator, entry.entry_id), MPCWeightedTemp(coordinator, entry.entry_id), MPCOutdoorTemperature(coordinator, entry.entry_id)]
+    entities += [MPCRoomPrediction(coordinator, entry.entry_id, room) for room in coordinator.rooms]
+    async_add_entities(entities)
+
+
+class MPCStatus(CoordinatorEntity, SensorEntity):
+    _attr_has_entity_name = True
+    _attr_name = "Heating MPC Status"
+    _attr_icon = "mdi:brain"
+
+    def __init__(self, coordinator, entry_id):
+        super().__init__(coordinator)
+        self._attr_unique_id = f"{entry_id}_status"
+
+    @property
+    def native_value(self):
+        return self.coordinator.data.get("status")
+
+    @property
+    def extra_state_attributes(self):
+        data = self.coordinator.data
+        return {key: value for key, value in data.items() if key not in ("status", "prediction_60m", "prediction_end")}
+
+
+class MPCWeightedTemp(CoordinatorEntity, SensorEntity):
+    _attr_has_entity_name = True
+    _attr_name = "Heating MPC Weighted Temperature"
+    _attr_device_class = SensorDeviceClass.TEMPERATURE
+    _attr_native_unit_of_measurement = UnitOfTemperature.CELSIUS
+
+    def __init__(self, coordinator, entry_id):
+        super().__init__(coordinator)
+        self._attr_unique_id = f"{entry_id}_weighted_temp"
+
+    @property
+    def native_value(self):
+        return self.coordinator.data.get("weighted_temperature")
+
+
+class MPCRoomPrediction(CoordinatorEntity, SensorEntity):
+    _attr_has_entity_name = True
+    _attr_device_class = SensorDeviceClass.TEMPERATURE
+    _attr_native_unit_of_measurement = UnitOfTemperature.CELSIUS
+    _attr_icon = "mdi:chart-line"
+
+    def __init__(self, coordinator, entry_id, room):
+        super().__init__(coordinator)
+        self.room = room
+        self._attr_unique_id = f"{entry_id}_{room}_predicted_60min"
+        self._attr_name = f"MPC {room} 60min Prediction"
+
+    @property
+    def native_value(self):
+        return self.coordinator.data.get("prediction_60m", {}).get(self.room)
+
+class MPCOutdoorTemperature(CoordinatorEntity, SensorEntity):
+    """Display chosen outdoor temperature and all currently valid alternatives."""
+    _attr_has_entity_name = True
+    _attr_name = "MPC Outdoor Temperature"
+    _attr_device_class = SensorDeviceClass.TEMPERATURE
+    _attr_native_unit_of_measurement = UnitOfTemperature.CELSIUS
+    _attr_icon = "mdi:thermometer"
+
+    def __init__(self, coordinator, entry_id):
+        super().__init__(coordinator)
+        self._attr_unique_id = f"{entry_id}_outside_temperature"
+
+    @property
+    def native_value(self):
+        return self.coordinator.data.get("weather", {}).get("temperature")
+
+    @property
+    def extra_state_attributes(self):
+        data = self.coordinator.data
+        return {
+            "active_source": data.get("weather_active_sources", {}).get("temperature"),
+            "alternatives": data.get("weather_comparison", {}).get("temperature", {}),
+        }