From 68c50d412f7344c132bfbf87f94d47aab2b1a6e7 Mon Sep 17 00:00:00 2001
From: Luke <conway220@gmail.com>
Date: Mon, 10 Feb 2025 22:25:29 -0500
Subject: [PATCH] fix: add some extra data protocol checking

---
 roborock/roborock_message.py                  | 1 +
 roborock/version_1_apis/roborock_client_v1.py | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/roborock/roborock_message.py b/roborock/roborock_message.py
index 1386a4d5..1774c3bb 100644
--- a/roborock/roborock_message.py
+++ b/roborock/roborock_message.py
@@ -36,6 +36,7 @@ class RoborockDataProtocol(RoborockEnum):
     TASK_CANCEL_IN_MOTION = 132
     CHARGE_STATUS = 133
     DRYING_STATUS = 134
+    OFFLINE_STATUS = 135
 
     @classmethod
     def _missing_(cls: type[RoborockEnum], key) -> RoborockEnum:
diff --git a/roborock/version_1_apis/roborock_client_v1.py b/roborock/version_1_apis/roborock_client_v1.py
index a8dbb098..00e1052e 100644
--- a/roborock/version_1_apis/roborock_client_v1.py
+++ b/roborock/version_1_apis/roborock_client_v1.py
@@ -430,6 +430,12 @@ def on_message_received(self, messages: list[RoborockMessage]) -> None:
                                     consumable = Consumable.from_dict(value)
                                     for listener in self.listener_model.protocol_handlers.get(data_protocol, []):
                                         listener(consumable)
+                                elif data_protocol in {
+                                    RoborockDataProtocol.ADDITIONAL_PROPS,
+                                    RoborockDataProtocol.DRYING_STATUS,
+                                }:
+                                    # Known data protocol, but not yet sure how to correctly utilize it.
+                                    return
                                 else:
                                     self._logger.warning(
                                         f"Unknown data protocol {data_point_number}, please create an "