Fix iCloud device battery level can be None (#31468)

This commit is contained in:
Quentame 2020-02-05 00:26:47 +01:00 committed by GitHub
parent 370e2ffa5a
commit 2c439af165
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -331,14 +331,13 @@ class IcloudDevice:
device_status = DEVICE_STATUS_CODES.get(self._status[DEVICE_STATUS], "error") device_status = DEVICE_STATUS_CODES.get(self._status[DEVICE_STATUS], "error")
self._attrs[ATTR_DEVICE_STATUS] = device_status self._attrs[ATTR_DEVICE_STATUS] = device_status
if self._status[DEVICE_BATTERY_STATUS] != "Unknown": self._battery_status = self._status[DEVICE_BATTERY_STATUS]
self._battery_level = int(self._status.get(DEVICE_BATTERY_LEVEL, 0) * 100) self._attrs[ATTR_BATTERY_STATUS] = self._battery_status
self._battery_status = self._status[DEVICE_BATTERY_STATUS] device_battery_level = self._status.get(DEVICE_BATTERY_LEVEL, 0)
low_power_mode = self._status[DEVICE_LOW_POWER_MODE] if self._battery_status != "Unknown" and device_battery_level is not None:
self._battery_level = int(device_battery_level * 100)
self._attrs[ATTR_BATTERY] = self._battery_level self._attrs[ATTR_BATTERY] = self._battery_level
self._attrs[ATTR_BATTERY_STATUS] = self._battery_status self._attrs[ATTR_LOW_POWER_MODE] = self._status[DEVICE_LOW_POWER_MODE]
self._attrs[ATTR_LOW_POWER_MODE] = low_power_mode
if ( if (
self._status[DEVICE_LOCATION] self._status[DEVICE_LOCATION]