From 893cea66b57fc58afc1bc9d3066c9ecc2e719617 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 6 Apr 2022 11:59:33 +0000 Subject: [PATCH] Use EntityFeature in mysensors --- homeassistant/components/mysensors/climate.py | 11 ++++------- 1 file changed, 4 insertions(+), 7 deletions(-) diff --git a/homeassistant/components/mysensors/climate.py b/homeassistant/components/mysensors/climate.py index 0157d06ca2e7..960360cd5d95 100644 --- a/homeassistant/components/mysensors/climate.py +++ b/homeassistant/components/mysensors/climate.py @@ -4,7 +4,7 @@ from __future__ import annotations from typing import Any from homeassistant.components import mysensors -from homeassistant.components.climate import ClimateEntity +from homeassistant.components.climate import ClimateEntity, ClimateEntityFeature from homeassistant.components.climate.const import ( ATTR_TARGET_TEMP_HIGH, ATTR_TARGET_TEMP_LOW, @@ -12,9 +12,6 @@ from homeassistant.components.climate.const import ( HVAC_MODE_COOL, HVAC_MODE_HEAT, HVAC_MODE_OFF, - SUPPORT_FAN_MODE, - SUPPORT_TARGET_TEMPERATURE, - SUPPORT_TARGET_TEMPERATURE_RANGE, ) from homeassistant.config_entries import ConfigEntry from homeassistant.const import ( @@ -84,14 +81,14 @@ class MySensorsHVAC(mysensors.device.MySensorsEntity, ClimateEntity): features = 0 set_req = self.gateway.const.SetReq if set_req.V_HVAC_SPEED in self._values: - features = features | SUPPORT_FAN_MODE + features = features | ClimateEntityFeature.FAN_MODE if ( set_req.V_HVAC_SETPOINT_COOL in self._values and set_req.V_HVAC_SETPOINT_HEAT in self._values ): - features = features | SUPPORT_TARGET_TEMPERATURE_RANGE + features = features | ClimateEntityFeature.TARGET_TEMPERATURE_RANGE else: - features = features | SUPPORT_TARGET_TEMPERATURE + features = features | ClimateEntityFeature.TARGET_TEMPERATURE return features @property