Fix capability_attributes when supported_features is None (#30993)

* Fix capability_attributes when supported_features is None (water_heater)

* Fix capability_attributes when supported_features is None (media_player)
This commit is contained in:
Jeff Irion 2020-01-19 23:06:35 -08:00 committed by Paulus Schoutsen
parent b2212ad445
commit bb37f7bb75
2 changed files with 2 additions and 2 deletions

View File

@ -785,7 +785,7 @@ class MediaPlayerDevice(Entity):
@property
def capability_attributes(self):
"""Return capabilitiy attributes."""
supported_features = self.supported_features
supported_features = self.supported_features or 0
data = {}
if supported_features & SUPPORT_SELECT_SOURCE:

View File

@ -146,7 +146,7 @@ class WaterHeaterDevice(Entity):
@property
def capability_attributes(self):
"""Return capabilitiy attributes."""
supported_features = self.supported_features
supported_features = self.supported_features or 0
data = {
ATTR_MIN_TEMP: show_temp(