Fix last_replaced dates #2280

andrew-codechimp · Nov 2, 2024 · cac9e18 · cac9e18
1 parent d783652
commit cac9e18
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/custom_components/battery_notes/device.py b/custom_components/battery_notes/device.py
@@ -193,13 +193,13 @@ async def async_setup(self) -> bool:
             if entity.device_id:
                 device_entry = device_registry.async_get(entity.device_id)
 
-                if device_entry.created_at.year > 1970:
+                if device_entry and device_entry.created_at.year > 1970:
                     last_replaced = device_entry.created_at.strftime(
                         "%Y-%m-%dT%H:%M:%S:%f"
                     )
             else:
                 entity = entity_registry.async_get(source_entity_id)
-                if entity.created_at.year > 1970:
+                if entity and entity.created_at.year > 1970:
                     last_replaced = entity.created_at.strftime("%Y-%m-%dT%H:%M:%S:%f")
 
             _LOGGER.debug(