python-roborock 5.4.1__tar.gz → 5.5.0__tar.gz

{python_roborock-5.4.1 → python_roborock-5.5.0}/PKG-INFO

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

{python_roborock-5.4.1 → python_roborock-5.5.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "python-roborock"
-version = "5.4.1"
+version = "5.5.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-5.4.1 → python_roborock-5.5.0}/roborock/data/containers.py

@@ -251,11 +251,26 @@ class HomeDataProduct(RoborockBase):
 
     @cached_property
     def supported_schema_codes(self) -> set[str]:
-        """Return a set of fields that are supported by the device."""
+        """Return a set of schema codes that are supported by the device.
+
+        These correspond with string field names like "state" or "error_code" that
+        correspond to RoborockDataProtocol or RoborockB01Protocol code values.
+        """
         if self.schema is None:
             return set()
         return {schema.code for schema in self.schema if schema.code is not None}
 
+    @cached_property
+    def supported_schema_ids(self) -> set[int]:
+        """Return a set of schema IDs (DPS integers) that are supported by the device.
+
+        These correspond to RoborockMessageProtocol and RoborockDataProtocol or
+        RoborockB01Protocol enum number values (depends on the device protocol versions).
+        """
+        if self.schema is None:
+            return set()
+        return {int(schema.id) for schema in self.schema if schema.id is not None}
+
 
 @dataclass
 class HomeDataDevice(RoborockBase):

{python_roborock-5.4.1 → python_roborock-5.5.0}/roborock/data/v1/v1_containers.py

@@ -37,6 +37,7 @@ from roborock.const import (
     ROBOROCK_G20S_Ultra,
 )
 from roborock.exceptions import RoborockException
+from roborock.roborock_message import RoborockDataProtocol
 
 from ..containers import NamedRoomMapping, RoborockBase, RoborockBaseTimer, _attr_repr
 from .v1_clean_modes import WashTowelModes
@@ -102,9 +103,8 @@ class StatusField(FieldNameBase):
     This is used with `roborock.devices.traits.v1.status.DeviceFeaturesTrait`
     to understand if a feature is supported by the device using `is_field_supported`.
 
-    The enum values are names of fields in the `Status` class. Each field is
-    annotated with `requires_schema_code` metadata to map the field to a schema
-    code in the product schema, which may have a different name than the field/attribute name.
+    The enum values are names of fields in the `Status` class. Each field is annotated
+    with a metadata value to determine if the field is supported by the device.
     """
 
     STATE = "state"
@@ -116,21 +116,17 @@ class StatusField(FieldNameBase):
     ERROR_CODE = "error_code"
 
 
-def _requires_schema_code(requires_schema_code: str, default=None) -> Any:
-    return field(metadata={"requires_schema_code": requires_schema_code}, default=default)
-
-
 @dataclass
 class Status(RoborockBase):
     """This status will be deprecated in favor of StatusV2."""
 
     msg_ver: int | None = None
     msg_seq: int | None = None
-    state: RoborockStateCode | None = _requires_schema_code("state")
-    battery: int | None = _requires_schema_code("battery")
+    state: RoborockStateCode | None = field(default=None, metadata={"dps": RoborockDataProtocol.STATE})
+    battery: int | None = field(default=None, metadata={"dps": RoborockDataProtocol.BATTERY})
     clean_time: int | None = None
     clean_area: int | None = None
-    error_code: RoborockErrorCode | None = _requires_schema_code("error_code")
+    error_code: RoborockErrorCode | None = field(default=None, metadata={"dps": RoborockDataProtocol.ERROR_CODE})
     map_present: int | None = None
     in_cleaning: RoborockInCleaning | None = None
     in_returning: int | None = None
@@ -140,12 +136,14 @@ class Status(RoborockBase):
     back_type: int | None = None
     wash_phase: int | None = None
     wash_ready: int | None = None
-    fan_power: RoborockFanPowerCode | None = _requires_schema_code("fan_power")
+    fan_power: RoborockFanPowerCode | None = field(default=None, metadata={"dps": RoborockDataProtocol.FAN_POWER})
     dnd_enabled: int | None = None
     map_status: int | None = None
     is_locating: int | None = None
     lock_status: int | None = None
-    water_box_mode: RoborockMopIntensityCode | None = _requires_schema_code("water_box_mode")
+    water_box_mode: RoborockMopIntensityCode | None = field(
+        default=None, metadata={"dps": RoborockDataProtocol.WATER_BOX_MODE}
+    )
     water_box_carriage_status: int | None = None
     mop_forbidden_enable: int | None = None
     camera_status: int | None = None
@@ -163,13 +161,13 @@ class Status(RoborockBase):
     collision_avoid_status: int | None = None
     switch_map_mode: int | None = None
     dock_error_status: RoborockDockErrorCode | None = None
-    charge_status: int | None = _requires_schema_code("charge_status")
+    charge_status: int | None = field(default=None, metadata={"dps": RoborockDataProtocol.CHARGE_STATUS})
     unsave_map_reason: int | None = None
     unsave_map_flag: int | None = None
     wash_status: int | None = None
     distance_off: int | None = None
     in_warmup: int | None = None
-    dry_status: int | None = _requires_schema_code("drying_status")
+    dry_status: int | None = field(default=None, metadata={"dps": RoborockDataProtocol.DRYING_STATUS})
     rdt: int | None = None
     clean_percent: int | None = None
     rss: int | None = None
@@ -294,11 +292,11 @@ class StatusV2(RoborockBase):
 
     msg_ver: int | None = None
     msg_seq: int | None = None
-    state: RoborockStateCode | None = _requires_schema_code("state")
-    battery: int | None = _requires_schema_code("battery")
+    state: RoborockStateCode | None = field(default=None, metadata={"dps": RoborockDataProtocol.STATE})
+    battery: int | None = field(default=None, metadata={"dps": RoborockDataProtocol.BATTERY})
     clean_time: int | None = None
     clean_area: int | None = None
-    error_code: RoborockErrorCode | None = _requires_schema_code("error_code")
+    error_code: RoborockErrorCode | None = field(default=None, metadata={"dps": RoborockDataProtocol.ERROR_CODE})
     map_present: int | None = None
     in_cleaning: RoborockInCleaning | None = None
     in_returning: int | None = None
@@ -308,12 +306,12 @@ class StatusV2(RoborockBase):
     back_type: int | None = None
     wash_phase: int | None = None
     wash_ready: int | None = None
-    fan_power: int | None = _requires_schema_code("fan_power")
+    fan_power: int | None = field(default=None, metadata={"dps": RoborockDataProtocol.FAN_POWER})
     dnd_enabled: int | None = None
     map_status: int | None = None
     is_locating: int | None = None
     lock_status: int | None = None
-    water_box_mode: int | None = _requires_schema_code("water_box_mode")
+    water_box_mode: int | None = field(default=None, metadata={"dps": RoborockDataProtocol.WATER_BOX_MODE})
     water_box_carriage_status: int | None = None
     mop_forbidden_enable: int | None = None
     camera_status: int | None = None
@@ -330,14 +328,14 @@ class StatusV2(RoborockBase):
     debug_mode: int | None = None
     collision_avoid_status: int | None = None
     switch_map_mode: int | None = None
-    dock_error_status: RoborockDockErrorCode | None = _requires_schema_code("dock_error_status")
-    charge_status: int | None = _requires_schema_code("charge_status")
+    dock_error_status: RoborockDockErrorCode | None = None
+    charge_status: int | None = field(default=None, metadata={"dps": RoborockDataProtocol.CHARGE_STATUS})
     unsave_map_reason: int | None = None
     unsave_map_flag: int | None = None
     wash_status: int | None = None
     distance_off: int | None = None
     in_warmup: int | None = None
-    dry_status: int | None = _requires_schema_code("drying_status")
+    dry_status: int | None = field(default=None, metadata={"dps": RoborockDataProtocol.DRYING_STATUS})
     rdt: int | None = None
     clean_percent: int | None = None
     rss: int | None = None
@@ -631,9 +629,8 @@ class ConsumableField(FieldNameBase):
     This is used with `roborock.devices.traits.v1.status.DeviceFeaturesTrait`
     to understand if a feature is supported by the device using `is_field_supported`.
 
-    The enum values are names of fields in the `Consumable` class. Each field is
-    annotated with `requires_schema_code` metadata to map the field to a schema
-    code in the product schema, which may have a different name than the field/attribute name.
+    The enum values are names of fields in the `Consumable` class. Each field is annotated
+    with a metadata value to determine if the field is supported by the device.
     """
 
     MAIN_BRUSH_WORK_TIME = "main_brush_work_time"
@@ -643,9 +640,9 @@ class ConsumableField(FieldNameBase):
 
 @dataclass
 class Consumable(RoborockBase):
-    main_brush_work_time: int | None = field(metadata={"requires_schema_code": "main_brush_life"}, default=None)
-    side_brush_work_time: int | None = field(metadata={"requires_schema_code": "side_brush_life"}, default=None)
-    filter_work_time: int | None = field(metadata={"requires_schema_code": "filter_life"}, default=None)
+    main_brush_work_time: int | None = field(default=None, metadata={"dps": RoborockDataProtocol.MAIN_BRUSH_WORK_TIME})
+    side_brush_work_time: int | None = field(default=None, metadata={"dps": RoborockDataProtocol.SIDE_BRUSH_WORK_TIME})
+    filter_work_time: int | None = field(default=None, metadata={"dps": RoborockDataProtocol.FILTER_WORK_TIME})
     filter_element_work_time: int | None = None
     sensor_dirty_time: int | None = None
     strainer_work_times: int | None = None

{python_roborock-5.4.1 → python_roborock-5.5.0}/roborock/devices/traits/v1/__init__.py

@@ -48,8 +48,8 @@ optional traits:
 Additionally, DeviceFeaturesTrait has a method `is_field_supported` that is used to
 check individual trait field values. This is a more fine grained version to allow
 optional fields in a dataclass, vs the above feature checks that apply to an entire
-trait. The `requires_schema_code` field metadata attribute is a string of the schema
-code in HomeDataProduct Schema that is required for the field to be supported.
+trait. The `dps` field metadata attribute references a schema code in
+HomeDataProduct Schema that is required for the field to be supported.
 """
 
 import logging

{python_roborock-5.4.1 → python_roborock-5.5.0}/roborock/devices/traits/v1/device_features.py

@@ -45,27 +45,26 @@ class DeviceFeaturesTrait(DeviceFeatures, common.V1TraitMixin):
         for field in fields(self):
             setattr(self, field.name, False)
 
+    @staticmethod
+    def _get_dataclass_field(cls: type[RoborockBase], field_name: FieldNameBase) -> Field:
+        """Look up a dataclass field by its FieldNameBase name."""
+        for f in fields(cls):
+            if f.name == field_name:
+                return f
+        raise ValueError(f"Field {field_name!r} not found in {cls}")
+
     def is_field_supported(self, cls: type[RoborockBase], field_name: FieldNameBase) -> bool:
         """Determines if the specified field is supported by this device.
 
-        We use dataclass attributes on the field to specify the schema code that is required
-        for the field to be supported and it is compared against the list of
-        supported schema codes for the device returned in the product information.
+        We use the `dps` dataclass field metadata to get the `RoborockDataProtocol`
+        integer ID and check it against the set of supported schema IDs for the
+        device returned in the product information.
         """
-        dataclass_field: Field | None = None
-        for field in fields(cls):
-            if field.name == field_name:
-                dataclass_field = field
-                break
-        if dataclass_field is None:
-            raise ValueError(f"Field {field_name} not found in {cls}")
-
-        requires_schema_code = dataclass_field.metadata.get("requires_schema_code", None)
-        if requires_schema_code is None:
-            # We assume the field is supported
+        dataclass_field = self._get_dataclass_field(cls, field_name)
+        if (dps := dataclass_field.metadata.get("dps")) is None:
+            # No DPS metadata — field is assumed always supported
             return True
-        # If the field requires a protocol that is not supported, we return False
-        return requires_schema_code in self._product.supported_schema_codes
+        return int(dps) in self._product.supported_schema_ids
 
     async def refresh(self) -> None:
         """Refresh the contents of this trait.

{python_roborock-5.4.1 → python_roborock-5.5.0}/roborock/devices/traits/v1/home.py

@@ -107,7 +107,10 @@ class HomeTrait(RoborockBase, common.V1TraitMixin):
 
         await self._maps_trait.refresh()
         if self._maps_trait.current_map_info is None:
-            raise RoborockException("Cannot perform home discovery without current map info")
+            _LOGGER.debug("Cannot perform home discovery without current map info")
+            self._discovery_completed = True
+            await self._update_home_cache({}, {})
+            return
 
         home_map_info, home_map_content = await self._build_home_map_info()
         _LOGGER.debug("Home discovery complete, caching data for %d maps", len(home_map_info))
@@ -198,7 +201,8 @@ class HomeTrait(RoborockBase, common.V1TraitMixin):
         if (current_map_info := self._maps_trait.current_map_info) is None or (
             map_flag := self._maps_trait.current_map
         ) is None:
-            raise RoborockException("Cannot refresh home data without current map info")
+            _LOGGER.debug("Cannot refresh home data without current map info")
+            return
 
         # Refresh the map content to ensure we have the latest image and object positions
         new_map_content = await self._refresh_map_content()

{python_roborock-5.4.1 → python_roborock-5.5.0}/roborock/roborock_message.py

@@ -2,7 +2,7 @@ from dataclasses import dataclass, field
 from enum import StrEnum
 from typing import Self
 
-from roborock import RoborockEnum
+from roborock.data.code_mappings import RoborockEnum
 from roborock.util import get_next_int, get_timestamp