evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Use faster contains check in fan (#106431)

Browse source
This commit is contained in:
J. Nick Koston 2023-12-26 12:19:02 -10:00 • committed by GitHub
parent 244a7bef39
commit 51a50fc134
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
homeassistant/components/fan/__init__.py
View file

@ -400,10 +400,11 @@ class FanEntity(ToggleEntity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
def capability_attributes(self) -> dict[str, list[str] | None]: def capability_attributes(self) -> dict[str, list[str] | None]:
"""Return capability attributes.""" """Return capability attributes."""
attrs = {} attrs = {}
supported_features = self.supported_features
if ( if (
self.supported_features & FanEntityFeature.SET_SPEED FanEntityFeature.SET_SPEED in supported_features
or self.supported_features & FanEntityFeature.PRESET_MODE or FanEntityFeature.PRESET_MODE in supported_features
): ):
attrs[ATTR_PRESET_MODES] = self.preset_modes attrs[ATTR_PRESET_MODES] = self.preset_modes
@ -416,20 +417,19 @@ class FanEntity(ToggleEntity, cached_properties=CACHED_PROPERTIES_WITH_ATTR_):
data: dict[str, float | str | None] = {} data: dict[str, float | str | None] = {}
supported_features = self.supported_features supported_features = self.supported_features
if supported_features & FanEntityFeature.DIRECTION: if FanEntityFeature.DIRECTION in supported_features:
data[ATTR_DIRECTION] = self.current_direction data[ATTR_DIRECTION] = self.current_direction
if supported_features & FanEntityFeature.OSCILLATE: if FanEntityFeature.OSCILLATE in supported_features:
data[ATTR_OSCILLATING] = self.oscillating data[ATTR_OSCILLATING] = self.oscillating
if supported_features & FanEntityFeature.SET_SPEED: has_set_speed = FanEntityFeature.SET_SPEED in supported_features
if has_set_speed:
data[ATTR_PERCENTAGE] = self.percentage data[ATTR_PERCENTAGE] = self.percentage
data[ATTR_PERCENTAGE_STEP] = self.percentage_step data[ATTR_PERCENTAGE_STEP] = self.percentage_step
if ( if has_set_speed or FanEntityFeature.PRESET_MODE in supported_features:
supported_features & FanEntityFeature.PRESET_MODE
or supported_features & FanEntityFeature.SET_SPEED
):
data[ATTR_PRESET_MODE] = self.preset_mode data[ATTR_PRESET_MODE] = self.preset_mode
return data return data