chore: refactor to seperate b01 ss and sc logic

Author: Lash-L

roborock/devices/device_manager.py

@@ -196,7 +196,12 @@ def device_creator(home_data: HomeData, device: HomeDataDevice, product: HomeDat
                 trait = a01.create(product, channel)
             case DeviceVersion.B01:
                 channel = create_mqtt_channel(user_data, mqtt_params, mqtt_session, device)
-                trait = b01.create(channel)
+                if "ss" in product.model.strip(".")[-1]:
+                    raise NotImplementedError(
+                        f"Device {device.name} has unsupported version B01_{product.model.strip('.')[-1]}"
+                    )
+                elif "sc" in product.model.strip(".")[-1]:
+                    trait = b01.sc.create(channel)
             case _:
                 raise NotImplementedError(f"Device {device.name} has unsupported version {device.pv}")
         return RoborockDevice(device, product, channel, trait)

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

@@ -0,0 +1,31 @@
+"""Traits for SC B01 devices.
+Right now, the only known Q7 devices are sc devices."""
+
+from roborock import RoborockB01Methods
+from roborock.devices.b01_channel import send_decoded_command
+from roborock.devices.mqtt_channel import MqttChannel
+from roborock.devices.traits import Trait
+from roborock.roborock_message import RoborockB01Props
+
+__all__ = [
+    "PropertiesApi",
+]
+
+
+class PropertiesApi(Trait):
+    """API for interacting with B01 devices."""
+
+    def __init__(self, channel: MqttChannel) -> None:
+        """Initialize the B01Props API."""
+        self._channel = channel
+
+    async def query_values(self, props: list[RoborockB01Props]) -> None:
+        """Query the device for the values of the given Dyad protocols."""
+        await send_decoded_command(
+            self._channel, dps=10000, command=RoborockB01Methods.GET_PROP, params={"property": props}
+        )
+
+
+def create(channel: MqttChannel) -> PropertiesApi:
+    """Create traits for B01 devices."""
+    return PropertiesApi(channel)

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

@@ -0,0 +1 @@
+"""Q10"""