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

Use climate enums in ephember (#70637)

Browse source
This commit is contained in:
epenet 2022-04-25 13:21:12 +02:00 • committed by GitHub
parent a5f5cfb156
commit 2efc25c7ca
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

28
homeassistant/components/ephember/climate.py
View file

@ -17,17 +17,11 @@ from pyephember.pyephember import (
) )
import voluptuous as vol import voluptuous as vol
from homeassistant.components.climate import ( from homeassistant.components.climate import PLATFORM_SCHEMA, ClimateEntity
PLATFORM_SCHEMA,
ClimateEntity,
ClimateEntityFeature,
)
from homeassistant.components.climate.const import ( from homeassistant.components.climate.const import (
CURRENT_HVAC_HEAT, ClimateEntityFeature,
CURRENT_HVAC_IDLE, HVACAction,
HVAC_MODE_HEAT, HVACMode,
HVAC_MODE_HEAT_COOL,
HVAC_MODE_OFF,
) )
from homeassistant.const import ( from homeassistant.const import (
ATTR_TEMPERATURE, ATTR_TEMPERATURE,
@ -45,16 +39,16 @@ _LOGGER = logging.getLogger(__name__)
# Return cached results if last scan was less then this time ago # Return cached results if last scan was less then this time ago
SCAN_INTERVAL = timedelta(seconds=120) SCAN_INTERVAL = timedelta(seconds=120)
OPERATION_LIST = [HVAC_MODE_HEAT_COOL, HVAC_MODE_HEAT, HVAC_MODE_OFF] OPERATION_LIST = [HVACMode.HEAT_COOL, HVACMode.HEAT, HVACMode.OFF]
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
{vol.Required(CONF_USERNAME): cv.string, vol.Required(CONF_PASSWORD): cv.string} {vol.Required(CONF_USERNAME): cv.string, vol.Required(CONF_PASSWORD): cv.string}
) )
EPH_TO_HA_STATE = { EPH_TO_HA_STATE = {
"AUTO": HVAC_MODE_HEAT_COOL, "AUTO": HVACMode.HEAT_COOL,
"ON": HVAC_MODE_HEAT, "ON": HVACMode.HEAT,
"OFF": HVAC_MODE_OFF, "OFF": HVACMode.OFF,
} }
HA_STATE_TO_EPH = {value: key for key, value in EPH_TO_HA_STATE.items()} HA_STATE_TO_EPH = {value: key for key, value in EPH_TO_HA_STATE.items()}
@ -132,9 +126,9 @@ class EphEmberThermostat(ClimateEntity):
def hvac_action(self): def hvac_action(self):
"""Return current HVAC action.""" """Return current HVAC action."""
if zone_is_active(self._zone): if zone_is_active(self._zone):
return CURRENT_HVAC_HEAT return HVACAction.HEATING
return CURRENT_HVAC_IDLE return HVACAction.IDLE
@property @property
def hvac_mode(self): def hvac_mode(self):
@ -216,4 +210,4 @@ class EphEmberThermostat(ClimateEntity):
@staticmethod @staticmethod
def map_mode_eph_hass(operation_mode): def map_mode_eph_hass(operation_mode):
"""Map from eph mode to Home Assistant mode.""" """Map from eph mode to Home Assistant mode."""
return EPH_TO_HA_STATE.get(operation_mode.name, HVAC_MODE_HEAT_COOL) return EPH_TO_HA_STATE.get(operation_mode.name, HVACMode.HEAT_COOL)