Add device and product information to devices API (#540)

* feat: Expose device and product information on the new Device API

* chore: fix lint

Author: allenporter

roborock/devices/device.py

@@ -8,7 +8,7 @@
 from abc import ABC
 from collections.abc import Callable
 
-from roborock.containers import HomeDataDevice
+from roborock.containers import HomeDataDevice, HomeDataProduct
 from roborock.roborock_message import RoborockMessage
 
 from .channel import Channel
@@ -37,6 +37,7 @@ class RoborockDevice(ABC, TraitsMixin):
     def __init__(
         self,
         device_info: HomeDataDevice,
+        product: HomeDataProduct,
         channel: Channel,
         trait: Trait,
     ) -> None:
@@ -49,6 +50,8 @@ def __init__(
         TraitsMixin.__init__(self, trait)
         self._duid = device_info.duid
         self._name = device_info.name
+        self._device_info = device_info
+        self._product = product
         self._channel = channel
         self._unsub: Callable[[], None] | None = None
 
@@ -62,6 +65,23 @@ def name(self) -> str:
         """Return the device name."""
         return self._name
 
+    @property
+    def device_info(self) -> HomeDataDevice:
+        """Return the device information.
+
+        This includes information specific to the device like its identifier or
+        firmware version.
+        """
+        return self._device_info
+
+    @property
+    def product(self) -> HomeDataProduct:
+        """Return the device product name.
+
+        This returns product level information such as the model name.
+        """
+        return self._product
+
     @property
     def is_connected(self) -> bool:
         """Return whether the device is connected."""

roborock/devices/device_manager.py

@@ -168,7 +168,7 @@ def device_creator(home_data: HomeData, device: HomeDataDevice, product: HomeDat
                 trait = b01.create(channel)
             case _:
                 raise NotImplementedError(f"Device {device.name} has unsupported version {device.pv}")
-        return RoborockDevice(device, channel, trait)
+        return RoborockDevice(device, product, channel, trait)
 
     manager = DeviceManager(home_data_api, device_creator, mqtt_session=mqtt_session, cache=cache)
     await manager.discover_devices()

tests/devices/test_v1_device.py

@@ -53,6 +53,7 @@ def device_fixture(channel: AsyncMock, rpc_channel: AsyncMock, mqtt_rpc_channel:
     """Fixture to set up the device for tests."""
     return RoborockDevice(
         device_info=HOME_DATA.devices[0],
+        product=HOME_DATA.products[0],
         channel=channel,
         trait=v1.create(HOME_DATA.products[0], HOME_DATA, rpc_channel, mqtt_rpc_channel, AsyncMock(), NoCache()),
     )

tests/devices/traits/v1/fixtures.py

@@ -57,6 +57,7 @@ def device_fixture(
     """Fixture to set up the device for tests."""
     return RoborockDevice(
         device_info=HOME_DATA.devices[0],
+        product=HOME_DATA.products[0],
         channel=channel,
         trait=v1.create(
             HOME_DATA.products[0],