fix: clean up some naming

Author: Lash-L

roborock/devices/device_manager.py

@@ -201,7 +201,8 @@ def device_creator(home_data: HomeData, device: HomeDataDevice, product: HomeDat
                         f"Device {device.name} has unsupported version B01_{product.model.strip('.')[-1]}"
                     )
                 elif "sc" in product.model.strip(".")[-1]:
-                    trait = b01.sc.create(channel)
+                    # Q7 devices start with 'sc' in their model naming.
+                    trait = b01.q7.create(channel)
             case _:
                 raise NotImplementedError(f"Device {device.name} has unsupported version {device.pv}")
         return RoborockDevice(device, product, channel, trait)

roborock/devices/traits/b01/__init__.py

@@ -1,5 +1,5 @@
 """Traits for B01 devices."""
 
-from .sc import PropertiesApi
+from .q7 import Q7PropertiesApi
 
-__all__ = ["PropertiesApi", "sc", "ss"]
+__all__ = ["q&PropertiesApi", "q7", "q10"]

…borock/devices/traits/b01/ss/__init__.py …orock/devices/traits/b01/q10/__init__.pyroborock/devices/traits/b01/ss/__init__.py renamed to roborock/devices/traits/b01/q10/__init__.py roborock/devices/traits/b01/ss/__init__.py renamed to roborock/devices/traits/b01/q10/__init__.py

@@ -0,0 +1,31 @@
+"""Traits for Q7 B01 devices.
+Potentially other devices may fall into this category in the future."""
+
+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
+from roborock.roborock_typing import RoborockB01Q7Methods
+
+__all__ = [
+    "Q7PropertiesApi",
+]
+
+
+class Q7PropertiesApi(Trait):
+    """API for interacting with Q7 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 Q7 protocols."""
+        await send_decoded_command(
+            self._channel, dps=10000, command=RoborockB01Q7Methods.GET_PROP, params={"property": props}
+        )
+
+
+def create(channel: MqttChannel) -> Q7PropertiesApi:
+    """Create traits for B01 devices."""
+    return Q7PropertiesApi(channel)

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

@@ -31,8 +31,8 @@ class TraitsMixin:
     zeo: a01.ZeoApi | None = None
     """Zeo API, if supported."""
 
-    b01_properties: b01.PropertiesApi | None = None
-    """B01 properties trait, if supported."""
+    b01_q7_properties: b01.Q7PropertiesApi | None = None
+    """B01 Q7 properties trait, if supported."""
 
     def __init__(self, trait: Trait) -> None:
         """Initialize the TraitsMixin with the given trait.

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

@@ -7,7 +7,7 @@
 from Crypto.Cipher import AES
 from Crypto.Util.Padding import pad, unpad
 
-from roborock import RoborockB01Methods
+from roborock import RoborockB01Q7Methods
 from roborock.exceptions import RoborockException
 from roborock.roborock_message import (
     RoborockMessage,
@@ -18,7 +18,7 @@
 _LOGGER = logging.getLogger(__name__)
 
 B01_VERSION = b"B01"
-CommandType = RoborockB01Methods | str
+CommandType = RoborockB01Q7Methods | str
 ParamsType = list | dict | int | None
 
 

roborock/devices/traits/traits_mixin.py

@@ -271,8 +271,8 @@ class RoborockCommand(str, Enum):
     APP_GET_ROBOT_SETTING = "app_get_robot_setting"
 
 
-class RoborockB01Methods(StrEnum):
-    """Methods used by the Roborock B01 model."""
+class RoborockB01Q7Methods(StrEnum):
+    """Methods used by the Roborock Q7 model."""
 
     GET_PROP = "prop.get"
     GET_MAP_LIST = "service.get_map_list"