fix: update ha state on command success

closes #102
alandtse · Dec 6, 2021 · 6ede864 · 6ede864
1 parent 3a3c0fb
commit 6ede864
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/custom_components/tesla_custom/climate.py b/custom_components/tesla_custom/climate.py
@@ -87,6 +87,8 @@ async def async_set_temperature(self, **kwargs):
         if temperature:
             _LOGGER.debug("%s: Setting temperature to %s", self.name, temperature)
             await self.tesla_device.set_temperature(temperature)
+            self.async_write_ha_state()
+
 
     @TeslaDevice.Decorators.check_for_reauth
     async def async_set_hvac_mode(self, hvac_mode):
@@ -96,13 +98,15 @@ async def async_set_hvac_mode(self, hvac_mode):
             await self.tesla_device.set_status(False)
         elif hvac_mode == HVAC_MODE_HEAT_COOL:
             await self.tesla_device.set_status(True)
+        self.async_write_ha_state()
 
     @TeslaDevice.Decorators.check_for_reauth
     async def async_set_preset_mode(self, preset_mode: str) -> None:
         """Set new preset mode."""
         _LOGGER.debug("%s: Setting preset_mode to: %s", self.name, preset_mode)
         try:
             await self.tesla_device.set_preset_mode(preset_mode)
+            self.async_write_ha_state()
         except UnknownPresetMode as ex:
             _LOGGER.error("%s", ex.message)
 

diff --git a/custom_components/tesla_custom/lock.py b/custom_components/tesla_custom/lock.py
@@ -29,12 +29,14 @@ async def async_lock(self, **kwargs):
         """Send the lock command."""
         _LOGGER.debug("Locking doors for: %s", self.name)
         await self.tesla_device.lock()
+        self.async_write_ha_state()
 
     @TeslaDevice.Decorators.check_for_reauth
     async def async_unlock(self, **kwargs):
         """Send the unlock command."""
         _LOGGER.debug("Unlocking doors for: %s", self.name)
         await self.tesla_device.unlock()
+        self.async_write_ha_state()
 
     @property
     def is_locked(self):