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

Use EntityFeature enum in esphome (#69386)

Browse source
This commit is contained in:
epenet 2022-04-06 11:52:59 +02:00 • committed by GitHub
parent 78045df227
commit dae2cf2827
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
5 changed files with 26 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

10
homeassistant/components/esphome/fan.py
View file

@ -9,10 +9,8 @@ from aioesphomeapi import FanDirection, FanInfo, FanSpeed, FanState
from homeassistant.components.fan import (
DIRECTION_FORWARD,
DIRECTION_REVERSE,
SUPPORT_DIRECTION,
SUPPORT_OSCILLATE,
SUPPORT_SET_SPEED,
FanEntity,
FanEntityFeature,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant
@ -161,9 +159,9 @@ class EsphomeFan(EsphomeEntity[FanInfo, FanState], FanEntity):
"""Flag supported features."""
flags = 0
if self._static_info.supports_oscillation:
flags |= SUPPORT_OSCILLATE
flags |= FanEntityFeature.OSCILLATE
if self._static_info.supports_speed:
flags |= SUPPORT_SET_SPEED
flags |= FanEntityFeature.SET_SPEED
if self._static_info.supports_direction:
flags |= SUPPORT_DIRECTION
flags |= FanEntityFeature.DIRECTION
return flags