python-roborock 5.13.0__py3-none-any.whl → 5.14.1__py3-none-any.whl

{python_roborock-5.13.0.dist-info → python_roborock-5.14.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-roborock
-Version: 5.13.0
+Version: 5.14.1
 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.13.0.dist-info → python_roborock-5.14.1.dist-info}/RECORD

@@ -11,12 +11,12 @@ roborock/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 roborock/roborock_message.py,sha256=BPDT_uzCeD4L_4LottCMeft1kYmYPoG9ktcqqj0SYhQ,6466
 roborock/roborock_typing.py,sha256=Rwx0c0oZCMY-4O8VMLnFSO_OUNb_D59cyG_ktZTZoZI,17134
 roborock/util.py,sha256=mjw329ujLjoWPPAr9mG0VZstaVL06dmVvNnYACB-Pxk,1650
-roborock/web_api.py,sha256=sPdvwGnevCkgEiQwyCSmA3JrGIyZ3t7enIGK9dyJJ8U,34348
+roborock/web_api.py,sha256=g1LT04LIth4LhTrkQi8csXHZZ3UmhctQdHqkcwjWhyM,34504
 roborock/data/__init__.py,sha256=DikqaVM7VPVddkBNkHSss5zb1_iKB7GYmc2GiHoMAkA,257
 roborock/data/code_mappings.py,sha256=CzD9q-VuPy70pDZdjZ6kgK4vcGdzNGV3Eo5z9QGOYp0,8399
 roborock/data/containers.py,sha256=-b6H2snGQTzsJSsctgil6sx8rLtCDihTTzdkxcFRdlE,16489
 roborock/data/b01_q10/__init__.py,sha256=E4lwJcz4AI1z92FcYOJ-cY3fcfHSoXqvYVgfA-TgaRk,71
-roborock/data/b01_q10/b01_q10_code_mappings.py,sha256=Ja_ZAVJL7z-bmiAlnwDgkPNdTUB5Qt0kZA-1HQgqKz4,8161
+roborock/data/b01_q10/b01_q10_code_mappings.py,sha256=1DH85rOz9MDqcyLvLjTHSZe-k4we16_I3Wc2LkzJv4g,8288
 roborock/data/b01_q10/b01_q10_containers.py,sha256=wFHeZPGj6NvljKL99j1209ldwiVNAdqKh52UvMFU640,3514
 roborock/data/b01_q7/__init__.py,sha256=D-Cd8eB0njf1PQPM6Exl_9A0AhpzIjrL0hapGOCEkkA,69
 roborock/data/b01_q7/b01_q7_code_mappings.py,sha256=EHSCZSv_Os049NcenKo4tkPjDOB4vpmQ2p2Psr9FRCc,10698
@@ -47,8 +47,9 @@ roborock/devices/traits/common.py,sha256=KUlXvx2UcFVeNXEhPiCAdy2APu7zrJmM_jME_bM
 roborock/devices/traits/traits_mixin.py,sha256=7t15l_Ty1cN-3-fKNtRH1XR6-mfadPnTBCQRZP4rAtc,2092
 roborock/devices/traits/a01/__init__.py,sha256=i-us9zLPWPfSQ5XWq8t0MiYLYe5NkcLvk3aEQTYT7E0,8104
 roborock/devices/traits/b01/__init__.py,sha256=pGN6wEocE6Fr2BTasG8Z_75crD6OdTwpju59Vw7eTDk,202
-roborock/devices/traits/b01/q10/__init__.py,sha256=chMyP30e7o0WT-PU06t-JN3NrHGeZ1sKfCq6OA5x_EI,2641
+roborock/devices/traits/b01/q10/__init__.py,sha256=Nhb_jwxMRPUup9T6pCqt2kBFiYPrLBIsv7dpHJnF8qY,2799
 roborock/devices/traits/b01/q10/command.py,sha256=mcWNRSq72bfa79_EjKRgMEIDTV3vWFh2ofBe4aF7EOE,1306
+roborock/devices/traits/b01/q10/remote.py,sha256=ZvURdYP023vsJF6Mx9rA004bi6tOPAjrL4o7dC1baFQ,1257
 roborock/devices/traits/b01/q10/status.py,sha256=dIuUpRVjOVXvFr-EA_7lV9KCF7XXjE4vQOFsd-Bbh8w,1133
 roborock/devices/traits/b01/q10/vacuum.py,sha256=7PUPWaLqj3jsNLw87oyf1oz2UXckF4Z5Fh6ZYqeY7Ww,2280
 roborock/devices/traits/b01/q7/__init__.py,sha256=jxvN9KVrqQPgwNjIbseGME2v8DX-ryFH-rd_k0ulnps,6719
@@ -96,8 +97,8 @@ roborock/protocols/a01_protocol.py,sha256=JCIhUuVNamcENu0gtZR2x8rBz6dpAZLcbA-NQg
 roborock/protocols/b01_q10_protocol.py,sha256=SimfXpQotkEkb58ThZq7xa7fCxPxF-hrpAJcRKI-yZE,3407
 roborock/protocols/b01_q7_protocol.py,sha256=o8repRrxhxknzspsg1K27ds643rs87ilWXbj8RJtOCs,4619
 roborock/protocols/v1_protocol.py,sha256=uA-EMXd1AUBAXjkjAm0VxZlPLre0v2X9Kei5UqyPzaY,11141
-python_roborock-5.13.0.dist-info/METADATA,sha256=bYwkKEfZUz-u_qYMtzd_zGRsXncAlC9U_ncnKLdKzbg,4963
-python_roborock-5.13.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
-python_roborock-5.13.0.dist-info/entry_points.txt,sha256=EvC1nMqi9ZXKgZnqlNXA33v_3nzgPNjnM0mzWlrehnY,47
-python_roborock-5.13.0.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-python_roborock-5.13.0.dist-info/RECORD,,
+python_roborock-5.14.1.dist-info/METADATA,sha256=49viaRvdbijogTJG3PwecIXMf-sRHEX-NScxOdJeYEw,4963
+python_roborock-5.14.1.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+python_roborock-5.14.1.dist-info/entry_points.txt,sha256=EvC1nMqi9ZXKgZnqlNXA33v_3nzgPNjnM0mzWlrehnY,47
+python_roborock-5.14.1.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+python_roborock-5.14.1.dist-info/RECORD,,

roborock/data/b01_q10/b01_q10_code_mappings.py

@@ -1,3 +1,5 @@
+from enum import IntEnum
+
 from ..code_mappings import RoborockModeEnum
 
 
@@ -217,3 +219,11 @@ class YXDeviceDustCollectionFrequency(RoborockModeEnum):
     INTERVAL_30 = "interval_30", 30
     INTERVAL_45 = "interval_45", 45
     INTERVAL_60 = "interval_60", 60
+
+
+class RemoteCommand(IntEnum):
+    FORWARD = 0
+    LEFT = 2
+    RIGHT = 3
+    STOP = 4
+    EXIT = 5

roborock/devices/traits/b01/q10/__init__.py

@@ -2,7 +2,6 @@
 
 import asyncio
 import logging
-from typing import Any
 
 from roborock.data.b01_q10.b01_q10_code_mappings import B01_Q10_DP
 from roborock.devices.rpc.b01_q10_channel import stream_decoded_responses
@@ -10,6 +9,7 @@ from roborock.devices.traits import Trait
 from roborock.devices.transport.mqtt_channel import MqttChannel
 
 from .command import CommandTrait
+from .remote import RemoteTrait
 from .status import StatusTrait
 from .vacuum import VacuumTrait
 
@@ -32,11 +32,15 @@ class Q10PropertiesApi(Trait):
     vacuum: VacuumTrait
     """Trait for sending vacuum related commands to Q10 devices."""
 
+    remote: RemoteTrait
+    """Trait for sending remote control related commands to Q10 devices."""
+
     def __init__(self, channel: MqttChannel) -> None:
         """Initialize the B01Props API."""
         self._channel = channel
         self.command = CommandTrait(channel)
         self.vacuum = VacuumTrait(self.command)
+        self.remote = RemoteTrait(self.command)
         self.status = StatusTrait()
         self._subscribe_task: asyncio.Task[None] | None = None
 

roborock/devices/traits/b01/q10/remote.py

@@ -0,0 +1,43 @@
+"""Traits for Q10 B01 devices."""
+
+from roborock.data.b01_q10.b01_q10_code_mappings import (
+    B01_Q10_DP,
+    RemoteCommand,
+)
+
+from .command import CommandTrait
+
+
+class RemoteTrait:
+    """Trait for sending vacuum commands.
+
+    This is a wrapper around the CommandTrait for sending remote related
+    commands to Q10 devices.
+    """
+
+    def __init__(self, command: CommandTrait) -> None:
+        """Initialize the RemoteTrait."""
+        self._command = command
+
+    async def _send_remote(self, action: RemoteCommand) -> None:
+        await self._command.send(B01_Q10_DP.COMMON, params={B01_Q10_DP.REMOTE: action.value})
+
+    async def forward(self) -> None:
+        """Move forward."""
+        await self._send_remote(RemoteCommand.FORWARD)
+
+    async def left(self) -> None:
+        """Turn left."""
+        await self._send_remote(RemoteCommand.LEFT)
+
+    async def right(self) -> None:
+        """Turn right."""
+        await self._send_remote(RemoteCommand.RIGHT)
+
+    async def stop(self) -> None:
+        """Stop last moving command or start remote control."""
+        await self._send_remote(RemoteCommand.STOP)
+
+    async def exit_remote(self) -> None:
+        """Exit remote control."""
+        await self._send_remote(RemoteCommand.EXIT)

roborock/web_api.py

@@ -692,6 +692,8 @@ class PreparedRequest:
             _LOGGER.info("Resp raw: %s", resp_raw)
             # Still raise the err so that it's clear it failed.
             raise err
+        except (aiohttp.ClientError, TimeoutError, OSError) as err:
+            raise RoborockException(f"Network error contacting {_url}: {err}") from err
         finally:
             if close_session:
                 await session.close()