feat(ed): add new query01 msg for ed device

midealocal/devices/ed/__init__.py

@@ -8,7 +8,13 @@
 from midealocal.device import MideaDevice
 from midealocal.message import ListTypes
 
-from .message import MessageEDResponse, MessageNewSet, MessageOldSet, MessageQuery
+from .message import (
+    MessageEDResponse,
+    MessageNewSet,
+    MessageOldSet,
+    MessageQuery,
+    MessageQuery01,
+)
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -77,9 +83,12 @@ def _use_new_set(self) -> bool:
         # if (self.sub_type > 342 or self.sub_type == 340) else False
         return True
 
-    def build_query(self) -> list[MessageQuery]:
+    def build_query(self) -> list[MessageQuery | MessageQuery01]:
         """Midea ED device build query."""
-        return [MessageQuery(self._message_protocol_version, self._device_class)]
+        return [
+            MessageQuery(self._message_protocol_version),
+            MessageQuery01(self._message_protocol_version),
+        ]
 
     def process_message(self, msg: bytes) -> dict[str, Any]:
         """Midea ED device process message."""

midealocal/devices/ed/message.py

@@ -64,7 +64,11 @@ def _body(self) -> bytearray:
 class MessageQuery(MessageEDBase):
     """ED message query."""
 
-    def __init__(self, protocol_version: int, body_type: ListTypes) -> None:
+    def __init__(
+        self,
+        protocol_version: int,
+        body_type: ListTypes = ListTypes.X00,
+    ) -> None:
         """Initialize ED message query."""
         super().__init__(
             protocol_version=protocol_version,
@@ -77,6 +81,26 @@ def _body(self) -> bytearray:
         return bytearray([0x01])
 
 
+class MessageQuery01(MessageEDBase):
+    """ED message query01."""
+
+    def __init__(
+        self,
+        protocol_version: int,
+        body_type: ListTypes = ListTypes.X01,
+    ) -> None:
+        """Initialize ED message query01."""
+        super().__init__(
+            protocol_version=protocol_version,
+            message_type=MessageType.query,
+            body_type=body_type,
+        )
+
+    @property
+    def _body(self) -> bytearray:
+        return bytearray([0x01])
+
+
 class MessageNewSet(MessageEDBase):
     """ED message new set."""