Improve handling empty map

PiotrMachowski · May 23, 2022 · 2764b3f · 2764b3f
1 parent 56f8cb6
commit 2764b3f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/custom_components/xiaomi_cloud_map_extractor/camera.py b/custom_components/xiaomi_cloud_map_extractor/camera.py
@@ -338,13 +338,15 @@ def _handle_map_data(self, map_name: str):
             # noinspection PyBroadException
             try:
                 _LOGGER.debug("Map data retrieved")
-                self._set_map_data(map_data)
                 self._map_saved = map_stored
-                if self._map_data.image.is_empty:
+                if map_data.image.is_empty:
                     _LOGGER.debug("Map is empty")
                     self._status = CameraStatus.EMPTY_MAP
+                    if self._map_data is None or self._map_data.image.is_empty:
+                        self._set_map_data(map_data)
                 else:
                     _LOGGER.debug("Map is ok")
+                    self._set_map_data(map_data)
                     self._status = CameraStatus.OK
             except:
                 _LOGGER.warning("Unable to parse map data")