Fix TypeError of vacuum battery level None (#62722)

Co-authored-by: Franck Nijhof <frenck@frenck.nl>
This commit is contained in:
htmltiger 2021-12-27 19:55:43 +00:00 committed by Franck Nijhof
parent c4fe1d57d9
commit fb26398ea7
No known key found for this signature in database
GPG Key ID: D62583BA8AB11CA3

View File

@ -641,6 +641,8 @@ class EnergyStorageTrait(_Trait):
def query_attributes(self): def query_attributes(self):
"""Return EnergyStorage query attributes.""" """Return EnergyStorage query attributes."""
battery_level = self.state.attributes.get(ATTR_BATTERY_LEVEL) battery_level = self.state.attributes.get(ATTR_BATTERY_LEVEL)
if battery_level is None:
return {}
if battery_level == 100: if battery_level == 100:
descriptive_capacity_remaining = "FULL" descriptive_capacity_remaining = "FULL"
elif 75 <= battery_level < 100: elif 75 <= battery_level < 100: