feat: single device api and discovery method

Author: humbertogontijo

mypy.ini

@@ -1,3 +1,5 @@
 [mypy]
 check_untyped_defs = True
-exclude = cli.py
+
+[mypy-construct]
+ignore_missing_imports = True

roborock/cli.py

@@ -83,7 +83,7 @@ async def _discover(ctx):
     home_data = await client.get_home_data(login_data.user_data)
     context.update(LoginData(**login_data.as_dict(), home_data=home_data))
     click.echo(
-        f"Discovered devices {', '.join([device.name for device in home_data.devices + home_data.received_devices])}"
+        f"Discovered devices {', '.join([device.name for device in home_data.get_all_devices()])}"
     )
 
 
@@ -123,8 +123,7 @@ async def command(ctx, cmd, device_id, params):
         await _discover(ctx)
         login_data = context.login_data()
     home_data = login_data.home_data
-    devices = home_data.devices + home_data.received_devices
-    device = next((device for device in devices if device.duid == device_id), None)
+    device = next((device for device in home_data.get_all_devices() if device.duid == device_id))
     device_info = RoborockDeviceInfo(device=device)
     mqtt_client = RoborockMqttClient(login_data.user_data, device_info)
     await mqtt_client.send_command(cmd, params)

roborock/containers.py

@@ -177,6 +177,14 @@ class HomeData(RoborockBase):
     received_devices: Optional[list[HomeDataDevice]] = None
     rooms: Optional[list[HomeDataRoom]] = None
 
+    def get_all_devices(self) -> list[HomeDataDevice]:
+        devices = []
+        if self.devices is not None:
+            devices += self.devices
+        if self.received_devices is not None:
+            devices += self.received_devices
+        return devices
+
 
 @dataclass
 class LoginData(RoborockBase):