allow setting wake up sleep schedule

Author: zweckj

lmcloud/client_cloud.py

@@ -26,7 +26,7 @@
     SmartStandbyMode,
 )
 from .exceptions import AuthFail, RequestNotSuccessful
-from .models import LaMarzoccoFirmware, LaMarzoccoDeviceInfo
+from .models import LaMarzoccoFirmware, LaMarzoccoDeviceInfo, LaMarzoccoWakeUpSleepEntry
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -274,6 +274,25 @@ async def set_dose_hot_water(
     #         return True
     #     return False
 
+    async def set_wake_up_sleep(
+        self, serial_number: str, wake_up_sleep_entry: LaMarzoccoWakeUpSleepEntry
+    ) -> bool:
+        """Enable or disable wake-up sleep mode"""
+
+        url = f"{GW_MACHINE_BASE_URL}/{serial_number}/wake-up-sleep/{wake_up_sleep_entry.entry_id}"
+        data = {
+            "days": [day.value for day in wake_up_sleep_entry.days],
+            "enable": wake_up_sleep_entry.enabled,
+            "id": wake_up_sleep_entry.entry_id,
+            "steam": wake_up_sleep_entry.steam,
+            "timeOff": wake_up_sleep_entry.time_off,
+            "timeOn": wake_up_sleep_entry.time_on,
+        }
+        response = await self._rest_api_call(url=url, method=HTTPMethod.PUT, data=data)
+        if await self._check_cloud_command_status(serial_number, response):
+            return True
+        return False
+
     async def set_smart_standby(
         self,
         serial_number: str,

lmcloud/lm_machine.py

@@ -35,6 +35,7 @@
     LaMarzoccoCoffeeStatistics,
     LaMarzoccoMachineConfig,
     LaMarzoccoSmartStandby,
+    LaMarzoccoWakeUpSleepEntry,
 )
 
 _LOGGER = logging.getLogger(__name__)
@@ -333,6 +334,18 @@ async def start_backflush(self) -> None:
     #     schedule.days[day] = day_settings
     #     return await self.set_schedule(schedule)
 
+    async def set_wake_up_sleep(self, wake_up_sleep_entry: LaMarzoccoWakeUpSleepEntry):
+        """Set wake up sleep"""
+
+        if await self.cloud_client.set_wake_up_sleep(
+            self.serial_number, wake_up_sleep_entry
+        ):
+            for idx, entry in enumerate(self.config.wake_up_sleep_entries):
+                if entry.entry_id == wake_up_sleep_entry.entry_id:
+                    self.config.wake_up_sleep_entries[idx] = wake_up_sleep_entry
+                    return True
+        return False
+
     async def set_smart_standby(
         self, enabled: bool, minutes: int, mode: SmartStandbyMode
     ) -> bool:

setup.py

@@ -7,7 +7,7 @@
 
 setuptools.setup(
     name="lmcloud",
-    version="1.1.6",
+    version="1.1.7",
     description="A Python implementation of the new La Marzocco API",
     long_description=readme,
     long_description_content_type="text/markdown",