24 files

custom_components/midea_ac_lan/sensor.py

@@ -13,7 +13,7 @@ from homeassistant.core import HomeAssistant
 from homeassistant.helpers.entity_platform import AddEntitiesCallback
 from homeassistant.helpers.restore_state import RestoreEntity
 from homeassistant.helpers.typing import StateType
-from midealocal.device import MideaDevice
+from midealan.device import MideaDevice
 
 from .const import DEVICES, DOMAIN, supports_model
 from .midea_devices import MIDEA_DEVICES
@@ -185,9 +185,7 @@ class MideaEstimatedUsageSensor(MideaSensor, RestoreEntity):
 
         self._last_progress = current_progress
         super().update_state(status)
-        if (
-            self.hass
-            and not self.hass.is_stopping
-            and ("progress" in status or "status" in status or "mode" in status)
+        if self.hass and (
+            "progress" in status or "status" in status or "mode" in status
         ):
-            self.schedule_update_ha_state()
+            self.schedule_update_if_running()