Fix temperature sensor entity (#68)

iMicknl · Mar 30, 2022 · f584050 · f584050
1 parent f5c2535
commit f584050
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/custom_components/nest_protect/entity.py b/custom_components/nest_protect/entity.py
@@ -74,11 +74,15 @@ def generate_device_info(self) -> DeviceInfo | None:
             )
 
         if self.bucket.object_key.startswith("kryptonite."):
+            identifier = (
+                self.bucket.value.get("serial_number") or self.bucket.object_key
+            )
+
             return DeviceInfo(
-                identifiers={(DOMAIN, self.bucket.value["serial_number"])},
+                identifiers={(DOMAIN, identifier)},
                 name=self._attr_name,
                 manufacturer="Google",
-                model=self.bucket.value["model"],
+                model=self.bucket.value.get("model"),
                 suggested_area=self.area,
             )