python-roborock 4.10.1__tar.gz → 4.11.0__tar.gz

{python_roborock-4.10.1 → python_roborock-4.11.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-roborock
-Version: 4.10.1
+Version: 4.11.0
 Summary: A package to control Roborock vacuums.
 Project-URL: Repository, https://github.com/humbertogontijo/python-roborock
 Project-URL: Documentation, https://python-roborock.readthedocs.io/

{python_roborock-4.10.1 → python_roborock-4.11.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "python-roborock"
-version = "4.10.1"
+version = "4.11.0"
 description = "A package to control Roborock vacuums."
 authors = [{ name = "humbertogontijo", email = "humbertogontijo@users.noreply.github.com" }, {name="Lash-L"}, {name="allenporter"}]
 requires-python = ">=3.11, <4"

{python_roborock-4.10.1 → python_roborock-4.11.0}/roborock/data/b01_q7/b01_q7_code_mappings.py

@@ -46,8 +46,15 @@ class CleanTypeMapping(RoborockModeEnum):
 class CleanRepeatMapping(RoborockModeEnum):
     """Maps the cleaning repeat parameter."""
 
-    ONCE = ("once", 0)
-    TWICE = ("twice", 1)
+    ONE = ("one", 0)
+    TWO = ("two", 1)
+
+
+class CleanPathPreferenceMapping(RoborockModeEnum):
+    """Maps the cleaning path preference parameter."""
+
+    BALANCED = ("balanced", 0)
+    DEEP = ("deep", 1)
 
 
 class SCDeviceCleanParam(RoborockModeEnum):

{python_roborock-4.10.1 → python_roborock-4.11.0}/roborock/data/b01_q7/b01_q7_containers.py

@@ -7,6 +7,8 @@ from ...exceptions import RoborockException
 from ..containers import RoborockBase
 from .b01_q7_code_mappings import (
     B01Fault,
+    CleanPathPreferenceMapping,
+    CleanRepeatMapping,
     CleanTypeMapping,
     SCWindMapping,
     WaterLevelMapping,
@@ -94,10 +96,10 @@ class B01Props(RoborockBase):
     mop_life: int | None = None
     main_sensor: int | None = None
     net_status: NetStatus | None = None
-    repeat_state: int | None = None
+    repeat_state: CleanRepeatMapping | None = None
     tank_state: int | None = None
     sweep_type: int | None = None
-    clean_path_preference: int | None = None
+    clean_path_preference: CleanPathPreferenceMapping | None = None
     cloth_state: int | None = None
     time_zone: int | None = None
     time_zone_info: str | None = None
@@ -210,6 +212,16 @@ class B01Props(RoborockBase):
         """Returns the name of the current work mode."""
         return self.work_mode.value if self.work_mode is not None else None
 
+    @property
+    def repeat_state_name(self) -> str | None:
+        """Returns the name of the current repeat state."""
+        return self.repeat_state.value if self.repeat_state is not None else None
+
+    @property
+    def clean_path_preference_name(self) -> str | None:
+        """Returns the name of the current clean path preference."""
+        return self.clean_path_preference.value if self.clean_path_preference is not None else None
+
 
 @dataclass
 class CleanRecordDetail(RoborockBase):

{python_roborock-4.10.1 → python_roborock-4.11.0}/roborock/devices/traits/b01/q7/__init__.py

@@ -5,6 +5,8 @@ from typing import Any
 
 from roborock import B01Props
 from roborock.data.b01_q7.b01_q7_code_mappings import (
+    CleanPathPreferenceMapping,
+    CleanRepeatMapping,
     CleanTaskTypeMapping,
     CleanTypeMapping,
     SCDeviceCleanParam,
@@ -66,6 +68,14 @@ class Q7PropertiesApi(Trait):
         """Set the cleaning mode (vacuum, mop, or vacuum and mop)."""
         await self.set_prop(RoborockB01Props.MODE, mode.code)
 
+    async def set_clean_path_preference(self, preference: CleanPathPreferenceMapping) -> None:
+        """Set the cleaning path preference (route)."""
+        await self.set_prop(RoborockB01Props.CLEAN_PATH_PREFERENCE, preference.code)
+
+    async def set_repeat_state(self, repeat: CleanRepeatMapping) -> None:
+        """Set the cleaning repeat state (cycles)."""
+        await self.set_prop(RoborockB01Props.REPEAT_STATE, repeat.code)
+
     async def start_clean(self) -> None:
         """Start cleaning."""
         await self.send(