Merge pull request #212 from mikey0000/fix-luba-data

remove device from saved data if it ends up there
mikey0000 · Nov 28, 2024 · 614368e · 614368e
2 parents e4224b6 + 42d42e7
commit 614368e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/custom_components/mammotion/coordinator.py b/custom_components/mammotion/coordinator.py
@@ -339,8 +339,10 @@ async def async_restore_data(self) -> None:
         restored_data = await store.async_load()
         try:
             if restored_data:
-                device_dict = LubaMsg().to_dict(casing=betterproto.Casing.SNAKE)
+                if restored_data.get("device"):
+                    del restored_data["device"]
                 mower_state = MowingDevice().from_dict(restored_data)
+                device_dict = LubaMsg().to_dict(casing=betterproto.Casing.SNAKE)
                 mower_state.update_raw(device_dict)
                 self.manager.get_device_by_name(
                     self.device_name
@@ -354,7 +356,7 @@ async def async_save_data(self, data: MowingDevice) -> None:
         """Get map data from the device."""
         store = Store(self.hass, version=1, key=self.device_name)
         stored_data = asdict(data)
-        stored_data["device"] = None
+        del stored_data["device"]
         await store.async_save(stored_data)