Add additional sensors and settings to Roborock vacuums

miio/device.py

@@ -7,8 +7,9 @@
 from .click_common import DeviceGroupMeta, LiteralParamType, command, format_output
 from .descriptors import (
     ButtonDescriptor,
+    EnumSettingDescriptor,
+    NumberSettingDescriptor,
     SensorDescriptor,
-    SettingDescriptor,
     SwitchDescriptor,
 )
 from .deviceinfo import DeviceInfo
@@ -242,9 +243,13 @@ def buttons(self) -> List[ButtonDescriptor]:
         """Return a list of button-like, clickable actions of the device."""
         return []
 
-    def settings(self) -> Dict[str, SettingDescriptor]:
+    def settings(
+        self,
+    ) -> Dict[str, Union[EnumSettingDescriptor, NumberSettingDescriptor]]:
         """Return list of settings."""
-        settings = self.status().settings()
+        settings = (
+            self.status().settings()
+        )  # NOTE that this already does IO so schould be run in executer job in HA
         for setting in settings.values():
             # TODO: Bind setter methods, this should probably done only once during init.
             if setting.setter is None:
@@ -255,6 +260,12 @@ def settings(self) -> Dict[str, SettingDescriptor]:
                     )
 
                 setting.setter = getattr(self, setting.setter_name)
+            if (
+                isinstance(setting, EnumSettingDescriptor)
+                and setting.choices_attribute is not None
+            ):
+                retrieve_choices_function = getattr(self, setting.choices_attribute)
+                setting.choices = retrieve_choices_function()
 
         return settings
 

miio/devicestatus.py

@@ -100,7 +100,9 @@ def switches(self) -> Dict[str, SwitchDescriptor]:
         """
         return self._switches  # type: ignore[attr-defined]
 
-    def settings(self) -> Dict[str, SettingDescriptor]:
+    def settings(
+        self,
+    ) -> Dict[str, Union[EnumSettingDescriptor, NumberSettingDescriptor]]:
         """Return the dict of settings exposed by the status container.
 
         You can use @setting decorator to define sensors inside your status class.
@@ -155,7 +157,8 @@ def sensor(name: str, *, unit: str = "", **kwargs):
     """
 
     def decorator_sensor(func):
-        property_name = func.__name__
+        property_name = str(func.__name__)
+        qualified_name = str(func.__qualname__)
 
         def _sensor_type_for_return_type(func):
             rtype = get_type_hints(func).get("return")
@@ -169,8 +172,8 @@ def _sensor_type_for_return_type(func):
 
         sensor_type = _sensor_type_for_return_type(func)
         descriptor = SensorDescriptor(
-            id=str(property_name),
-            property=str(property_name),
+            id=qualified_name,
+            property=property_name,
             name=name,
             unit=unit,
             type=sensor_type,
@@ -194,12 +197,13 @@ def switch(name: str, *, setter_name: str, **kwargs):
     and can be interpreted downstream users as they wish.
     """
 
-    def decorator_sensor(func):
-        property_name = func.__name__
+    def decorator_switch(func):
+        property_name = str(func.__name__)
+        qualified_name = str(func.__qualname__)
 
         descriptor = SwitchDescriptor(
-            id=str(property_name),
-            property=str(property_name),
+            id=qualified_name,
+            property=property_name,
             name=name,
             setter_name=setter_name,
             extras=kwargs,
@@ -208,7 +212,7 @@ def decorator_sensor(func):
 
         return func
 
-    return decorator_sensor
+    return decorator_switch
 
 
 def setting(
@@ -235,15 +239,16 @@ def setting(
     """
 
     def decorator_setting(func):
-        property_name = func.__name__
+        property_name = str(func.__name__)
+        qualified_name = str(func.__qualname__)
 
         if setter is None and setter_name is None:
             raise Exception("Either setter or setter_name needs to be defined")
 
         if min_value or max_value:
             descriptor = NumberSettingDescriptor(
-                id=str(property_name),
-                property=str(property_name),
+                id=qualified_name,
+                property=property_name,
                 name=name,
                 unit=unit,
                 setter=setter,
@@ -254,13 +259,13 @@ def decorator_setting(func):
                 extras=kwargs,
             )
         elif choices or choices_attribute:
-            if choices_attribute is not None:
-                # TODO: adding choices from attribute is a bit more complex, as it requires a way to
-                # construct enums pointed by the attribute
-                raise NotImplementedError("choices_attribute is not yet implemented")
+            # if choices_attribute is not None:
+            # TODO: adding choices from attribute is a bit more complex, as it requires a way to
+            # construct enums pointed by the attribute
+            # raise NotImplementedError("choices_attribute is not yet implemented")
             descriptor = EnumSettingDescriptor(
-                id=str(property_name),
-                property=str(property_name),
+                id=qualified_name,
+                property=property_name,
                 name=name,
                 unit=unit,
                 setter=setter,

miio/integrations/vacuum/roborock/tests/test_vacuum.py

@@ -68,6 +68,45 @@ def __init__(self, *args, **kwargs):
                 1487548800,
             ],
         ]
+        self.dummies["dnd_timer"] = [
+            {
+                "enabled": 1,
+                "start_minute": 0,
+                "end_minute": 0,
+                "start_hour": 22,
+                "end_hour": 8,
+            }
+        ]
+        self.dummies["multi_maps"] = [
+            {
+                "max_multi_map": 4,
+                "max_bak_map": 1,
+                "multi_map_count": 3,
+                "map_info": [
+                    {
+                        "mapFlag": 0,
+                        "add_time": 1664448893,
+                        "length": 10,
+                        "name": "Downstairs",
+                        "bak_maps": [{"mapFlag": 4, "add_time": 1663577737}],
+                    },
+                    {
+                        "mapFlag": 1,
+                        "add_time": 1663580330,
+                        "length": 8,
+                        "name": "Upstairs",
+                        "bak_maps": [{"mapFlag": 5, "add_time": 1663577752}],
+                    },
+                    {
+                        "mapFlag": 2,
+                        "add_time": 1663580384,
+                        "length": 5,
+                        "name": "Attic",
+                        "bak_maps": [{"mapFlag": 6, "add_time": 1663577765}],
+                    },
+                ],
+            }
+        ]
 
         self.return_values = {
             "get_status": lambda x: [self.state],
@@ -81,8 +120,16 @@ def __init__(self, *args, **kwargs):
             "app_zoned_clean": lambda x: self.change_mode("zoned clean"),
             "app_charge": lambda x: self.change_mode("charge"),
             "miIO.info": "dummy info",
+            "get_clean_record": lambda x: [[1488347071, 1488347123, 16, 0, 0, 0]],
+            "get_dnd_timer": lambda x: self.dummies["dnd_timer"],
+            "get_multi_maps_list": lambda x: self.dummies["multi_maps"],
         }
 
+        self._multi_maps = None
+
+        self._floor_clean_details = {}
+        self._searched_clean_id = None
+
         super().__init__(args, kwargs)
 
     def change_mode(self, new_mode):