mirror of
https://github.com/home-assistant/core
synced 2024-09-03 08:14:07 +02:00
238 lines
7.1 KiB
Python
238 lines
7.1 KiB
Python
"""Support for Nexia / Trane XL Thermostats."""
|
|
from __future__ import annotations
|
|
|
|
from nexia.const import UNIT_CELSIUS
|
|
from nexia.thermostat import NexiaThermostat
|
|
|
|
from homeassistant.components.sensor import (
|
|
SensorDeviceClass,
|
|
SensorEntity,
|
|
SensorStateClass,
|
|
)
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import PERCENTAGE, TEMP_CELSIUS, TEMP_FAHRENHEIT
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
|
|
from .const import DOMAIN
|
|
from .coordinator import NexiaDataUpdateCoordinator
|
|
from .entity import NexiaThermostatEntity, NexiaThermostatZoneEntity
|
|
from .util import percent_conv
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
config_entry: ConfigEntry,
|
|
async_add_entities: AddEntitiesCallback,
|
|
) -> None:
|
|
"""Set up sensors for a Nexia device."""
|
|
|
|
coordinator: NexiaDataUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id]
|
|
nexia_home = coordinator.nexia_home
|
|
entities: list[NexiaThermostatEntity] = []
|
|
|
|
# Thermostat / System Sensors
|
|
for thermostat_id in nexia_home.get_thermostat_ids():
|
|
thermostat: NexiaThermostat = nexia_home.get_thermostat_by_id(thermostat_id)
|
|
|
|
entities.append(
|
|
NexiaThermostatSensor(
|
|
coordinator,
|
|
thermostat,
|
|
"get_system_status",
|
|
"System Status",
|
|
None,
|
|
None,
|
|
None,
|
|
)
|
|
)
|
|
# Air cleaner
|
|
entities.append(
|
|
NexiaThermostatSensor(
|
|
coordinator,
|
|
thermostat,
|
|
"get_air_cleaner_mode",
|
|
"Air Cleaner Mode",
|
|
None,
|
|
None,
|
|
None,
|
|
)
|
|
)
|
|
# Compressor Speed
|
|
if thermostat.has_variable_speed_compressor():
|
|
entities.append(
|
|
NexiaThermostatSensor(
|
|
coordinator,
|
|
thermostat,
|
|
"get_current_compressor_speed",
|
|
"Current Compressor Speed",
|
|
None,
|
|
PERCENTAGE,
|
|
SensorStateClass.MEASUREMENT,
|
|
percent_conv,
|
|
)
|
|
)
|
|
entities.append(
|
|
NexiaThermostatSensor(
|
|
coordinator,
|
|
thermostat,
|
|
"get_requested_compressor_speed",
|
|
"Requested Compressor Speed",
|
|
None,
|
|
PERCENTAGE,
|
|
SensorStateClass.MEASUREMENT,
|
|
percent_conv,
|
|
)
|
|
)
|
|
# Outdoor Temperature
|
|
if thermostat.has_outdoor_temperature():
|
|
unit = (
|
|
TEMP_CELSIUS
|
|
if thermostat.get_unit() == UNIT_CELSIUS
|
|
else TEMP_FAHRENHEIT
|
|
)
|
|
entities.append(
|
|
NexiaThermostatSensor(
|
|
coordinator,
|
|
thermostat,
|
|
"get_outdoor_temperature",
|
|
"Outdoor Temperature",
|
|
SensorDeviceClass.TEMPERATURE,
|
|
unit,
|
|
SensorStateClass.MEASUREMENT,
|
|
)
|
|
)
|
|
# Relative Humidity
|
|
if thermostat.has_relative_humidity():
|
|
entities.append(
|
|
NexiaThermostatSensor(
|
|
coordinator,
|
|
thermostat,
|
|
"get_relative_humidity",
|
|
"Relative Humidity",
|
|
SensorDeviceClass.HUMIDITY,
|
|
PERCENTAGE,
|
|
SensorStateClass.MEASUREMENT,
|
|
percent_conv,
|
|
)
|
|
)
|
|
|
|
# Zone Sensors
|
|
for zone_id in thermostat.get_zone_ids():
|
|
zone = thermostat.get_zone_by_id(zone_id)
|
|
unit = (
|
|
TEMP_CELSIUS
|
|
if thermostat.get_unit() == UNIT_CELSIUS
|
|
else TEMP_FAHRENHEIT
|
|
)
|
|
# Temperature
|
|
entities.append(
|
|
NexiaThermostatZoneSensor(
|
|
coordinator,
|
|
zone,
|
|
"get_temperature",
|
|
"Temperature",
|
|
SensorDeviceClass.TEMPERATURE,
|
|
unit,
|
|
SensorStateClass.MEASUREMENT,
|
|
None,
|
|
)
|
|
)
|
|
# Zone Status
|
|
entities.append(
|
|
NexiaThermostatZoneSensor(
|
|
coordinator, zone, "get_status", "Zone Status", None, None, None
|
|
)
|
|
)
|
|
# Setpoint Status
|
|
entities.append(
|
|
NexiaThermostatZoneSensor(
|
|
coordinator,
|
|
zone,
|
|
"get_setpoint_status",
|
|
"Zone Setpoint Status",
|
|
None,
|
|
None,
|
|
None,
|
|
)
|
|
)
|
|
|
|
async_add_entities(entities)
|
|
|
|
|
|
class NexiaThermostatSensor(NexiaThermostatEntity, SensorEntity):
|
|
"""Provides Nexia thermostat sensor support."""
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator,
|
|
thermostat,
|
|
sensor_call,
|
|
sensor_name,
|
|
sensor_class,
|
|
sensor_unit,
|
|
state_class,
|
|
modifier=None,
|
|
):
|
|
"""Initialize the sensor."""
|
|
super().__init__(
|
|
coordinator,
|
|
thermostat,
|
|
name=f"{thermostat.get_name()} {sensor_name}",
|
|
unique_id=f"{thermostat.thermostat_id}_{sensor_call}",
|
|
)
|
|
self._call = sensor_call
|
|
self._modifier = modifier
|
|
self._attr_device_class = sensor_class
|
|
self._attr_native_unit_of_measurement = sensor_unit
|
|
self._attr_state_class = state_class
|
|
|
|
@property
|
|
def native_value(self):
|
|
"""Return the state of the sensor."""
|
|
val = getattr(self._thermostat, self._call)()
|
|
if self._modifier:
|
|
val = self._modifier(val)
|
|
if isinstance(val, float):
|
|
val = round(val, 1)
|
|
return val
|
|
|
|
|
|
class NexiaThermostatZoneSensor(NexiaThermostatZoneEntity, SensorEntity):
|
|
"""Nexia Zone Sensor Support."""
|
|
|
|
def __init__(
|
|
self,
|
|
coordinator,
|
|
zone,
|
|
sensor_call,
|
|
sensor_name,
|
|
sensor_class,
|
|
sensor_unit,
|
|
state_class,
|
|
modifier=None,
|
|
):
|
|
"""Create a zone sensor."""
|
|
|
|
super().__init__(
|
|
coordinator,
|
|
zone,
|
|
name=f"{zone.get_name()} {sensor_name}",
|
|
unique_id=f"{zone.zone_id}_{sensor_call}",
|
|
)
|
|
self._call = sensor_call
|
|
self._modifier = modifier
|
|
self._attr_device_class = sensor_class
|
|
self._attr_native_unit_of_measurement = sensor_unit
|
|
self._attr_state_class = state_class
|
|
|
|
@property
|
|
def native_value(self):
|
|
"""Return the state of the sensor."""
|
|
val = getattr(self._zone, self._call)()
|
|
if self._modifier:
|
|
val = self._modifier(val)
|
|
if isinstance(val, float):
|
|
val = round(val, 1)
|
|
return val
|