fix missing await

pail23 · Jun 28, 2024 · ea5e2d9 · ea5e2d9
1 parent 7751e4d
commit ea5e2d9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/custom_components/stiebel_eltron_isg/climate.py b/custom_components/stiebel_eltron_isg/climate.py
@@ -212,12 +212,12 @@ def set_temperature(self, **kwargs) -> None:
         """Set new target temperature."""
         value = kwargs["temperature"]
         if self.coordinator.data.get(OPERATION_MODE) == ECO_MODE:
-            self.coordinator.set_data(
+            await self.coordinator.set_data(
                 TEMPERATURE_KEY_MAP[self.entity_description.key][0],
                 value,
             )
         else:
-            self.coordinator.set_data(
+            await self.coordinator.set_data(
                 TEMPERATURE_KEY_MAP[self.entity_description.key][1],
                 value,
             )
@@ -260,7 +260,7 @@ def hvac_mode(self) -> HVACMode | None:
     def set_hvac_mode(self, hvac_mode: HVACMode) -> None:
         """Set new operation mode."""
         new_mode = HA_TO_WPM_HVAC.get(hvac_mode)
-        self.coordinator.set_data(OPERATION_MODE, new_mode)
+        await self.coordinator.set_data(OPERATION_MODE, new_mode)
 
     @property
     def preset_mode(self) -> str | None:
@@ -270,7 +270,7 @@ def preset_mode(self) -> str | None:
     def set_preset_mode(self, preset_mode):
         """Set new target preset mode."""
         new_mode = HA_TO_WPM_PRESET.get(preset_mode)
-        self.coordinator.set_data(OPERATION_MODE, new_mode)
+        await self.coordinator.set_data(OPERATION_MODE, new_mode)
 
 
 class StiebelEltronLWZClimateEntity(StiebelEltronISGClimateEntity):
@@ -302,7 +302,7 @@ def hvac_mode(self) -> HVACMode | None:
     def set_hvac_mode(self, hvac_mode: HVACMode) -> None:
         """Set new operation mode."""
         new_mode = HA_TO_LWZ_HVAC.get(hvac_mode)
-        self.coordinator.set_data(OPERATION_MODE, new_mode)
+        await self.coordinator.set_data(OPERATION_MODE, new_mode)
 
     @property
     def preset_mode(self) -> str | None:
@@ -312,7 +312,7 @@ def preset_mode(self) -> str | None:
     def set_preset_mode(self, preset_mode):
         """Set new target preset mode."""
         new_mode = HA_TO_LWZ_PRESET.get(preset_mode)
-        self.coordinator.set_data(OPERATION_MODE, new_mode)
+        await self.coordinator.set_data(OPERATION_MODE, new_mode)
 
     @property
     def fan_mode(self) -> str | None:
@@ -325,6 +325,6 @@ def set_fan_mode(self, fan_mode: str) -> None:
         """Set new target fan mode."""
         new_mode = HA_TO_LWZ_FAN.get(fan_mode)
         if self.coordinator.data.get(OPERATION_MODE) == ECO_MODE:
-            self.coordinator.set_data(FAN_LEVEL_NIGHT, new_mode)
+            await self.coordinator.set_data(FAN_LEVEL_NIGHT, new_mode)
         else:
-            self.coordinator.set_data(FAN_LEVEL_DAY, new_mode)
+            await self.coordinator.set_data(FAN_LEVEL_DAY, new_mode)
diff --git a/custom_components/stiebel_eltron_isg/switch.py b/custom_components/stiebel_eltron_isg/switch.py
@@ -103,12 +103,12 @@ def is_on(self) -> bool:
 
     async def async_turn_on(self, **kwargs: Any) -> None:
         """Turn the switch on."""
-        self.coordinator.set_data(self.entity_description.key, 1)
+        await self.coordinator.set_data(self.entity_description.key, 1)
         await self.async_update()
 
     async def async_turn_off(self, **kwargs: Any) -> None:
         """Turn the device off."""
-        self.coordinator.set_data(self.entity_description.key, 0)
+        await self.coordinator.set_data(self.entity_description.key, 0)
         await self.async_update()
 
     @property