Control air conditioners

custom_components/hon/climate.py

@@ -0,0 +1,148 @@
+import logging
+
+from homeassistant.components.climate import (
+    ClimateEntity,
+    ClimateEntityDescription,
+)
+from homeassistant.components.climate.const import (
+    FAN_OFF,
+    SWING_OFF,
+    SWING_BOTH,
+    SWING_VERTICAL,
+    SWING_HORIZONTAL,
+    ClimateEntityFeature,
+    HVACMode,
+)
+from homeassistant.config_entries import ConfigEntry
+from homeassistant.const import (
+    ATTR_TEMPERATURE,
+    PRECISION_WHOLE,
+    TEMP_CELSIUS,
+)
+from homeassistant.core import callback
+from pyhon import Hon
+from pyhon.appliance import HonAppliance
+
+from custom_components.hon.const import HON_HVAC_MODE, HON_FAN, HON_HVAC_PROGRAM, DOMAIN
+from custom_components.hon.hon import HonEntity, HonCoordinator
+
+_LOGGER = logging.getLogger(__name__)
+
+CLIMATES = {
+    "AC": (ClimateEntityDescription(key="startProgram"),),
+}
+
+
+async def async_setup_entry(hass, entry: ConfigEntry, async_add_entities) -> None:
+    hon: Hon = hass.data[DOMAIN][entry.unique_id]
+    coordinators = hass.data[DOMAIN]["coordinators"]
+    appliances = []
+    for device in hon.appliances:
+        if device.unique_id in coordinators:
+            coordinator = hass.data[DOMAIN]["coordinators"][device.unique_id]
+        else:
+            coordinator = HonCoordinator(hass, device)
+            hass.data[DOMAIN]["coordinators"][device.unique_id] = coordinator
+        await coordinator.async_config_entry_first_refresh()
+
+        if descriptions := CLIMATES.get(device.appliance_type):
+            for description in descriptions:
+                if not device.settings.get(description.key):
+                    continue
+                appliances.extend(
+                    [HonClimateEntity(hass, coordinator, entry, device, description)]
+                )
+    async_add_entities(appliances)
+
+
+class HonClimateEntity(HonEntity, ClimateEntity):
+    def __init__(
+        self, hass, coordinator, entry, device: HonAppliance, description
+    ) -> None:
+        super().__init__(hass, entry, coordinator, device)
+        self._coordinator = coordinator
+        self._device = coordinator.device
+        self.entity_description = description
+        self._hass = hass
+        self._attr_unique_id = f"{super().unique_id}climate"
+
+        self._attr_temperature_unit = TEMP_CELSIUS
+        self._attr_target_temperature_step = PRECISION_WHOLE
+        self._attr_max_temp = device.settings["tempSel"].max
+        self._attr_min_temp = device.settings["tempSel"].min
+
+        self._attr_hvac_modes = [HVACMode.OFF] + [
+            HON_HVAC_MODE[mode] for mode in device.settings["machMode"].values
+        ]
+        self._attr_fan_modes = [FAN_OFF] + [
+            HON_FAN[mode] for mode in device.settings["windSpeed"].values
+        ]
+        self._attr_swing_modes = [
+            SWING_OFF,
+            SWING_VERTICAL,
+            SWING_HORIZONTAL,
+            SWING_BOTH,
+        ]
+        self._attr_supported_features = (
+            ClimateEntityFeature.TARGET_TEMPERATURE
+            | ClimateEntityFeature.FAN_MODE
+            | ClimateEntityFeature.SWING_MODE
+        )
+
+    async def async_set_hvac_mode(self, hvac_mode):
+        if hvac_mode == HVACMode.OFF:
+            self._device.commands["stopProgram"].send()
+        else:
+            self._device.settings["program"].value = HON_HVAC_PROGRAM[hvac_mode]
+            self._device.commands["startProgram"].send()
+        self._attr_hvac_mode = hvac_mode
+
+    async def async_set_fan_mode(self, fan_mode):
+        mode_number = list(HON_FAN.values()).index(fan_mode)
+        self._device.settings["windSpeed"].value = list(HON_FAN.keys())[mode_number]
+        self._device.commands["startProgram"].send()
+
+    async def async_set_swing_mode(self, swing_mode):
+        horizontal = self._device.settings["windDirectionHorizontal"]
+        vertical = self._device.settings["windDirectionVertical"]
+        if swing_mode in [SWING_BOTH, SWING_HORIZONTAL]:
+            horizontal.value = "7"
+        if swing_mode in [SWING_BOTH, SWING_VERTICAL]:
+            vertical.value = "8"
+        if swing_mode in [SWING_OFF, SWING_HORIZONTAL] and vertical.value == "8":
+            vertical.value = "5"
+        if swing_mode in [SWING_OFF, SWING_VERTICAL] and horizontal.value == "7":
+            horizontal.value = "0"
+        self._attr_swing_mode = swing_mode
+        self._device.commands["startProgram"].send()
+
+    async def async_set_temperature(self, **kwargs):
+        if (temperature := kwargs.get(ATTR_TEMPERATURE)) is None:
+            return False
+        self._device.settings["selTemp"].value = temperature
+        self._device.commands["startProgram"].send()
+
+    @callback
+    def _handle_coordinator_update(self, update=True) -> None:
+        self._attr_target_temperature = int(float(self._device.get("tempSel")))
+        self._attr_current_temperature = float(self._device.get("tempIndoor"))
+        self._attr_max_temp = self._device.settings["tempSel"].max
+        self._attr_min_temp = self._device.settings["tempSel"].min
+
+        if self._device.get("onOffStatus") == "0":
+            self._attr_hvac_mode = HVACMode.OFF
+        else:
+            self._attr_hvac_mode = HON_HVAC_MODE[self._device.get("machMode")]
+
+        self._attr_fan_mode = HON_FAN[self._device.settings["windSpeed"].value]
+
+        horizontal = self._device.settings["windDirectionHorizontal"]
+        vertical = self._device.settings["windDirectionVertical"]
+        if horizontal == "7" and vertical == "8":
+            self._attr_swing_mode = SWING_BOTH
+        elif horizontal == "7":
+            self._attr_swing_mode = SWING_HORIZONTAL
+        elif vertical == "8":
+            self._attr_swing_mode = SWING_VERTICAL
+        else:
+            self._attr_swing_mode = SWING_OFF