2341d510d3
Co-authored-by: pre-commit-ci-lite[bot] <117423508+pre-commit-ci-lite[bot]@users.noreply.github.com> Co-authored-by: Jonathan Swoboda <154711427+swoboda1337@users.noreply.github.com>
77 lines
2.2 KiB
Python
77 lines
2.2 KiB
Python
import esphome.config_validation as cv
|
|
from esphome.const import CONF_MAX_VALUE, CONF_MIN_VALUE, CONF_MODE, CONF_VALUE
|
|
|
|
from ..defines import (
|
|
BAR_MODES,
|
|
CONF_ANIMATED,
|
|
CONF_INDICATOR,
|
|
CONF_MAIN,
|
|
CONF_START_VALUE,
|
|
literal,
|
|
)
|
|
from ..lv_validation import animated, lv_int
|
|
from ..lvcode import lv
|
|
from ..types import LvNumber
|
|
from . import NumberType, Widget
|
|
|
|
# Note this file cannot be called "bar.py" because that name is disallowed.
|
|
|
|
CONF_BAR = "bar"
|
|
|
|
|
|
def validate_bar(config):
|
|
if config.get(CONF_MODE) != "LV_BAR_MODE_RANGE" and CONF_START_VALUE in config:
|
|
raise cv.Invalid(
|
|
f"{CONF_START_VALUE} is only allowed when {CONF_MODE} is set to 'RANGE'"
|
|
)
|
|
if (CONF_MIN_VALUE in config) != (CONF_MAX_VALUE in config):
|
|
raise cv.Invalid(
|
|
f"If either {CONF_MIN_VALUE} or {CONF_MAX_VALUE} is set, both must be set"
|
|
)
|
|
return config
|
|
|
|
|
|
BAR_SCHEMA = cv.Schema(
|
|
{
|
|
cv.Optional(CONF_VALUE): lv_int,
|
|
cv.Optional(CONF_START_VALUE): lv_int,
|
|
cv.Optional(CONF_MIN_VALUE): lv_int,
|
|
cv.Optional(CONF_MAX_VALUE): lv_int,
|
|
cv.Optional(CONF_MODE): BAR_MODES.one_of,
|
|
cv.Optional(CONF_ANIMATED, default=True): animated,
|
|
}
|
|
).add_extra(validate_bar)
|
|
|
|
|
|
class BarType(NumberType):
|
|
def __init__(self):
|
|
super().__init__(
|
|
CONF_BAR,
|
|
LvNumber("lv_bar_t"),
|
|
parts=(CONF_MAIN, CONF_INDICATOR),
|
|
schema=BAR_SCHEMA,
|
|
)
|
|
|
|
async def to_code(self, w: Widget, config):
|
|
var = w.obj
|
|
if mode := config.get(CONF_MODE):
|
|
lv.bar_set_mode(var, literal(mode))
|
|
is_animated = literal(config[CONF_ANIMATED])
|
|
if CONF_MIN_VALUE in config:
|
|
lv.bar_set_range(
|
|
var,
|
|
await lv_int.process(config[CONF_MIN_VALUE]),
|
|
await lv_int.process(config[CONF_MAX_VALUE]),
|
|
)
|
|
if value := await lv_int.process(config.get(CONF_VALUE)):
|
|
lv.bar_set_value(var, value, is_animated)
|
|
if start_value := await lv_int.process(config.get(CONF_START_VALUE)):
|
|
lv.bar_set_start_value(var, start_value, is_animated)
|
|
|
|
@property
|
|
def animated(self):
|
|
return True
|
|
|
|
|
|
bar_spec = BarType()
|