feat: add dust collection mode name for typing ease (#271)

Python-roborock · Jan 9, 2025 · c85232a · c85232a
1 parent 2356c16
commit c85232a
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/roborock/roborock_typing.py b/roborock/roborock_typing.py
@@ -461,6 +461,11 @@ class DeviceProp(RoborockBase):
     consumable: Consumable = field(default_factory=Consumable)
     last_clean_record: CleanRecord | None = None
     dock_summary: DockSummary | None = None
+    dust_collection_mode_name: str | None = None
+
+    def __post_init__(self) -> None:
+        if self.dock_summary and self.dock_summary.dust_collection_mode and self.dock_summary.dust_collection_mode.mode:
+            self.dust_collection_mode_name = self.dock_summary.dust_collection_mode.mode.name
 
     def update(self, device_prop: DeviceProp) -> None:
         if device_prop.status:
@@ -473,3 +478,4 @@ def update(self, device_prop: DeviceProp) -> None:
             self.last_clean_record = device_prop.last_clean_record
         if device_prop.dock_summary:
             self.dock_summary = device_prop.dock_summary
+        self.__post_init__()