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]:
"""Return capability attributes."""
attrs = {}
supported_features = self.supported_features
if (
self.supported_features & FanEntityFeature.SET_SPEED
or self.supported_features & FanEntityFeature.PRESET_MODE
FanEntityFeature.SET_SPEED in supported_features
or FanEntityFeature.PRESET_MODE in supported_features
):
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] = {}
supported_features = self.supported_features
if supported_features & FanEntityFeature.DIRECTION:
if FanEntityFeature.DIRECTION in supported_features:
data[ATTR_DIRECTION] = self.current_direction
if supported_features & FanEntityFeature.OSCILLATE:
if FanEntityFeature.OSCILLATE in supported_features:
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_STEP] = self.percentage_step
if (
supported_features & FanEntityFeature.PRESET_MODE
or supported_features & FanEntityFeature.SET_SPEED
):
if has_set_speed or FanEntityFeature.PRESET_MODE in supported_features:
data[ATTR_PRESET_MODE] = self.preset_mode
return data