python-roborock 7.3.0__tar.gz → 7.4.0__tar.gz

{python_roborock-7.3.0 → python_roborock-7.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: python-roborock
-Version: 7.3.0
+Version: 7.4.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-7.3.0 → python_roborock-7.4.0}/pyproject.toml

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

@@ -348,6 +348,24 @@ class HomeDataScene(RoborockBase):
     name: str
 
 
+@dataclass
+class FirmwareInfo(RoborockBase):
+    """Firmware/OTA info from the cloud (`ota/firmware/{duid}/updatev2`)."""
+
+    version: str | None = None
+    """Latest available firmware version."""
+    current_version: str | None = None
+    """Currently installed firmware version."""
+    updatable: bool | None = None
+    """Whether a newer firmware is available to install."""
+    desc: str | None = None
+    """Release notes / description."""
+    release_time: str | None = None
+    """Release date of the available firmware as an ISO-8601 date (``YYYY-MM-DD``)."""
+    force_update: bool | None = None
+    """Whether the update is mandatory (cannot be skipped)."""
+
+
 @dataclass
 class HomeDataSchedule(RoborockBase):
     id: int

{python_roborock-7.3.0 → python_roborock-7.4.0}/roborock/web_api.py

@@ -15,7 +15,7 @@ from aiohttp import ContentTypeError, FormData
 from pyrate_limiter import Duration, Limiter, Rate
 
 from roborock import HomeDataSchedule
-from roborock.data import HomeData, HomeDataRoom, HomeDataScene, ProductResponse, RRiot, UserData
+from roborock.data import FirmwareInfo, HomeData, HomeDataRoom, HomeDataScene, ProductResponse, RRiot, UserData
 from roborock.exceptions import (
     RoborockAccountDoesNotExist,
     RoborockException,
@@ -608,6 +608,73 @@ class RoborockApiClient:
         else:
             raise RoborockException("scene_response result was an unexpected type")
 
+    async def get_firmware_info(self, user_data: UserData, device_id: str) -> FirmwareInfo:
+        """Get firmware/OTA info for a device (latest version + updatable flag)."""
+        rriot = user_data.rriot
+        if rriot is None:
+            raise RoborockException("rriot is none")
+        if rriot.r.a is None:
+            raise RoborockException("Missing field 'a' in rriot reference")
+        path = f"/ota/firmware/{device_id}/updatev2"
+        params = {"lang": "en"}
+        firmware_request = PreparedRequest(
+            rriot.r.a,
+            self.session,
+            {
+                "Authorization": _get_hawk_authentication(rriot, path, params=params),
+            },
+        )
+        firmware_response = await firmware_request.request("get", path, params=params)
+        if not firmware_response or not firmware_response.get("success"):
+            raise RoborockException(firmware_response)
+        return FirmwareInfo.from_dict(firmware_response.get("result") or {})
+
+    async def start_firmware_update(self, user_data: UserData, device_id: str) -> None:
+        """Trigger the (irreversible) firmware update for a device.
+
+        The device downloads and flashes the latest firmware reported by
+        :meth:`get_firmware_info`.
+        """
+        rriot = user_data.rriot
+        if rriot is None:
+            raise RoborockException("rriot is none")
+        if rriot.r.a is None:
+            raise RoborockException("Missing field 'a' in rriot reference")
+        path = f"/ota/device/{device_id}/upgrade"
+        upgrade_request = PreparedRequest(
+            rriot.r.a,
+            self.session,
+            {
+                "Authorization": _get_hawk_authentication(rriot, path),
+            },
+        )
+        upgrade_response = await upgrade_request.request("post", path)
+        if not upgrade_response or not upgrade_response.get("success"):
+            raise RoborockException(upgrade_response)
+
+    async def set_silent_ota(self, user_data: UserData, device_id: str, enabled: bool) -> None:
+        """Enable/disable automatic (silent) firmware updates for a device.
+
+        The current value is reported as ``silent_ota_switch`` on the home-data device.
+        """
+        rriot = user_data.rriot
+        if rriot is None:
+            raise RoborockException("rriot is none")
+        if rriot.r.a is None:
+            raise RoborockException("Missing field 'a' in rriot reference")
+        path = f"/user/devices/{device_id}"
+        formdata = {"silentOtaSwitch": "true" if enabled else "false"}
+        silent_ota_request = PreparedRequest(
+            rriot.r.a,
+            self.session,
+            {
+                "Authorization": _get_hawk_authentication(rriot, path, formdata=formdata),
+            },
+        )
+        silent_ota_response = await silent_ota_request.request("put", path, data=formdata)
+        if not silent_ota_response or not silent_ota_response.get("success"):
+            raise RoborockException(silent_ota_response)
+
     async def execute_scene(self, user_data: UserData, scene_id: int) -> None:
         rriot = user_data.rriot
         if rriot is None: