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

17
homeassistant/components/esphome/climate.py
View file

@ -13,7 +13,7 @@ from aioesphomeapi import (
ClimateSwingMode,
)
from homeassistant.components.climate import ClimateEntity
from homeassistant.components.climate import ClimateEntity, ClimateEntityFeature
from homeassistant.components.climate.const import (
ATTR_HVAC_MODE,
ATTR_TARGET_TEMP_HIGH,
@ -48,11 +48,6 @@ from homeassistant.components.climate.const import (
PRESET_HOME,
PRESET_NONE,
PRESET_SLEEP,
SUPPORT_FAN_MODE,
SUPPORT_PRESET_MODE,
SUPPORT_SWING_MODE,
SUPPORT_TARGET_TEMPERATURE,
SUPPORT_TARGET_TEMPERATURE_RANGE,
SWING_BOTH,
SWING_HORIZONTAL,
SWING_OFF,
@ -223,15 +218,15 @@ class EsphomeClimateEntity(EsphomeEntity[ClimateInfo, ClimateState], ClimateEnti
"""Return the list of supported features."""
features = 0
if self._static_info.supports_two_point_target_temperature:
features |= SUPPORT_TARGET_TEMPERATURE_RANGE
features |= ClimateEntityFeature.TARGET_TEMPERATURE_RANGE
else:
features |= SUPPORT_TARGET_TEMPERATURE
features |= ClimateEntityFeature.TARGET_TEMPERATURE
if self.preset_modes:
features |= SUPPORT_PRESET_MODE
features |= ClimateEntityFeature.PRESET_MODE
if self.fan_modes:
features |= SUPPORT_FAN_MODE
features |= ClimateEntityFeature.FAN_MODE
if self.swing_modes:
features |= SUPPORT_SWING_MODE
features |= ClimateEntityFeature.SWING_MODE
return features
@esphome_state_property