From d611b169acff9b1908fec2e15253b10ac8cade88 Mon Sep 17 00:00:00 2001 From: Maciej Bieniek Date: Sat, 5 Aug 2023 11:05:15 +0000 Subject: [PATCH] Don't assume that `battery_level` value is always present in Tractive `hw_info` (#97766) Don't assume that battery_level value is always present in hw_info --- homeassistant/components/tractive/device_tracker.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/homeassistant/components/tractive/device_tracker.py b/homeassistant/components/tractive/device_tracker.py index e9739819734..a97ea963362 100644 --- a/homeassistant/components/tractive/device_tracker.py +++ b/homeassistant/components/tractive/device_tracker.py @@ -43,7 +43,7 @@ class TractiveDeviceTracker(TractiveEntity, TrackerEntity): """Initialize tracker entity.""" super().__init__(user_id, item.trackable, item.tracker_details) - self._battery_level: int = item.hw_info["battery_level"] + self._battery_level: int | None = item.hw_info.get("battery_level") self._latitude: float = item.pos_report["latlong"][0] self._longitude: float = item.pos_report["latlong"][1] self._accuracy: int = item.pos_report["pos_uncertainty"] @@ -75,7 +75,7 @@ class TractiveDeviceTracker(TractiveEntity, TrackerEntity): return self._accuracy @property - def battery_level(self) -> int: + def battery_level(self) -> int | None: """Return the battery level of the device.""" return self._battery_level