deebot-client 3.0.3.dev0__tar.gz → 4.0.0__tar.gz

{deebot-client-3.0.3.dev0 → deebot-client-4.0.0}/.devcontainer.json

@@ -5,6 +5,7 @@
     // Configure properties specific to VS Code.
     "vscode": {
       "extensions": [
+        "charliermarsh.ruff",
         "eamodio.gitlens",
         "github.vscode-pull-request-github",
         "ms-python.python",
@@ -15,6 +16,9 @@
       ],
       // Set *default* container specific settings.json values on container create.
       "settings": {
+        "[python]": {
+          "editor.defaultFormatter": "charliermarsh.ruff"
+        },
         "editor.formatOnPaste": false,
         "editor.formatOnSave": true,
         "editor.formatOnType": true,

{deebot-client-3.0.3.dev0 → deebot-client-4.0.0}/.github/ISSUE_TEMPLATE/bug_report.yml

@@ -41,7 +41,7 @@ body:
     validations:
       required: true
     attributes:
-      label: On which deebot vacuum you have the issue?
+      label: On which deebot device (vacuum) you have the issue?
       placeholder: Deebot Ozmo 950
   - type: input
     id: version

{deebot-client-3.0.3.dev0 → deebot-client-4.0.0}/.github/workflows/ci.yml

@@ -18,7 +18,7 @@ jobs:
       - uses: "actions/checkout@v4"
       - name: Set up Python ${{ env.DEFAULT_PYTHON }}
         id: python
-        uses: actions/setup-python@v4.7.1
+        uses: actions/setup-python@v5.0.0
         with:
           python-version: ${{ env.DEFAULT_PYTHON }}
           cache: "pip"
@@ -42,7 +42,7 @@ jobs:
       - uses: "actions/checkout@v4"
       - name: Set up Python ${{ env.DEFAULT_PYTHON }}
         id: python
-        uses: actions/setup-python@v4.7.1
+        uses: actions/setup-python@v5.0.0
         with:
           python-version: ${{ env.DEFAULT_PYTHON }}
           cache: "pip"

{deebot-client-3.0.3.dev0 → deebot-client-4.0.0}/.github/workflows/python-publish.yml

@@ -23,7 +23,7 @@ jobs:
         uses: actions/checkout@v4
 
       - name: Set up Python
-        uses: actions/setup-python@v4.7.1
+        uses: actions/setup-python@v5.0.0
         with:
           python-version: "3.11"
 

{deebot-client-3.0.3.dev0 → deebot-client-4.0.0}/.pre-commit-config.yaml

@@ -1,4 +1,5 @@
 ci:
+  autofix_prs: false
   skip:
     # This steps run in the ci workflow. Keep in sync
     - mypy
@@ -10,23 +11,19 @@ default_language_version:
 
 repos:
   - repo: https://github.com/astral-sh/ruff-pre-commit
-    rev: v0.1.1
+    rev: v0.1.6
     hooks:
       - id: ruff
         args:
           - --fix
+          # - --unsafe-fixes
+      - id: ruff-format
   - repo: https://github.com/asottile/pyupgrade
     rev: v3.15.0
     hooks:
       - id: pyupgrade
         args:
           - --py311-plus
-  - repo: https://github.com/psf/black-pre-commit-mirror
-    rev: 23.10.0
-    hooks:
-      - id: black
-        args:
-          - --quiet
   - repo: https://github.com/codespell-project/codespell
     rev: v2.2.6
     hooks:
@@ -48,16 +45,16 @@ repos:
         args: [--branch, main, --branch, dev]
       - id: requirements-txt-fixer
   - repo: https://github.com/pre-commit/mirrors-prettier
-    rev: v3.0.3
+    rev: v3.1.0
     hooks:
       - id: prettier
         additional_dependencies:
-          - prettier@3.0.3
-          - prettier-plugin-sort-json@3.0.1
+          - prettier@3.1.0
+          - prettier-plugin-sort-json@3.1.0
         exclude_types:
           - python
   - repo: https://github.com/adrienverge/yamllint.git
-    rev: v1.32.0
+    rev: v1.33.0
     hooks:
       - id: yamllint
   - repo: local

{deebot-client-3.0.3.dev0/deebot_client.egg-info → deebot-client-4.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: deebot-client
-Version: 3.0.3.dev0
+Version: 4.0.0
 Summary: Deebot client library in python 3
 Author-email: Robert Resch <robert@resch.dev>
 License: GPL-3.0
@@ -24,7 +24,7 @@ Requires-Dist: defusedxml
 Requires-Dist: numpy<2.0,>=1.23.2
 Requires-Dist: Pillow<11.0,>=10.0.1
 
-# Client Library for Deebot Vacuums
+# Client Library for Deebot devices (Vacuums)
 
 [![PyPI - Downloads](https://img.shields.io/pypi/dw/deebot-client?style=for-the-badge)](https://pypi.org/project/deebot-client)
 <a href="https://www.buymeacoffee.com/edenhaus" target="_blank"><img src="https://cdn.buymeacoffee.com/buttons/default-black.png" width="150px" height="35px" alt="Buy Me A Coffee" style="height: 35px !important;width: 150px !important;" ></a>
@@ -57,7 +57,7 @@ from deebot_client.events import BatteryEvent
 from deebot_client.models import Configuration
 from deebot_client.mqtt_client import MqttClient, MqttConfiguration
 from deebot_client.util import md5
-from deebot_client.vacuum_bot import VacuumBot
+from deebot_client.device import Device
 
 device_id = md5(str(time.time()))
 account_id = "your email or phonenumber (cn)"
@@ -78,7 +78,7 @@ async def main():
 
     devices_ = await api_client.get_devices()
 
-    bot = VacuumBot(devices_[0], authenticator)
+    bot = Device(devices_[0], authenticator)
 
     mqtt_config = MqttConfiguration(config=config)
     mqtt = MqttClient(mqtt_config, authenticator)

{deebot-client-3.0.3.dev0 → deebot-client-4.0.0}/README.md

@@ -1,4 +1,4 @@
-# Client Library for Deebot Vacuums
+# Client Library for Deebot devices (Vacuums)
 
 [![PyPI - Downloads](https://img.shields.io/pypi/dw/deebot-client?style=for-the-badge)](https://pypi.org/project/deebot-client)
 <a href="https://www.buymeacoffee.com/edenhaus" target="_blank"><img src="https://cdn.buymeacoffee.com/buttons/default-black.png" width="150px" height="35px" alt="Buy Me A Coffee" style="height: 35px !important;width: 150px !important;" ></a>
@@ -31,7 +31,7 @@ from deebot_client.events import BatteryEvent
 from deebot_client.models import Configuration
 from deebot_client.mqtt_client import MqttClient, MqttConfiguration
 from deebot_client.util import md5
-from deebot_client.vacuum_bot import VacuumBot
+from deebot_client.device import Device
 
 device_id = md5(str(time.time()))
 account_id = "your email or phonenumber (cn)"
@@ -52,7 +52,7 @@ async def main():
 
     devices_ = await api_client.get_devices()
 
-    bot = VacuumBot(devices_[0], authenticator)
+    bot = Device(devices_[0], authenticator)
 
     mqtt_config = MqttConfiguration(config=config)
     mqtt = MqttClient(mqtt_config, authenticator)

{deebot-client-3.0.3.dev0 → deebot-client-4.0.0}/deebot_client/api_client.py

@@ -15,7 +15,7 @@ _LOGGER = get_logger(__name__)
 class ApiClient:
     """Api client."""
 
-    def __init__(self, authenticator: Authenticator):
+    def __init__(self, authenticator: Authenticator) -> None:
         self._authenticator = authenticator
 
     async def get_devices(self) -> list[DeviceInfo]:
@@ -38,9 +38,8 @@ class ApiClient:
                     _LOGGER.debug("Skipping device as it is not supported: %s", device)
             return devices
         _LOGGER.error("Failed to get devices: %s", resp)
-        raise ApiError(
-            f"failure {resp.get('error', '')} ({resp.get('errno', '')}) on getting devices"
-        )
+        msg = f"failure {resp.get('error', '')} ({resp.get('errno', '')}) on getting devices"
+        raise ApiError(msg)
 
     async def get_product_iot_map(self) -> dict[str, Any]:
         """Get product iot map."""
@@ -55,6 +54,5 @@ class ApiClient:
                 result[entry["classid"]] = entry["product"]
             return result
         _LOGGER.error("Failed to get product iot map")
-        raise ApiError(
-            f"failure {resp['error']} ({resp['errno']}) on getting product iot map"
-        )
+        msg = f"failure {resp['error']} ({resp['errno']}) on getting product iot map"
+        raise ApiError(msg)

{deebot-client-3.0.3.dev0 → deebot-client-4.0.0}/deebot_client/authentication.py

@@ -51,7 +51,7 @@ class _AuthClient:
         config: Configuration,
         account_id: str,
         password_hash: str,
-    ):
+    ) -> None:
         self._config = config
         self._account_id = account_id
         self._password_hash = password_hash
@@ -107,7 +107,7 @@ class _AuthClient:
             content_type = res.headers.get(hdrs.CONTENT_TYPE, "").lower()
             json = await res.json(content_type=content_type)
             _LOGGER.debug("got %s", json)
-            # todo better error handling # pylint: disable=fixme
+            # TODO better error handling # pylint: disable=fixme
             if json["code"] == "0000":
                 data: dict[str, Any] = json["data"]
                 return data
@@ -115,9 +115,8 @@ class _AuthClient:
                 raise InvalidAuthenticationError
 
             _LOGGER.error("call to %s failed with %s", url, json)
-            raise AuthenticationError(
-                f"failure code {json['code']} ({json['msg']}) for call {url}"
-            )
+            msg = f"failure code {json['code']} ({json['msg']}) for call {url}"
+            raise AuthenticationError(msg)
 
     async def __call_login_api(
         self, account_id: str, password_hash: str
@@ -204,9 +203,10 @@ class _AuthClient:
                 continue
 
             _LOGGER.error("call to %s failed with %s", _PATH_USERS_USER, resp)
-            raise AuthenticationError(
+            msg = (
                 f"failure {resp['error']} ({resp['errno']}) for call {_PATH_USERS_USER}"
             )
+            raise AuthenticationError(msg)
 
         raise AuthenticationError("failed to login with token")
 
@@ -302,7 +302,7 @@ class Authenticator:
         config: Configuration,
         account_id: str,
         password_hash: str,
-    ):
+    ) -> None:
         self._auth_client = _AuthClient(
             config,
             account_id,
@@ -317,7 +317,7 @@ class Authenticator:
         self._refresh_handle: asyncio.TimerHandle | None = None
         self._tasks: set[asyncio.Future[Any]] = set()
 
-    async def authenticate(self, force: bool = False) -> Credentials:
+    async def authenticate(self, *, force: bool = False) -> Credentials:
         """Authenticate on ecovacs servers."""
         async with self._lock:
             if (
@@ -379,7 +379,7 @@ class Authenticator:
 
             async def async_refresh() -> None:
                 try:
-                    await self.authenticate(True)
+                    await self.authenticate(force=True)
                 except Exception:  # pylint: disable=broad-except
                     _LOGGER.exception("An exception occurred during refreshing token")
 

{deebot-client-3.0.3.dev0 → deebot-client-4.0.0}/deebot_client/capabilities.py

@@ -4,8 +4,7 @@ from dataclasses import dataclass, field, fields, is_dataclass
 from types import MappingProxyType
 from typing import TYPE_CHECKING, Any, Generic, TypeVar
 
-from deebot_client.command import Command
-from deebot_client.commands.json.common import SetCommand
+from deebot_client.command import Command, SetCommand
 from deebot_client.events import (
     AdvancedModeEvent,
     AvailabilityEvent,
@@ -35,12 +34,14 @@ from deebot_client.events import (
     StatsEvent,
     TotalStatsEvent,
     TrueDetectEvent,
+    VoiceAssistantStateEvent,
     VolumeEvent,
     WaterAmount,
     WaterInfoEvent,
     WorkMode,
     WorkModeEvent,
 )
+from deebot_client.events.efficiency_mode import EfficiencyMode, EfficiencyModeEvent
 from deebot_client.models import CleanAction, CleanMode
 
 if TYPE_CHECKING:
@@ -121,7 +122,7 @@ class CapabilityClean:
     action: CapabilityCleanAction
     continuous: CapabilitySetEnable[ContinuousCleaningEvent]
     count: CapabilitySet[CleanCountEvent, int] | None = None
-    log: CapabilityEvent[CleanLogEvent]
+    log: CapabilityEvent[CleanLogEvent] | None = None
     preference: CapabilitySetEnable[CleanPreferenceEvent] | None = None
     work_mode: CapabilitySetTypes[WorkModeEvent, WorkMode] | None = None
 
@@ -169,7 +170,11 @@ class CapabilitySettings:
 
     advanced_mode: CapabilitySetEnable[AdvancedModeEvent]
     carpet_auto_fan_boost: CapabilitySetEnable[CarpetAutoFanBoostEvent]
+    efficiency_mode: (
+        CapabilitySetTypes[EfficiencyModeEvent, EfficiencyMode] | None
+    ) = None
     true_detect: CapabilitySetEnable[TrueDetectEvent] | None = None
+    voice_assistant: CapabilitySetEnable[VoiceAssistantStateEvent] | None = None
     volume: CapabilitySet[VolumeEvent, int]
 
 

{deebot-client-3.0.3.dev0 → deebot-client-4.0.0}/deebot_client/command.py

@@ -2,15 +2,17 @@
 from abc import ABC, abstractmethod
 import asyncio
 from dataclasses import dataclass, field
+from types import MappingProxyType
 from typing import Any, final
 
+from deebot_client.events import AvailabilityEvent
 from deebot_client.exceptions import DeebotError
 
 from .authentication import Authenticator
 from .const import PATH_API_IOT_DEVMANAGER, REQUEST_HEADERS, DataType
 from .event_bus import EventBus
 from .logging_filter import get_logger
-from .message import HandlingResult, HandlingState
+from .message import HandlingResult, HandlingState, MessageBody
 from .models import DeviceInfo
 
 _LOGGER = get_logger(__name__)
@@ -53,7 +55,7 @@ class Command(ABC):
     @classmethod
     @abstractmethod
     def data_type(cls) -> DataType:
-        """Data type."""  # noqa: D401
+        """Data type."""
 
     @abstractmethod
     def _get_payload(self) -> dict[str, Any] | list[Any] | str:
@@ -189,6 +191,49 @@ class Command(ABC):
         return hash(self.name) + hash(self._args)
 
 
+class CommandWithMessageHandling(Command, MessageBody, ABC):
+    """Command, which handle response by itself."""
+
+    _is_available_check: bool = False
+
+    def _handle_response(
+        self, event_bus: EventBus, response: dict[str, Any]
+    ) -> CommandResult:
+        """Handle response from a command.
+
+        :return: A message response
+        """
+        if response.get("ret") == "ok":
+            data = response.get("resp", response)
+            result = self.handle(event_bus, data)
+            return CommandResult(result.state, result.args)
+
+        if errno := response.get("errno", None):
+            match errno:
+                case 4200:
+                    # bot offline
+                    _LOGGER.info(
+                        'Device is offline. Could not execute command "%s"', self.name
+                    )
+                    event_bus.notify(AvailabilityEvent(available=False))
+                    return CommandResult(HandlingState.FAILED)
+                case 500:
+                    if self._is_available_check:
+                        _LOGGER.info(
+                            'No response received for command "%s" during availability-check.',
+                            self.name,
+                        )
+                    else:
+                        _LOGGER.warning(
+                            'No response received for command "%s". This can happen if the device has network issues or does not support the command',
+                            self.name,
+                        )
+                    return CommandResult(HandlingState.FAILED)
+
+        _LOGGER.warning('Command "%s" was not successfully.', self.name)
+        return CommandResult(HandlingState.ANALYSE)
+
+
 @dataclass
 class InitParam:
     """Init param."""
@@ -200,7 +245,7 @@ class InitParam:
 class CommandMqttP2P(Command, ABC):
     """Command which can handle mqtt p2p messages."""
 
-    _mqtt_params: dict[str, InitParam | None]
+    _mqtt_params: MappingProxyType[str, InitParam | None]
 
     @abstractmethod
     def handle_mqtt_p2p(self, event_bus: EventBus, response: dict[str, Any]) -> None:
@@ -230,10 +275,29 @@ def _pop_or_raise(name: str, type_: type, data: dict[str, Any]) -> Any:
     try:
         value = data.pop(name)
     except KeyError as err:
-        raise DeebotError(f'"{name}" is missing in {data}') from err
+        msg = f'"{name}" is missing in {data}'
+        raise DeebotError(msg) from err
     try:
         return type_(value)
     except ValueError as err:
-        raise DeebotError(
-            f'Could not convert "{value}" of {name} into {type_}'
-        ) from err
+        msg = f'Could not convert "{value}" of {name} into {type_}'
+        raise DeebotError(msg) from err
+
+
+class SetCommand(CommandWithMessageHandling, CommandMqttP2P, ABC):
+    """Base set command.
+
+    Command needs to be linked to the "get" command, for handling (updating) the sensors.
+    """
+
+    @property
+    @abstractmethod
+    def get_command(self) -> type[CommandWithMessageHandling]:
+        """Return the corresponding "get" command."""
+        raise NotImplementedError  # pragma: no cover
+
+    def handle_mqtt_p2p(self, event_bus: EventBus, response: dict[str, Any]) -> None:
+        """Handle response received over the mqtt channel "p2p"."""
+        result = self.handle(event_bus, response)
+        if result.state == HandlingState.SUCCESS and isinstance(self._args, dict):
+            self.get_command.handle(event_bus, self._args)

{deebot-client-3.0.3.dev0 → deebot-client-4.0.0}/deebot_client/commands/json/__init__.py

@@ -1,4 +1,4 @@
-"""Commands module."""
+"""Json commands module."""
 from deebot_client.command import Command, CommandMqttP2P
 
 from .advanced_mode import GetAdvancedMode, SetAdvancedMode
@@ -12,6 +12,7 @@ from .clean_logs import GetCleanLogs
 from .clean_preference import GetCleanPreference, SetCleanPreference
 from .common import JsonCommand
 from .continuous_cleaning import GetContinuousCleaning, SetContinuousCleaning
+from .efficiency import GetEfficiencyMode, SetEfficiencyMode
 from .error import GetError
 from .fan_speed import GetFanSpeed, SetFanSpeed
 from .life_span import GetLifeSpan, ResetLifeSpan
@@ -30,6 +31,7 @@ from .pos import GetPos
 from .relocation import SetRelocationState
 from .stats import GetStats, GetTotalStats
 from .true_detect import GetTrueDetect, SetTrueDetect
+from .voice_assistant_state import GetVoiceAssistantState, SetVoiceAssistantState
 from .volume import GetVolume, SetVolume
 from .water_info import GetWaterInfo, SetWaterInfo
 from .work_mode import GetWorkMode, SetWorkMode
@@ -52,6 +54,8 @@ __all__ = [
     "GetCleanLogs",
     "GetContinuousCleaning",
     "SetContinuousCleaning",
+    "GetEfficiencyMode",
+    "SetEfficiencyMode",
     "GetError",
     "GetFanSpeed",
     "SetFanSpeed",
@@ -73,6 +77,8 @@ __all__ = [
     "GetTotalStats",
     "GetTrueDetect",
     "SetTrueDetect",
+    "GetVoiceAssistantState",
+    "SetVoiceAssistantState",
     "GetVolume",
     "SetVolume",
     "GetWaterInfo",
@@ -111,6 +117,9 @@ _COMMANDS: list[type[JsonCommand]] = [
     GetContinuousCleaning,
     SetContinuousCleaning,
 
+    GetEfficiencyMode,
+    SetEfficiencyMode,
+
     GetError,
 
     GetFanSpeed,
@@ -143,6 +152,9 @@ _COMMANDS: list[type[JsonCommand]] = [
     GetTrueDetect,
     SetTrueDetect,
 
+    GetVoiceAssistantState,
+    SetVoiceAssistantState,
+
     GetVolume,
     SetVolume,
 
@@ -155,7 +167,8 @@ _COMMANDS: list[type[JsonCommand]] = [
 # fmt: on
 
 COMMANDS: dict[str, type[Command]] = {
-    cmd.name: cmd for cmd in _COMMANDS  # type: ignore[misc]
+    cmd.name: cmd  # type: ignore[misc]
+    for cmd in _COMMANDS
 }
 
 COMMANDS_WITH_MQTT_P2P_HANDLING: dict[str, type[CommandMqttP2P]] = {

{deebot-client-3.0.3.dev0 → deebot-client-4.0.0}/deebot_client/commands/json/battery.py

@@ -1,14 +1,14 @@
 """Battery commands."""
 from deebot_client.messages.json import OnBattery
 
-from .common import CommandWithMessageHandling
+from .common import JsonCommandWithMessageHandling
 
 
-class GetBattery(OnBattery, CommandWithMessageHandling):
+class GetBattery(OnBattery, JsonCommandWithMessageHandling):
     """Get battery command."""
 
     name = "getBattery"
 
-    def __init__(self, is_available_check: bool = False) -> None:
+    def __init__(self, *, is_available_check: bool = False) -> None:
         super().__init__()
         self._is_available_check = is_available_check

{deebot-client-3.0.3.dev0 → deebot-client-4.0.0}/deebot_client/commands/json/charge.py

@@ -5,7 +5,7 @@ from deebot_client.event_bus import EventBus
 from deebot_client.events import StateEvent
 from deebot_client.logging_filter import get_logger
 from deebot_client.message import HandlingResult
-from deebot_client.models import VacuumState
+from deebot_client.models import State
 
 from .common import ExecuteCommand
 from .const import CODE
@@ -29,12 +29,12 @@ class Charge(ExecuteCommand):
         """
         code = int(body.get(CODE, -1))
         if code == 0:
-            event_bus.notify(StateEvent(VacuumState.RETURNING))
+            event_bus.notify(StateEvent(State.RETURNING))
             return HandlingResult.success()
 
         if code == 30007:
             # bot is already charging
-            event_bus.notify(StateEvent(VacuumState.DOCKED))
+            event_bus.notify(StateEvent(State.DOCKED))
             return HandlingResult.success()
 
         return super()._handle_body(event_bus, body)

{deebot-client-3.0.3.dev0 → deebot-client-4.0.0}/deebot_client/commands/json/charge_state.py

@@ -4,13 +4,13 @@ from typing import Any
 from deebot_client.event_bus import EventBus
 from deebot_client.events import StateEvent
 from deebot_client.message import HandlingResult, MessageBodyDataDict
-from deebot_client.models import VacuumState
+from deebot_client.models import State
 
-from .common import CommandWithMessageHandling
+from .common import JsonCommandWithMessageHandling
 from .const import CODE
 
 
-class GetChargeState(CommandWithMessageHandling, MessageBodyDataDict):
+class GetChargeState(JsonCommandWithMessageHandling, MessageBodyDataDict):
     """Get charge state command."""
 
     name = "getChargeState"
@@ -24,7 +24,7 @@ class GetChargeState(CommandWithMessageHandling, MessageBodyDataDict):
         :return: A message response
         """
         if data.get("isCharging") == 1:
-            event_bus.notify(StateEvent(VacuumState.DOCKED))
+            event_bus.notify(StateEvent(State.DOCKED))
         return HandlingResult.success()
 
     @classmethod
@@ -33,17 +33,17 @@ class GetChargeState(CommandWithMessageHandling, MessageBodyDataDict):
             # Call this also if code is not in the body
             return super()._handle_body(event_bus, body)
 
-        status: VacuumState | None = None
+        status: State | None = None
         if body.get("msg", None) == "fail":
             if body["code"] == "30007":  # Already charging
-                status = VacuumState.DOCKED
+                status = State.DOCKED
             elif body["code"] in ("3", "5"):
                 # 3 -> Bot in stuck state, example dust bin out
                 # 5 -> Busy with another command
-                status = VacuumState.ERROR
+                status = State.ERROR
 
         if status:
-            event_bus.notify(StateEvent(VacuumState.DOCKED))
+            event_bus.notify(StateEvent(State.DOCKED))
             return HandlingResult.success()
 
         return HandlingResult.analyse()

{deebot-client-3.0.3.dev0 → deebot-client-4.0.0}/deebot_client/commands/json/clean.py

@@ -7,9 +7,9 @@ from deebot_client.event_bus import EventBus
 from deebot_client.events import StateEvent
 from deebot_client.logging_filter import get_logger
 from deebot_client.message import HandlingResult, MessageBodyDataDict
-from deebot_client.models import CleanAction, CleanMode, DeviceInfo, VacuumState
+from deebot_client.models import CleanAction, CleanMode, DeviceInfo, State
 
-from .common import CommandWithMessageHandling, ExecuteCommand
+from .common import ExecuteCommand, JsonCommandWithMessageHandling
 
 _LOGGER = get_logger(__name__)
 
@@ -30,12 +30,12 @@ class Clean(ExecuteCommand):
         if state and isinstance(self._args, dict):
             if (
                 self._args["act"] == CleanAction.RESUME.value
-                and state.state != VacuumState.PAUSED
+                and state.state != State.PAUSED
             ):
                 self._args = self.__get_args(CleanAction.START)
             elif (
                 self._args["act"] == CleanAction.START.value
-                and state.state == VacuumState.PAUSED
+                and state.state == State.PAUSED
             ):
                 self._args = self.__get_args(CleanAction.RESUME)
 
@@ -62,7 +62,7 @@ class CleanArea(Clean):
         self._args["count"] = cleanings
 
 
-class GetCleanInfo(CommandWithMessageHandling, MessageBodyDataDict):
+class GetCleanInfo(JsonCommandWithMessageHandling, MessageBodyDataDict):
     """Get clean info command."""
 
     name = "getCleanInfo"
@@ -75,20 +75,19 @@ class GetCleanInfo(CommandWithMessageHandling, MessageBodyDataDict):
 
         :return: A message response
         """
-
-        status: VacuumState | None = None
+        status: State | None = None
         state = data.get("state")
         if data.get("trigger") == "alert":
-            status = VacuumState.ERROR
+            status = State.ERROR
         elif state == "clean":
             clean_state = data.get("cleanState", {})
             motion_state = clean_state.get("motionState")
             if motion_state == "working":
-                status = VacuumState.CLEANING
+                status = State.CLEANING
             elif motion_state == "pause":
-                status = VacuumState.PAUSED
+                status = State.PAUSED
             elif motion_state == "goCharging":
-                status = VacuumState.RETURNING
+                status = State.RETURNING
 
             clean_type = clean_state.get("type")
             content = clean_state.get("content", {})
@@ -103,9 +102,9 @@ class GetCleanInfo(CommandWithMessageHandling, MessageBodyDataDict):
                 _LOGGER.debug("Last custom area values (x1,y1,x2,y2): %s", area_values)
 
         elif state == "goCharging":
-            status = VacuumState.RETURNING
+            status = State.RETURNING
         elif state == "idle":
-            status = VacuumState.IDLE
+            status = State.IDLE
 
         if status:
             event_bus.notify(StateEvent(status))