python-roborock 2.1.0__tar.gz → 2.2.0__tar.gz

{python_roborock-2.1.0 → python_roborock-2.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-roborock
-Version: 2.1.0
+Version: 2.2.0
 Summary: A package to control Roborock vacuums.
 Home-page: https://github.com/humbertogontijo/python-roborock
 License: GPL-3.0-only

{python_roborock-2.1.0 → python_roborock-2.2.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "python-roborock"
-version = "2.1.0"
+version = "2.2.0"
 description = "A package to control Roborock vacuums."
 authors = ["humbertogontijo <humbertogontijo@users.noreply.github.com>"]
 license = "GPL-3.0-only"

{python_roborock-2.1.0 → python_roborock-2.2.0}/roborock/api.py

@@ -49,10 +49,10 @@ class RoborockClient:
     def __del__(self) -> None:
         self.release()
 
-    def release(self):
+    def release(self) -> None:
         self.sync_disconnect()
 
-    async def async_release(self):
+    async def async_release(self) -> None:
         await self.async_disconnect()
 
     @property

{python_roborock-2.1.0 → python_roborock-2.2.0}/roborock/version_1_apis/roborock_client_v1.py

@@ -277,6 +277,8 @@ class RoborockClientV1(RoborockClient):
         """Gets the mapping from segment id -> iot id. Only works on local api."""
         mapping: list = await self.send_command(RoborockCommand.GET_ROOM_MAPPING)
         if isinstance(mapping, list):
+            if not isinstance(mapping[0], list) and len(mapping) == 2:
+                return [RoomMapping(segment_id=mapping[0], iot_id=mapping[1])]
             return [
                 RoomMapping(segment_id=segment_id, iot_id=iot_id)  # type: ignore
                 for segment_id, iot_id in [unpack_list(room, 2) for room in mapping if isinstance(room, list)]

{python_roborock-2.1.0 → python_roborock-2.2.0}/roborock/version_1_apis/roborock_mqtt_client_v1.py

@@ -71,5 +71,5 @@ class RoborockMqttClientV1(RoborockMqttClient, RoborockClientV1):
         roborock_message = RoborockMessage(timestamp=timestamp, protocol=request_protocol, payload=payload)
         return await self.send_message(roborock_message)
 
-    async def get_map_v1(self):
+    async def get_map_v1(self) -> bytes | None:
         return await self.send_command(RoborockCommand.GET_MAP_V1)

{python_roborock-2.1.0 → python_roborock-2.2.0}/roborock/version_a01_apis/roborock_client_a01.py

@@ -137,6 +137,8 @@ class RoborockClientA01(RoborockClient):
                         if queue and queue.protocol == protocol:
                             queue.resolve((converted_response, None))
 
-    async def update_values(self, dyad_data_protocols: list[RoborockDyadDataProtocol | RoborockZeoProtocol]):
+    async def update_values(
+        self, dyad_data_protocols: list[RoborockDyadDataProtocol | RoborockZeoProtocol]
+    ) -> dict[RoborockDyadDataProtocol | RoborockZeoProtocol, typing.Any]:
         """This should handle updating for each given protocol."""
         raise NotImplementedError

{python_roborock-2.1.0 → python_roborock-2.2.0}/roborock/version_a01_apis/roborock_mqtt_client_a01.py

@@ -57,7 +57,9 @@ class RoborockMqttClientA01(RoborockMqttClient, RoborockClientA01):
                     dps_responses[dps] = response[0]
         return dps_responses
 
-    async def update_values(self, dyad_data_protocols: list[RoborockDyadDataProtocol | RoborockZeoProtocol]):
+    async def update_values(
+        self, dyad_data_protocols: list[RoborockDyadDataProtocol | RoborockZeoProtocol]
+    ) -> dict[RoborockDyadDataProtocol | RoborockZeoProtocol, typing.Any]:
         payload = {"dps": {RoborockDyadDataProtocol.ID_QUERY: str([int(protocol) for protocol in dyad_data_protocols])}}
         return await self.send_message(
             RoborockMessage(