v0.9.7

home-assistant/custom_components/gree_controller/select.py

@@ -1,4 +1,4 @@
-"""Whole-house and climate-group thermostat selectors for GREE Controller."""
+"""Whole-house, zone and climate-group thermostat selectors for GREE Controller."""
 
 from __future__ import annotations
 
@@ -45,12 +45,16 @@ async def async_setup_entry(
     entry: ConfigEntry,
     async_add_entities: AddConfigEntryEntitiesCallback,
 ) -> None:
-    """Create whole-house thermostat mode and work-profile selectors."""
+    """Create whole-house, zone and climate-group thermostat selectors."""
     runtime: GreeControllerRuntimeData = entry.runtime_data
     entities: list[SelectEntity] = [
         GreeControllerHouseModeSelect(runtime.coordinator),
         GreeControllerHousePresetSelect(runtime.coordinator),
     ]
+    for zone in runtime.coordinator.plan.get("zones", []):
+        zone_id = str(zone.get("zone_id") or "").strip()
+        if zone_id:
+            entities.append(GreeControllerZonePresetSelect(runtime.coordinator, zone_id))
     for group_id in runtime.coordinator.groups:
         entities.append(GreeControllerGroupModeSelect(runtime.coordinator, group_id))
         entities.append(GreeControllerGroupPresetSelect(runtime.coordinator, group_id))
@@ -116,6 +120,56 @@ class GreeControllerHousePresetSelect(_GreeControllerHouseSelect):
         await self.coordinator.async_request_refresh()
 
 
+class GreeControllerZonePresetSelect(CoordinatorEntity[GreeControllerCoordinator], SelectEntity):
+    """Select the temporary work profile for one thermostat zone."""
+
+    _attr_has_entity_name = True
+    _attr_name = "Work profile"
+    _attr_options = list(HOUSE_PRESET_VALUES)
+
+    def __init__(self, coordinator: GreeControllerCoordinator, zone_id: str) -> None:
+        super().__init__(coordinator)
+        self._zone_id = zone_id
+        self._attr_unique_id = f"{zone_id}-work-profile"
+
+    @property
+    def _zone(self) -> dict[str, Any]:
+        for zone in self.coordinator.plan.get("zones", []):
+            if str(zone.get("zone_id")) == self._zone_id:
+                return zone
+        return {}
+
+    @property
+    def available(self) -> bool:
+        return super().available and bool(self._zone)
+
+    @property
+    def current_option(self) -> str | None:
+        value = str(self._zone.get("preset_override") or "auto")
+        return HOUSE_PRESET_LABELS.get(value)
+
+    @property
+    def device_info(self) -> DeviceInfo:
+        zone = self._zone
+        return DeviceInfo(
+            identifiers={(DOMAIN, f"zone:{self._zone_id}")},
+            name=str(zone.get("zone_name") or self._zone_id),
+            manufacturer="GREE Controller",
+            model="Zone thermostat",
+            via_device=(DOMAIN, "controller"),
+        )
+
+    @property
+    def extra_state_attributes(self) -> dict[str, Any]:
+        return {"active_profile": self._zone.get("preset")}
+
+    async def async_select_option(self, option: str) -> None:
+        value = HOUSE_PRESET_VALUES.get(option)
+        if value is None:
+            return
+        await self.coordinator.async_zone_control(self._zone_id, {"preset": value})
+
+
 class _GreeControllerGroupSelect(CoordinatorEntity[GreeControllerCoordinator], SelectEntity):
     """Base class for climate-group selectors."""