Add image changes detection

Author: PiotrMachowski

custom_components/xiaomi_cloud_map_extractor/connector/__init__.py

@@ -1,3 +1,4 @@
+import hashlib
 import logging
 from collections.abc import Callable
 from datetime import datetime
@@ -60,6 +61,7 @@ class XiaomiCloudMapExtractorConnector:
     _connector_config: XiaomiCloudConnectorConfig | None
     _forced_refresh: bool
     _auto_update: bool
+    _last_hash: str | None
 
     def __init__(
         self: Self,
@@ -78,6 +80,7 @@ def __init__(
         self._used_api = self._config.used_api
         self._forced_refresh = False
         self._auto_update = True
+        self._last_hash = None
 
     async def get_data(self: Self) -> XiaomiCloudMapExtractorData:
         if self._should_get_map():
@@ -155,6 +158,10 @@ async def _get_map(self: Self) -> None:
             self._map_cache.status = XiaomiCloudMapExtractorConnectorStatus.OK
             self._map_cache.last_successful_update_timestamp = datetime.now()
             self._map_cache.two_factor_url = None
+            if self._last_hash != (new_hash := hashlib.sha256(self._map_cache.map_data_raw).hexdigest()):
+                self._last_hash = new_hash
+                self._map_cache.last_real_update_timestamp = self._map_cache.last_successful_update_timestamp
+
         except DeviceNotFoundException as e:
             self._map_cache.status = XiaomiCloudMapExtractorConnectorStatus.DEVICE_NOT_FOUND
             raise e

custom_components/xiaomi_cloud_map_extractor/connector/model.py

@@ -56,21 +56,17 @@ class XiaomiCloudMapExtractorData:
     two_factor_url: str | None = None
     last_update_timestamp: datetime | None = None
     last_successful_update_timestamp: datetime | None = None
+    last_real_update_timestamp: datetime | None = None
     status: XiaomiCloudMapExtractorConnectorStatus = XiaomiCloudMapExtractorConnectorStatus.UNKNOWN
     additional_vacuum_data: dict[str, Any] | None = None
 
     def as_dict(self: Self) -> dict[str, Any]:
-        map_image_dict = self.map_data and self.map_data.image and {
-            **self.map_data.image.as_dict(),
-            "image_bytes": base64.b64encode(self.map_data.image.data.tobytes()).decode()
-        }
-
         map_data_dict = self.map_data and {
             "blocks": self.map_data.blocks,
             "charger": self.map_data.charger and self.map_data.charger.as_dict(),
             "goto": self.map_data.goto and self.map_data.goto.as_dict(),
             "goto_path": self.map_data.goto_path and self.map_data.goto_path.as_dict(),
-            "image": map_image_dict,
+            "image": self.map_data and self.map_data.image and self.map_data.image.as_dict(),
             "no_go_areas": as_list_of_dict(self.map_data.no_go_areas),
             "no_mopping_areas": as_list_of_dict(self.map_data.no_mopping_areas),
             "no_carpet_areas": as_list_of_dict(self.map_data.no_carpet_areas),
@@ -102,5 +98,7 @@ def as_dict(self: Self) -> dict[str, Any]:
                 "%Y-%m-%d %H:%M:%S"),
             "last_successful_update_timestamp": self.last_successful_update_timestamp and self.last_successful_update_timestamp.strftime(
                 "%Y-%m-%d %H:%M:%S"),
+            "last_real_update_timestamp": self.last_real_update_timestamp and self.last_real_update_timestamp.strftime(
+                "%Y-%m-%d %H:%M:%S"),
             "status": self.status,
         }

custom_components/xiaomi_cloud_map_extractor/image.py

@@ -1,3 +1,4 @@
+from datetime import datetime
 import logging
 from typing import Self, Any
 
@@ -45,11 +46,11 @@ def image(self: Self) -> bytes | None:
         return data.map_image
 
     @property
-    def image_last_updated(self: Self):
+    def image_last_updated(self: Self) -> datetime | None:
         data = self._data()
         if data is None:
             return None
-        return data.last_successful_update_timestamp
+        return data.last_real_update_timestamp
 
     @property
     def extra_state_attributes(self: Self) -> dict[str, Any]: