Merge pull request #23 from vincentwolsink/async

Async
vincentwolsink · Aug 10, 2023 · cce4c63 · cce4c63
2 parents 87e0f60 + bba7868
commit cce4c63
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 13 deletions.
diff --git a/custom_components/aguaiot/climate.py b/custom_components/aguaiot/climate.py
@@ -21,7 +21,6 @@
     PRECISION_WHOLE,
     TEMP_CELSIUS,
 )
-
 from .const import (
     ATTR_DEVICE_ALARM,
     ATTR_DEVICE_STATUS,
@@ -34,6 +33,7 @@
     AGUA_STATUS_ON,
     CURRENT_HVAC_MAP_AGUA_HEAT,
 )
+from py_agua_iot import Error as AguaIOTError
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -169,47 +169,57 @@ def hvac_action(self):
             return CURRENT_HVAC_MAP_AGUA_HEAT.get(self._device.status_translated)
         return CURRENT_HVAC_IDLE
 
-    def turn_off(self):
+    async def async_turn_off(self):
         """Turn device off."""
         try:
-            self._device.turn_off()
+            await self.hass.async_add_executor_job(self._device.turn_off)
+            await self.coordinator.async_request_refresh()
         except AguaIOTError as err:
             _LOGGER.error("Failed to turn off device, error: %s", err)
 
-    def turn_on(self):
+    async def async_turn_on(self):
         """Turn device on."""
         try:
-            self._device.turn_on()
+            await self.hass.async_add_executor_job(self._device.turn_on)
+            await self.coordinator.async_request_refresh()
         except AguaIOTError as err:
             _LOGGER.error("Failed to turn on device, error: %s", err)
 
-    def set_temperature(self, **kwargs):
+    async def async_set_temperature(self, **kwargs):
         """Set new target temperature."""
         temperature = kwargs.get(ATTR_TEMPERATURE)
         if temperature is None:
             return
 
         try:
             if self._device.air_temperature is not None:
-                self._device.set_air_temperature = temperature
+                await self.hass.async_add_executor_job(
+                    setattr, self._device, "set_air_temperature", temperature
+                )
             elif self._device.water_temperature is not None:
-                self._device.set_water_temperature = temperature
+                await self.hass.async_add_executor_job(
+                    setattr, self._device, "set_water_temperature", temperature
+                )
+            await self.coordinator.async_request_refresh()
         except (ValueError, AguaIOTError) as err:
             _LOGGER.error("Failed to set temperature, error: %s", err)
 
-    def set_fan_mode(self, fan_mode):
+    async def async_set_fan_mode(self, fan_mode):
         """Set new target fan mode."""
         if fan_mode is None or not fan_mode.isdigit():
             return
 
         try:
-            self._device.set_power = int(fan_mode)
+            await self.hass.async_add_executor_job(
+                setattr, self._device, "set_power", int(fan_mode)
+            )
+            await self.coordinator.async_request_refresh()
         except AguaIOTError as err:
             _LOGGER.error("Failed to set fan mode, error: %s", err)
 
-    def set_hvac_mode(self, hvac_mode):
+    async def async_set_hvac_mode(self, hvac_mode):
         """Set new target hvac mode."""
         if hvac_mode == HVAC_MODE_OFF:
-            self.turn_off()
+            await self.async_turn_off()
         elif hvac_mode == HVAC_MODE_HEAT:
-            self.turn_on()
+            await self.async_turn_on()
diff --git a/custom_components/aguaiot/const.py b/custom_components/aguaiot/const.py
@@ -56,6 +56,7 @@
     SensorEntityDescription(
         key="status_translated",
         name="Status",
+        icon="mdi:fire",
         native_unit_of_measurement=None,
         state_class=None,
         device_class=None,