mirror of
https://github.com/home-assistant/core.git
synced 2026-08-07 13:55:32 +01:00
87 lines
3.0 KiB
Python
87 lines
3.0 KiB
Python
"""Provides conditions for temperature."""
|
|
|
|
from typing import override
|
|
|
|
from homeassistant.components.climate import (
|
|
DOMAIN as CLIMATE_DOMAIN,
|
|
ClimateEntityStateAttribute,
|
|
)
|
|
from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN, SensorDeviceClass
|
|
from homeassistant.components.water_heater import (
|
|
DOMAIN as WATER_HEATER_DOMAIN,
|
|
WaterHeaterStateAttribute,
|
|
)
|
|
from homeassistant.components.weather import (
|
|
DOMAIN as WEATHER_DOMAIN,
|
|
WeatherEntityStateAttribute,
|
|
)
|
|
from homeassistant.const import EntityStateAttribute, UnitOfTemperature
|
|
from homeassistant.core import HomeAssistant, State
|
|
from homeassistant.helpers.automation import DomainSpec
|
|
from homeassistant.helpers.condition import (
|
|
Condition,
|
|
EntityNumericalConditionWithUnitBase,
|
|
)
|
|
from homeassistant.util.unit_conversion import TemperatureConverter
|
|
|
|
TEMPERATURE_DOMAIN_SPECS: dict[str, DomainSpec] = {
|
|
CLIMATE_DOMAIN: DomainSpec(
|
|
value_source=ClimateEntityStateAttribute.CURRENT_TEMPERATURE,
|
|
),
|
|
SENSOR_DOMAIN: DomainSpec(
|
|
device_class=SensorDeviceClass.TEMPERATURE,
|
|
),
|
|
WATER_HEATER_DOMAIN: DomainSpec(
|
|
value_source=WaterHeaterStateAttribute.CURRENT_TEMPERATURE,
|
|
),
|
|
WEATHER_DOMAIN: DomainSpec(
|
|
value_source=WeatherEntityStateAttribute.TEMPERATURE,
|
|
),
|
|
}
|
|
|
|
|
|
class TemperatureCondition(EntityNumericalConditionWithUnitBase):
|
|
"""Condition for temperature value."""
|
|
|
|
_base_unit = UnitOfTemperature.CELSIUS
|
|
_domain_specs = TEMPERATURE_DOMAIN_SPECS
|
|
_unit_converter = TemperatureConverter
|
|
|
|
@override
|
|
def _should_include(self, state: State) -> bool:
|
|
"""Skip attribute-source entities that lack the temperature attribute.
|
|
|
|
Mirrors the temperature trigger: for climate / water_heater /
|
|
weather (attribute-based), the entity is filtered when the source
|
|
attribute is absent; sensor entities (state-value-based) fall
|
|
through to the base impl.
|
|
"""
|
|
if not super()._should_include(state):
|
|
return False
|
|
domain_spec = self._domain_specs[state.domain]
|
|
if domain_spec.value_source is None:
|
|
return True
|
|
return state.attributes.get(domain_spec.value_source) is not None
|
|
|
|
@override
|
|
def _get_entity_unit(self, entity_state: State) -> str | None:
|
|
"""Get the temperature unit of an entity from its state."""
|
|
if entity_state.domain == SENSOR_DOMAIN:
|
|
return entity_state.attributes.get(EntityStateAttribute.UNIT_OF_MEASUREMENT)
|
|
if entity_state.domain == WEATHER_DOMAIN:
|
|
return entity_state.attributes.get(
|
|
WeatherEntityStateAttribute.TEMPERATURE_UNIT
|
|
)
|
|
# Climate and water_heater: show_temp converts to system unit
|
|
return self._hass.config.units.temperature_unit
|
|
|
|
|
|
CONDITIONS: dict[str, type[Condition]] = {
|
|
"is_value": TemperatureCondition,
|
|
}
|
|
|
|
|
|
async def async_get_conditions(hass: HomeAssistant) -> dict[str, type[Condition]]:
|
|
"""Return the temperature conditions."""
|
|
return CONDITIONS
|