Use climate enums in tfiac (#70746)

This commit is contained in:
epenet 2022-04-26 09:01:54 +02:00 committed by GitHub
parent ba7cff505e
commit ab1dd7887e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -8,26 +8,18 @@ import logging
from pytfiac import Tfiac from pytfiac import Tfiac
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 (
FAN_AUTO, FAN_AUTO,
FAN_HIGH, FAN_HIGH,
FAN_LOW, FAN_LOW,
FAN_MEDIUM, FAN_MEDIUM,
HVAC_MODE_AUTO,
HVAC_MODE_COOL,
HVAC_MODE_DRY,
HVAC_MODE_FAN_ONLY,
HVAC_MODE_HEAT,
HVAC_MODE_OFF,
SWING_BOTH, SWING_BOTH,
SWING_HORIZONTAL, SWING_HORIZONTAL,
SWING_OFF, SWING_OFF,
SWING_VERTICAL, SWING_VERTICAL,
ClimateEntityFeature,
HVACMode,
) )
from homeassistant.const import ATTR_TEMPERATURE, CONF_HOST, TEMP_FAHRENHEIT from homeassistant.const import ATTR_TEMPERATURE, CONF_HOST, TEMP_FAHRENHEIT
from homeassistant.core import HomeAssistant from homeassistant.core import HomeAssistant
@ -45,12 +37,12 @@ MIN_TEMP = 61
MAX_TEMP = 88 MAX_TEMP = 88
HVAC_MAP = { HVAC_MAP = {
HVAC_MODE_HEAT: "heat", HVACMode.HEAT: "heat",
HVAC_MODE_AUTO: "selfFeel", HVACMode.AUTO: "selfFeel",
HVAC_MODE_DRY: "dehumi", HVACMode.DRY: "dehumi",
HVAC_MODE_FAN_ONLY: "fan", HVACMode.FAN_ONLY: "fan",
HVAC_MODE_COOL: "cool", HVACMode.COOL: "cool",
HVAC_MODE_OFF: "off", HVACMode.OFF: "off",
} }
HVAC_MAP_REV = {v: k for k, v in HVAC_MAP.items()} HVAC_MAP_REV = {v: k for k, v in HVAC_MAP.items()}
@ -140,19 +132,19 @@ class TfiacClimate(ClimateEntity):
return self._client.status["current_temp"] return self._client.status["current_temp"]
@property @property
def hvac_mode(self): def hvac_mode(self) -> HVACMode | None:
"""Return hvac operation ie. heat, cool mode. """Return hvac operation ie. heat, cool mode.
Need to be one of HVAC_MODE_*. Need to be one of HVAC_MODE_*.
""" """
if self._client.status[ON_MODE] != "on": if self._client.status[ON_MODE] != "on":
return HVAC_MODE_OFF return HVACMode.OFF
state = self._client.status["operation"] state = self._client.status["operation"]
return HVAC_MAP_REV.get(state) return HVAC_MAP_REV.get(state)
@property @property
def hvac_modes(self): def hvac_modes(self) -> list[HVACMode]:
"""Return the list of available hvac operation modes. """Return the list of available hvac operation modes.
Need to be a subset of HVAC_MODES. Need to be a subset of HVAC_MODES.
@ -184,9 +176,9 @@ class TfiacClimate(ClimateEntity):
if (temp := kwargs.get(ATTR_TEMPERATURE)) is not None: if (temp := kwargs.get(ATTR_TEMPERATURE)) is not None:
await self._client.set_state(TARGET_TEMP, temp) await self._client.set_state(TARGET_TEMP, temp)
async def async_set_hvac_mode(self, hvac_mode): async def async_set_hvac_mode(self, hvac_mode: HVACMode) -> None:
"""Set new target hvac mode.""" """Set new target hvac mode."""
if hvac_mode == HVAC_MODE_OFF: if hvac_mode == HVACMode.OFF:
await self._client.set_state(ON_MODE, "off") await self._client.set_state(ON_MODE, "off")
else: else:
await self._client.set_state(OPERATION_MODE, HVAC_MAP[hvac_mode]) await self._client.set_state(OPERATION_MODE, HVAC_MAP[hvac_mode])