feat: Add from_any_optional method to CodeMapping for flexible enum resolution with corresponding tests.

Author: allenporter

roborock/data/code_mappings.py

@@ -100,6 +100,30 @@ def from_name(cls, name: str) -> Self:
                 return member
         raise ValueError(f"{name} is not a valid name for {cls.__name__}")
 
+    @classmethod
+    def from_any_optional(cls, value: str | int) -> Self | None:
+        """Resolve a string or int to an enum member.
+
+        Triest to lookup by enum name, string value, or integer code
+        and returns None if no match is found.
+        """
+        # Try enum name lookup (e.g. "SEEK")
+        try:
+            return cls.from_name(str(value))
+        except ValueError:
+            pass
+        # Try DP string value lookup (e.g. "dpSeek")
+        try:
+            return cls.from_value(str(value))
+        except ValueError:
+            pass
+        # Try integer code lookup (e.g. "11")
+        try:
+            return cls.from_code(int(value))
+        except (ValueError, TypeError):
+            pass
+        return None
+
     @classmethod
     def keys(cls) -> list[str]:
         """Returns a list of all member values."""

tests/data/test_code_mappings.py

@@ -52,6 +52,30 @@ def test_invalid_from_value() -> None:
         B01_Q10_DP.from_value("invalid_value")
 
 
+@pytest.mark.parametrize(
+    "input, expected",
+    [
+        ("START_CLEAN", B01_Q10_DP.START_CLEAN),
+        ("start_clean", B01_Q10_DP.START_CLEAN),
+        ("dpStartClean", B01_Q10_DP.START_CLEAN),
+        (201, B01_Q10_DP.START_CLEAN),
+        ("PAUSE", B01_Q10_DP.PAUSE),
+        ("pause", B01_Q10_DP.PAUSE),
+        ("dpPause", B01_Q10_DP.PAUSE),
+        (204, B01_Q10_DP.PAUSE),
+        ("STOP", B01_Q10_DP.STOP),
+        ("stop", B01_Q10_DP.STOP),
+        ("dpStop", B01_Q10_DP.STOP),
+        (206, B01_Q10_DP.STOP),
+        ("invalid_value", None),
+        (999999, None),
+    ],
+)
+def test_from_any_optional(input: str | int, expected: B01_Q10_DP | None) -> None:
+    """Test from_name method."""
+    assert B01_Q10_DP.from_any_optional(input) == expected
+
+
 def test_homedata_product_unknown_category():
     """Test that HomeDataProduct can handle unknown categories."""
     data = {