Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use climate enums in toon (#70744)

pull/70764/head
epenet 2022-04-26 08:46:39 +02:00 committed by GitHub
parent 9342a1b577
commit b9379b5a88
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
homeassistant/components/toon/climate.py
View File

@ -10,15 +10,15 @@ from toonapi import (
ACTIVE_STATE_SLEEP, ACTIVE_STATE_SLEEP,
) )
from homeassistant.components.climate import ClimateEntity, ClimateEntityFeature from homeassistant.components.climate import ClimateEntity
from homeassistant.components.climate.const import ( from homeassistant.components.climate.const import (
CURRENT_HVAC_HEAT,
CURRENT_HVAC_IDLE,
HVAC_MODE_HEAT,
PRESET_AWAY, PRESET_AWAY,
PRESET_COMFORT, PRESET_COMFORT,
PRESET_HOME, PRESET_HOME,
PRESET_SLEEP, PRESET_SLEEP,
ClimateEntityFeature,
HVACAction,
HVACMode,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_TEMPERATURE, TEMP_CELSIUS from homeassistant.const import ATTR_TEMPERATURE, TEMP_CELSIUS
@ -42,7 +42,7 @@ async def async_setup_entry(
class ToonThermostatDevice(ToonDisplayDeviceEntity, ClimateEntity): class ToonThermostatDevice(ToonDisplayDeviceEntity, ClimateEntity):
"""Representation of a Toon climate device.""" """Representation of a Toon climate device."""
_attr_hvac_mode = HVAC_MODE_HEAT _attr_hvac_mode = HVACMode.HEAT
_attr_icon = "mdi:thermostat" _attr_icon = "mdi:thermostat"
_attr_max_temp = DEFAULT_MAX_TEMP _attr_max_temp = DEFAULT_MAX_TEMP
_attr_min_temp = DEFAULT_MIN_TEMP _attr_min_temp = DEFAULT_MIN_TEMP
@ -58,7 +58,7 @@ class ToonThermostatDevice(ToonDisplayDeviceEntity, ClimateEntity):
) -> None: ) -> None:
"""Initialize Toon climate entity.""" """Initialize Toon climate entity."""
super().__init__(coordinator) super().__init__(coordinator)
self._attr_hvac_modes = [HVAC_MODE_HEAT] self._attr_hvac_modes = [HVACMode.HEAT]
self._attr_preset_modes = [ self._attr_preset_modes = [
PRESET_AWAY, PRESET_AWAY,
PRESET_COMFORT, PRESET_COMFORT,
@ -70,11 +70,11 @@ class ToonThermostatDevice(ToonDisplayDeviceEntity, ClimateEntity):
) )
@property @property
def hvac_action(self) -> str | None: def hvac_action(self) -> HVACAction:
"""Return the current running hvac operation.""" """Return the current running hvac operation."""
if self.coordinator.data.thermostat.heating: if self.coordinator.data.thermostat.heating:
return CURRENT_HVAC_HEAT return HVACAction.HEATING
return CURRENT_HVAC_IDLE return HVACAction.IDLE
@property @property
def preset_mode(self) -> str | None: def preset_mode(self) -> str | None:
@ -120,7 +120,7 @@ class ToonThermostatDevice(ToonDisplayDeviceEntity, ClimateEntity):
if preset_mode in mapping: if preset_mode in mapping:
await self.coordinator.toon.set_active_state(mapping[preset_mode]) await self.coordinator.toon.set_active_state(mapping[preset_mode])
def set_hvac_mode(self, hvac_mode: str) -> None: def set_hvac_mode(self, hvac_mode: HVACMode) -> None:
"""Set new target hvac mode.""" """Set new target hvac mode."""
# Intentionally left empty # Intentionally left empty
# The HAVC mode is always HEAT # The HAVC mode is always HEAT