feat: add time datetime for valley (#65)

Author: Lash-L

roborock/api.py

@@ -35,6 +35,7 @@
     SmartWashParams,
     Status,
     UserData,
+    ValleyElectricityTimer,
     WashTowelMode,
 )
 from .exceptions import (
@@ -46,6 +47,7 @@
     RoborockNoUserAgreement,
     RoborockTimeout,
     RoborockUrlException,
+    UnknownMethodError,
     VacuumError,
 )
 from .protocol import Utils
@@ -189,6 +191,8 @@ async def _async_response(self, request_id: int, protocol_id: int = 0) -> tuple[
             queue = RoborockFuture(protocol_id)
             self._waiting_queue[request_id] = queue
             (response, err) = await queue.async_get(QUEUE_TIMEOUT)
+            if response == "unknown_method":
+                raise UnknownMethodError("Unknown method")
             return response, err
         except (asyncio.TimeoutError, asyncio.CancelledError):
             raise RoborockTimeout(f"id={request_id} Timeout after {QUEUE_TIMEOUT} seconds") from None
@@ -244,6 +248,13 @@ async def get_dnd_timer(self) -> DnDTimer | None:
             return DnDTimer.from_dict(dnd_timer)
         return None
 
+    @fallback_cache
+    async def get_valley_electricity_timer(self) -> ValleyElectricityTimer | None:
+        valley_electricity_timer = await self.send_command(RoborockCommand.GET_VALLEY_ELECTRICITY_TIMER)
+        if isinstance(valley_electricity_timer, dict):
+            return ValleyElectricityTimer.from_dict(valley_electricity_timer)
+        return None
+
     @fallback_cache
     async def get_clean_summary(self) -> CleanSummary | None:
         clean_summary = await self.send_command(RoborockCommand.GET_CLEAN_SUMMARY)

roborock/containers.py

@@ -3,6 +3,7 @@
 import logging
 import re
 from dataclasses import asdict, dataclass
+from datetime import time
 from enum import Enum
 from typing import Any, Optional, Type
 
@@ -342,6 +343,39 @@ class DnDTimer(RoborockBase):
     end_hour: Optional[int] = None
     end_minute: Optional[int] = None
     enabled: Optional[int] = None
+    start_time: Optional[time] = None
+    end_time: Optional[time] = None
+
+    def __post_init__(self) -> None:
+        self.start_time = (
+            time(hour=self.start_hour, minute=self.start_minute) if self.start_hour and self.start_minute else None
+        )
+        self.end_time = (
+            time(hour=self.end_hour, minute=self.end_minute)
+            if self.end_hour is not None and self.end_minute is not None
+            else None
+        )
+
+
+@dataclass
+class ValleyElectricityTimer(RoborockBase):
+    start_hour: Optional[int] = None
+    start_minute: Optional[int] = None
+    end_hour: Optional[int] = None
+    end_minute: Optional[int] = None
+    enabled: Optional[int] = None
+    start_time: Optional[time] = None
+    end_time: Optional[time] = None
+
+    def __post_init__(self) -> None:
+        self.start_time = (
+            time(hour=self.start_hour, minute=self.start_minute) if self.start_hour and self.start_minute else None
+        )
+        self.end_time = (
+            time(hour=self.end_hour, minute=self.end_minute)
+            if self.end_hour is not None and self.end_minute is not None
+            else None
+        )
 
 
 @dataclass

roborock/exceptions.py

@@ -29,6 +29,10 @@ def __init__(self, command: str, vacuum_error: VacuumError):
         super().__init__(self.message)
 
 
+class UnknownMethodError(RoborockException):
+    """Class for an invalid method being sent."""
+
+
 class RoborockAccountDoesNotExist(RoborockException):
     """Class for Roborock account does not exist exceptions."""