102 lines
3.3 KiB
Python
102 lines
3.3 KiB
Python
"""Adds sensors for each appliance."""
|
|
|
|
from homeassistant.components.sensor import (
|
|
SensorEntity,
|
|
SensorStateClass,
|
|
SensorDeviceClass
|
|
)
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import (
|
|
PERCENTAGE,
|
|
UnitOfTemperature,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
|
|
from custom_components.midea_dehumidifier_lan.appliance_coordinator import (
|
|
ApplianceEntity,
|
|
)
|
|
from custom_components.midea_dehumidifier_lan.const import DOMAIN, UNIQUE_CLIMATE_PREFIX
|
|
from custom_components.midea_dehumidifier_lan.hub import Hub
|
|
|
|
|
|
async def async_setup_entry(
|
|
hass: HomeAssistant,
|
|
config_entry: ConfigEntry,
|
|
async_add_entities: AddEntitiesCallback,
|
|
) -> None:
|
|
"""Sets up current environment humidity and temperature sensors"""
|
|
|
|
hub: Hub = hass.data[DOMAIN][config_entry.entry_id]
|
|
|
|
# Dehumidifier sensors
|
|
async_add_entities(
|
|
CurrentHumiditySensor(c) for c in hub.coordinators if c.is_dehumidifier()
|
|
)
|
|
async_add_entities(
|
|
CurrentTemperatureSensor(c) for c in hub.coordinators if c.is_dehumidifier()
|
|
)
|
|
async_add_entities(
|
|
TankLevelSensor(c)
|
|
for c in hub.coordinators
|
|
if c.is_dehumidifier() and c.dehumidifier().capabilities.get("water_level")
|
|
)
|
|
# Climate sensors
|
|
async_add_entities(
|
|
OutsideTemperatureSensor(c) for c in hub.coordinators if c.is_climate()
|
|
)
|
|
|
|
|
|
class CurrentHumiditySensor(ApplianceEntity, SensorEntity):
|
|
"""Crrent environment humidity sensor"""
|
|
|
|
_attr_device_class = SensorDeviceClass.HUMIDITY
|
|
_attr_native_unit_of_measurement = PERCENTAGE
|
|
_attr_state_class = SensorStateClass.MEASUREMENT
|
|
_name_suffix = " Humidity"
|
|
|
|
def on_update(self) -> None:
|
|
self._attr_native_value = self.dehumidifier().current_humidity
|
|
|
|
|
|
class CurrentTemperatureSensor(ApplianceEntity, SensorEntity):
|
|
"""Current environment temperature sensor"""
|
|
|
|
_attr_device_class = SensorDeviceClass.TEMPERATURE
|
|
_attr_native_unit_of_measurement = UnitOfTemperature.CELSIUS
|
|
_attr_state_class = SensorStateClass.MEASUREMENT
|
|
_name_suffix = " Temperature"
|
|
|
|
def on_update(self) -> None:
|
|
self._attr_native_value = self.dehumidifier().current_temperature
|
|
|
|
|
|
class TankLevelSensor(ApplianceEntity, SensorEntity):
|
|
"""Current tank water level sensor"""
|
|
|
|
_attr_native_unit_of_measurement = PERCENTAGE
|
|
_attr_state_class = SensorStateClass.MEASUREMENT
|
|
_name_suffix = " Water Level"
|
|
|
|
def on_online(self, update: bool) -> None:
|
|
self._attr_entity_registry_enabled_default = (
|
|
self.dehumidifier().capabilities.get("water_level", False)
|
|
)
|
|
return super().on_online(update)
|
|
|
|
def on_update(self) -> None:
|
|
self._attr_native_value = self.dehumidifier().tank_level
|
|
|
|
|
|
class OutsideTemperatureSensor(ApplianceEntity, SensorEntity):
|
|
"""Current outside temperature sensor"""
|
|
|
|
_attr_device_class = SensorDeviceClass.TEMPERATURE
|
|
_attr_native_unit_of_measurement = UnitOfTemperature.CELSIUS
|
|
_attr_state_class = SensorStateClass.MEASUREMENT
|
|
_unique_id_prefx = UNIQUE_CLIMATE_PREFIX
|
|
_name_suffix = " Outdoor Temperature"
|
|
|
|
def on_update(self) -> None:
|
|
self._attr_native_value = self.airconditioner().outdoor_temperature
|