feat: adding valley_electricity_timer to props

humbertogontijo · Jun 14, 2023 · 0844067 · 0844067
1 parent d80aab0
commit 0844067
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/roborock/api.py b/roborock/api.py
@@ -323,12 +323,13 @@ async def get_dock_summary(self, dock_type: RoborockDockTypeCode) -> DockSummary
     @fallback_cache
     async def get_prop(self) -> DeviceProp | None:
         """Gets device general properties."""
-        [status, dnd_timer, clean_summary, consumable] = await asyncio.gather(
+        [status, clean_summary, consumable, dnd_timer, valley_electricity_timer] = await asyncio.gather(
             *[
                 self.get_status(),
-                self.get_dnd_timer(),
                 self.get_clean_summary(),
                 self.get_consumable(),
+                self.get_dnd_timer(),
+                self.get_valley_electricity_timer(),
             ]
         )
         last_clean_record = None
@@ -340,9 +341,10 @@ async def get_prop(self) -> DeviceProp | None:
         if any([status, dnd_timer, clean_summary, consumable]):
             return DeviceProp(
                 status,
-                dnd_timer,
                 clean_summary,
                 consumable,
+                dnd_timer,
+                valley_electricity_timer,
                 last_clean_record,
                 dock_summary,
             )

diff --git a/roborock/roborock_typing.py b/roborock/roborock_typing.py
@@ -13,7 +13,7 @@
     RoborockBase,
     SmartWashParams,
     Status,
-    WashTowelMode,
+    ValleyElectricityTimer, WashTowelMode,
 )
 
 
@@ -311,9 +311,10 @@ class DockSummary(RoborockBase):
 @dataclass
 class DeviceProp(RoborockBase):
     status: Optional[Status] = None
-    dnd_timer: Optional[DnDTimer] = None
     clean_summary: Optional[CleanSummary] = None
     consumable: Optional[Consumable] = None
+    dnd_timer: Optional[DnDTimer] = None
+    valley_electricity_timer: Optional[ValleyElectricityTimer] = None
     last_clean_record: Optional[CleanRecord] = None
     dock_summary: Optional[DockSummary] = None