mirror of
https://github.com/home-assistant/core
synced 2024-07-30 21:18:57 +02:00
![springstan](/assets/img/avatar_default.png)
* Use f-strings in integrations starting with "H" * Use f-strings in integrations starting with "I" * Use f-strings in integrations starting with "J" * Use f-strings in integrations starting with "K" * Use f-strings in integrations starting with "L" * Fix lint error * Use join instead of f-string in homekit_controller * Use local variables with f-strings * Fix lint error * Escape the characters in f-string * Sort imports with isort in homeworks light * Fix pylint error * Fix broken tests * Fix broken tests v2
314 lines
10 KiB
Python
314 lines
10 KiB
Python
"""Support to set a numeric value from a slider or text box."""
|
|
import logging
|
|
import typing
|
|
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.const import (
|
|
ATTR_EDITABLE,
|
|
ATTR_MODE,
|
|
ATTR_UNIT_OF_MEASUREMENT,
|
|
CONF_ICON,
|
|
CONF_ID,
|
|
CONF_MODE,
|
|
CONF_NAME,
|
|
SERVICE_RELOAD,
|
|
)
|
|
from homeassistant.core import callback
|
|
from homeassistant.helpers import collection
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.helpers.entity_component import EntityComponent
|
|
from homeassistant.helpers.restore_state import RestoreEntity
|
|
import homeassistant.helpers.service
|
|
from homeassistant.helpers.storage import Store
|
|
from homeassistant.helpers.typing import ConfigType, HomeAssistantType, ServiceCallType
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
DOMAIN = "input_number"
|
|
|
|
CONF_INITIAL = "initial"
|
|
CONF_MIN = "min"
|
|
CONF_MAX = "max"
|
|
CONF_STEP = "step"
|
|
|
|
MODE_SLIDER = "slider"
|
|
MODE_BOX = "box"
|
|
|
|
ATTR_INITIAL = "initial"
|
|
ATTR_VALUE = "value"
|
|
ATTR_MIN = "min"
|
|
ATTR_MAX = "max"
|
|
ATTR_STEP = "step"
|
|
|
|
SERVICE_SET_VALUE = "set_value"
|
|
SERVICE_INCREMENT = "increment"
|
|
SERVICE_DECREMENT = "decrement"
|
|
|
|
|
|
def _cv_input_number(cfg):
|
|
"""Configure validation helper for input number (voluptuous)."""
|
|
minimum = cfg.get(CONF_MIN)
|
|
maximum = cfg.get(CONF_MAX)
|
|
if minimum >= maximum:
|
|
raise vol.Invalid(
|
|
f"Maximum ({minimum}) is not greater than minimum ({maximum})"
|
|
)
|
|
state = cfg.get(CONF_INITIAL)
|
|
if state is not None and (state < minimum or state > maximum):
|
|
raise vol.Invalid(f"Initial value {state} not in range {minimum}-{maximum}")
|
|
return cfg
|
|
|
|
|
|
CREATE_FIELDS = {
|
|
vol.Required(CONF_NAME): vol.All(str, vol.Length(min=1)),
|
|
vol.Required(CONF_MIN): vol.Coerce(float),
|
|
vol.Required(CONF_MAX): vol.Coerce(float),
|
|
vol.Optional(CONF_INITIAL): vol.Coerce(float),
|
|
vol.Optional(CONF_STEP, default=1): vol.All(vol.Coerce(float), vol.Range(min=1e-3)),
|
|
vol.Optional(CONF_ICON): cv.icon,
|
|
vol.Optional(ATTR_UNIT_OF_MEASUREMENT): cv.string,
|
|
vol.Optional(CONF_MODE, default=MODE_SLIDER): vol.In([MODE_BOX, MODE_SLIDER]),
|
|
}
|
|
|
|
UPDATE_FIELDS = {
|
|
vol.Optional(CONF_NAME): cv.string,
|
|
vol.Optional(CONF_MIN): vol.Coerce(float),
|
|
vol.Optional(CONF_MAX): vol.Coerce(float),
|
|
vol.Optional(CONF_INITIAL): vol.Coerce(float),
|
|
vol.Optional(CONF_STEP): vol.All(vol.Coerce(float), vol.Range(min=1e-3)),
|
|
vol.Optional(CONF_ICON): cv.icon,
|
|
vol.Optional(ATTR_UNIT_OF_MEASUREMENT): cv.string,
|
|
vol.Optional(CONF_MODE): vol.In([MODE_BOX, MODE_SLIDER]),
|
|
}
|
|
|
|
CONFIG_SCHEMA = vol.Schema(
|
|
{
|
|
DOMAIN: cv.schema_with_slug_keys(
|
|
vol.All(
|
|
{
|
|
vol.Optional(CONF_NAME): cv.string,
|
|
vol.Required(CONF_MIN): vol.Coerce(float),
|
|
vol.Required(CONF_MAX): vol.Coerce(float),
|
|
vol.Optional(CONF_INITIAL): vol.Coerce(float),
|
|
vol.Optional(CONF_STEP, default=1): vol.All(
|
|
vol.Coerce(float), vol.Range(min=1e-3)
|
|
),
|
|
vol.Optional(CONF_ICON): cv.icon,
|
|
vol.Optional(ATTR_UNIT_OF_MEASUREMENT): cv.string,
|
|
vol.Optional(CONF_MODE, default=MODE_SLIDER): vol.In(
|
|
[MODE_BOX, MODE_SLIDER]
|
|
),
|
|
},
|
|
_cv_input_number,
|
|
)
|
|
)
|
|
},
|
|
extra=vol.ALLOW_EXTRA,
|
|
)
|
|
RELOAD_SERVICE_SCHEMA = vol.Schema({})
|
|
STORAGE_KEY = DOMAIN
|
|
STORAGE_VERSION = 1
|
|
|
|
|
|
async def async_setup(hass: HomeAssistantType, config: ConfigType) -> bool:
|
|
"""Set up an input slider."""
|
|
component = EntityComponent(_LOGGER, DOMAIN, hass)
|
|
id_manager = collection.IDManager()
|
|
|
|
yaml_collection = collection.YamlCollection(
|
|
logging.getLogger(f"{__name__}.yaml_collection"), id_manager
|
|
)
|
|
collection.attach_entity_component_collection(
|
|
component, yaml_collection, InputNumber.from_yaml
|
|
)
|
|
|
|
storage_collection = NumberStorageCollection(
|
|
Store(hass, STORAGE_VERSION, STORAGE_KEY),
|
|
logging.getLogger(f"{__name__}.storage_collection"),
|
|
id_manager,
|
|
)
|
|
collection.attach_entity_component_collection(
|
|
component, storage_collection, InputNumber
|
|
)
|
|
|
|
await yaml_collection.async_load(
|
|
[{CONF_ID: id_, **(conf or {})} for id_, conf in config.get(DOMAIN, {}).items()]
|
|
)
|
|
await storage_collection.async_load()
|
|
|
|
collection.StorageCollectionWebsocket(
|
|
storage_collection, DOMAIN, DOMAIN, CREATE_FIELDS, UPDATE_FIELDS
|
|
).async_setup(hass)
|
|
|
|
collection.attach_entity_registry_cleaner(hass, DOMAIN, DOMAIN, yaml_collection)
|
|
collection.attach_entity_registry_cleaner(hass, DOMAIN, DOMAIN, storage_collection)
|
|
|
|
async def reload_service_handler(service_call: ServiceCallType) -> None:
|
|
"""Reload yaml entities."""
|
|
conf = await component.async_prepare_reload(skip_reset=True)
|
|
if conf is None:
|
|
conf = {DOMAIN: {}}
|
|
await yaml_collection.async_load(
|
|
[{CONF_ID: id_, **conf} for id_, conf in conf.get(DOMAIN, {}).items()]
|
|
)
|
|
|
|
homeassistant.helpers.service.async_register_admin_service(
|
|
hass,
|
|
DOMAIN,
|
|
SERVICE_RELOAD,
|
|
reload_service_handler,
|
|
schema=RELOAD_SERVICE_SCHEMA,
|
|
)
|
|
|
|
component.async_register_entity_service(
|
|
SERVICE_SET_VALUE,
|
|
{vol.Required(ATTR_VALUE): vol.Coerce(float)},
|
|
"async_set_value",
|
|
)
|
|
|
|
component.async_register_entity_service(SERVICE_INCREMENT, {}, "async_increment")
|
|
|
|
component.async_register_entity_service(SERVICE_DECREMENT, {}, "async_decrement")
|
|
|
|
return True
|
|
|
|
|
|
class NumberStorageCollection(collection.StorageCollection):
|
|
"""Input storage based collection."""
|
|
|
|
CREATE_SCHEMA = vol.Schema(vol.All(CREATE_FIELDS, _cv_input_number))
|
|
UPDATE_SCHEMA = vol.Schema(UPDATE_FIELDS)
|
|
|
|
async def _process_create_data(self, data: typing.Dict) -> typing.Dict:
|
|
"""Validate the config is valid."""
|
|
return self.CREATE_SCHEMA(data)
|
|
|
|
@callback
|
|
def _get_suggested_id(self, info: typing.Dict) -> str:
|
|
"""Suggest an ID based on the config."""
|
|
return info[CONF_NAME]
|
|
|
|
async def _update_data(self, data: dict, update_data: typing.Dict) -> typing.Dict:
|
|
"""Return a new updated data object."""
|
|
update_data = self.UPDATE_SCHEMA(update_data)
|
|
return _cv_input_number({**data, **update_data})
|
|
|
|
|
|
class InputNumber(RestoreEntity):
|
|
"""Representation of a slider."""
|
|
|
|
def __init__(self, config: typing.Dict):
|
|
"""Initialize an input number."""
|
|
self._config = config
|
|
self.editable = True
|
|
self._current_value = config.get(CONF_INITIAL)
|
|
|
|
@classmethod
|
|
def from_yaml(cls, config: typing.Dict) -> "InputNumber":
|
|
"""Return entity instance initialized from yaml storage."""
|
|
input_num = cls(config)
|
|
input_num.entity_id = f"{DOMAIN}.{config[CONF_ID]}"
|
|
input_num.editable = False
|
|
return input_num
|
|
|
|
@property
|
|
def should_poll(self):
|
|
"""If entity should be polled."""
|
|
return False
|
|
|
|
@property
|
|
def _minimum(self) -> float:
|
|
"""Return minimum allowed value."""
|
|
return self._config[CONF_MIN]
|
|
|
|
@property
|
|
def _maximum(self) -> float:
|
|
"""Return maximum allowed value."""
|
|
return self._config[CONF_MAX]
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the input slider."""
|
|
return self._config.get(CONF_NAME)
|
|
|
|
@property
|
|
def icon(self):
|
|
"""Return the icon to be used for this entity."""
|
|
return self._config.get(CONF_ICON)
|
|
|
|
@property
|
|
def state(self):
|
|
"""Return the state of the component."""
|
|
return self._current_value
|
|
|
|
@property
|
|
def _step(self) -> int:
|
|
"""Return entity's increment/decrement step."""
|
|
return self._config[CONF_STEP]
|
|
|
|
@property
|
|
def unit_of_measurement(self):
|
|
"""Return the unit the value is expressed in."""
|
|
return self._config.get(ATTR_UNIT_OF_MEASUREMENT)
|
|
|
|
@property
|
|
def unique_id(self) -> typing.Optional[str]:
|
|
"""Return unique id of the entity."""
|
|
return self._config[CONF_ID]
|
|
|
|
@property
|
|
def state_attributes(self):
|
|
"""Return the state attributes."""
|
|
return {
|
|
ATTR_INITIAL: self._config.get(CONF_INITIAL),
|
|
ATTR_EDITABLE: self.editable,
|
|
ATTR_MIN: self._minimum,
|
|
ATTR_MAX: self._maximum,
|
|
ATTR_STEP: self._step,
|
|
ATTR_MODE: self._config[CONF_MODE],
|
|
}
|
|
|
|
async def async_added_to_hass(self):
|
|
"""Run when entity about to be added to hass."""
|
|
await super().async_added_to_hass()
|
|
if self._current_value is not None:
|
|
return
|
|
|
|
state = await self.async_get_last_state()
|
|
value = state and float(state.state)
|
|
|
|
# Check against None because value can be 0
|
|
if value is not None and self._minimum <= value <= self._maximum:
|
|
self._current_value = value
|
|
else:
|
|
self._current_value = self._minimum
|
|
|
|
async def async_set_value(self, value):
|
|
"""Set new value."""
|
|
num_value = float(value)
|
|
|
|
if num_value < self._minimum or num_value > self._maximum:
|
|
raise vol.Invalid(
|
|
f"Invalid value for {self.entity_id}: {value} (range {self._minimum} - {self._maximum})"
|
|
)
|
|
|
|
self._current_value = num_value
|
|
self.async_write_ha_state()
|
|
|
|
async def async_increment(self):
|
|
"""Increment value."""
|
|
await self.async_set_value(min(self._current_value + self._step, self._maximum))
|
|
|
|
async def async_decrement(self):
|
|
"""Decrement value."""
|
|
await self.async_set_value(max(self._current_value - self._step, self._minimum))
|
|
|
|
async def async_update_config(self, config: typing.Dict) -> None:
|
|
"""Handle when the config is updated."""
|
|
self._config = config
|
|
# just in case min/max values changed
|
|
self._current_value = min(self._current_value, self._maximum)
|
|
self._current_value = max(self._current_value, self._minimum)
|
|
self.async_write_ha_state()
|