feat: Add from_any_optional method to RoborockModeEnum (#788)

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

* refactor: simplify B01_Q10 command parsing by removing a helper function and utilizing `from_any_optional`.

* chore: apply suggestions from code review

Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

---------

Co-authored-by: Luke Lashley <conway220@gmail.com>
Co-authored-by: Copilot Autofix powered by AI <175728472+Copilot@users.noreply.github.com>

Author: 3 people

roborock/cli.py

@@ -764,21 +764,6 @@ async def network_info(ctx, device_id: str):
     await _display_v1_trait(context, device_id, lambda v1: v1.network_info)
 
 
-def _parse_b01_q10_command(cmd: str) -> B01_Q10_DP:
-    """Parse B01_Q10 command from either enum name or value."""
-    try:
-        return B01_Q10_DP(int(cmd))
-    except ValueError:
-        try:
-            return B01_Q10_DP.from_name(cmd)
-        except ValueError:
-            try:
-                return B01_Q10_DP.from_value(cmd)
-            except ValueError:
-                pass
-    raise RoborockException(f"Invalid command {cmd} for B01_Q10 device")
-
-
 @click.command()
 @click.option("--device_id", required=True)
 @click.option("--cmd", required=True)
@@ -795,7 +780,8 @@ async def command(ctx, cmd, device_id, params):
         if result:
             click.echo(dump_json(result))
     elif device.b01_q10_properties is not None:
-        cmd_value = _parse_b01_q10_command(cmd)
+        if cmd_value := B01_Q10_DP.from_any_optional(cmd) is None:
+            raise RoborockException(f"Invalid command {cmd} for B01_Q10 device")
         command_trait: Trait = device.b01_q10_properties.command
         await command_trait.send(cmd_value, json.loads(params) if params is not None else None)
         click.echo("Command sent successfully; Enable debug logging (-d) to see responses.")

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.
+
+        Tries to look up 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_any_optional method."""
+    assert B01_Q10_DP.from_any_optional(input) == expected
+
+
 def test_homedata_product_unknown_category():
     """Test that HomeDataProduct can handle unknown categories."""
     data = {