fix: protocol check in B6 devices (#320)

midea-lan · Oct 6, 2024 · 55659c1 · 55659c1
1 parent 2f4e5d1
commit 55659c1
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/midealocal/devices/b6/message.py b/midealocal/devices/b6/message.py
@@ -15,6 +15,8 @@
 FAN_LEVEL_RANGE_3 = 170
 MIN_FAN_LEVEL_RANGE = 100
 
+MESSAGE_PROTOCOL_VERSION = 2
+
 
 class MessageB6Base(MessageRequest):
     """B6 message base."""
@@ -46,7 +48,9 @@ def __init__(self, protocol_version: int) -> None:
         super().__init__(
             protocol_version=protocol_version,
             message_type=MessageType.query,
-            body_type=BodyType.X11 if protocol_version == 2 else BodyType.X31,
+            body_type=BodyType.X11
+            if protocol_version == MESSAGE_PROTOCOL_VERSION
+            else BodyType.X31,
         )
 
     @property