From 7c0b71783bbe583d8e57edc7e9aeeaae7fbb76ab Mon Sep 17 00:00:00 2001 From: fustom Date: Wed, 13 Sep 2023 15:44:14 +0200 Subject: [PATCH] Add device attribute to features available check --- ariston/device.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ariston/device.py b/ariston/device.py index 9ceb60f..5b73337 100644 --- a/ariston/device.py +++ b/ariston/device.py @@ -364,7 +364,7 @@ def _set_energy_features(self): def are_device_features_available( self, - device_features: Optional[list[DeviceFeatures]], + device_features: Optional[list[DeviceFeatures or CustomDeviceFeatures or DeviceAttribute]], system_types: Optional[list[SystemType]], whe_types: Optional[list[WheType]], ) -> bool: @@ -380,6 +380,7 @@ def are_device_features_available( if ( self.features.get(str(device_feature)) is not True and self.custom_features.get(str(device_feature)) is not True + and self.attributes.get(str(device_feature)) is not True ): return False