await the on/off switch and update pymammotion

custom_components/mammotion/coordinator.py

@@ -87,18 +87,18 @@ def __init__(
         self.update_failures = 0
         self.enabled = True
 
-    def set_scheduled_updates(self, enabled: bool) -> None:
+    async def set_scheduled_updates(self, enabled: bool) -> None:
         self.enabled = enabled
         device = self.manager.get_device_by_name(self.device_name)
         if self.enabled:
             if device.has_cloud():
-                device.cloud().start()
+                await device.cloud().start()
         else:
             if device.has_cloud():
-                device.cloud().stop()
+                await device.cloud().stop()
                 device.cloud().mqtt.disconnect()
             if device.has_ble():
-                device.ble().stop()
+                await device.ble().stop()
 
     async def async_login(self) -> None:
         """Login to cloud servers."""

custom_components/mammotion/manifest.json

@@ -30,6 +30,6 @@
   ],
   "iot_class": "local_push",
   "requirements": [
-    "pymammotion==0.2.67"
+    "pymammotion==0.2.68"
   ]
 }

custom_components/mammotion/switch.py

@@ -26,7 +26,7 @@ class MammotionUpdateSwitchEntityDescription(SwitchEntityDescription):
     """Describes Mammotion switch entity."""
 
     key: str
-    set_fn: Callable[[MammotionDataUpdateCoordinator, bool], None]
+    set_fn: Callable[[MammotionDataUpdateCoordinator, bool], Awaitable[None]]
     is_on_func: Callable[[MammotionDataUpdateCoordinator], bool]
 
 
@@ -214,12 +214,12 @@ def is_on(self) -> bool:
 
     async def async_turn_on(self, **kwargs: Any) -> None:
         self._attr_is_on = True
-        self.entity_description.set_fn(self.coordinator, True)
+        await self.entity_description.set_fn(self.coordinator, True)
         self.async_write_ha_state()
 
     async def async_turn_off(self, **kwargs: Any) -> None:
         self._attr_is_on = False
-        self.entity_description.set_fn(self.coordinator, False)
+        await self.entity_description.set_fn(self.coordinator, False)
         self.async_write_ha_state()
 
 

pyproject.toml

@@ -7,7 +7,7 @@ readme = "README.md"
 
 [tool.poetry.dependencies]
 python = "~3.12.0"
-pymammotion = "0.2.67"
+pymammotion = "0.2.68"
 homeassistant = "^2024.9.0"
 autotyping = "^24.3.0"