mirror of
https://github.com/esphome/esphome.git
synced 2025-04-15 07:10:33 +01:00
131 lines
3.9 KiB
Python
131 lines
3.9 KiB
Python
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome.components import text_sensor
|
|
from esphome.const import (
|
|
CONF_ADDRESS,
|
|
CONF_BYTES,
|
|
CONF_DIV_RATIO,
|
|
CONF_ENTITY_ID,
|
|
CONF_ID,
|
|
CONF_MODE,
|
|
)
|
|
from .. import optolink_ns, CONF_OPTOLINK_ID, SENSOR_BASE_SCHEMA
|
|
|
|
DEPENDENCIES = ["optolink", "api"]
|
|
CODEOWNERS = ["@j0ta29"]
|
|
|
|
TextSensorMode = optolink_ns.enum("TextSensorMode")
|
|
MODE = {
|
|
"MAP": TextSensorMode.MAP,
|
|
"RAW": TextSensorMode.RAW,
|
|
"DAY_SCHEDULE": TextSensorMode.DAY_SCHEDULE,
|
|
"DAY_SCHEDULE_SYNCHRONIZED": TextSensorMode.DAY_SCHEDULE_SYNCHRONIZED,
|
|
"DEVICE_INFO": TextSensorMode.DEVICE_INFO,
|
|
}
|
|
DAY_OF_WEEK = {
|
|
"MONDAY": 0,
|
|
"TUESDAY": 1,
|
|
"WEDNESDAY": 2,
|
|
"THURSDAY": 3,
|
|
"FRIDAY": 4,
|
|
"SATURDAY": 5,
|
|
"SUNDAY": 6,
|
|
}
|
|
CONF_DAY_OF_WEEK = "day_of_week"
|
|
|
|
OptolinkTextSensor = optolink_ns.class_(
|
|
"OptolinkTextSensor", text_sensor.TextSensor, cg.PollingComponent
|
|
)
|
|
|
|
|
|
def check_bytes():
|
|
def validator_(config):
|
|
bytes_needed = config[CONF_MODE] in ["MAP", "RAW"]
|
|
bytes_defined = CONF_BYTES in config
|
|
if bytes_needed and not bytes_defined:
|
|
raise cv.Invalid(f"{CONF_BYTES} is required in mode MAP or RAW")
|
|
if not bytes_needed and bytes_defined:
|
|
raise cv.Invalid(
|
|
f"{CONF_BYTES} is not allowed in mode DAY_SCHEDULE and DAY_SCHEDULE_SYNCHRONIZED"
|
|
)
|
|
return config
|
|
|
|
return validator_
|
|
|
|
|
|
def check_dow():
|
|
def validator_(config):
|
|
if (
|
|
config[CONF_MODE] in ["DAY_SCHEDULE", "DAY_SCHEDULE_SYNCHRONIZED"]
|
|
and CONF_DAY_OF_WEEK not in config
|
|
):
|
|
raise cv.Invalid(f"{CONF_DAY_OF_WEEK} is required in mode DAY_SCHEDULE")
|
|
if (
|
|
config[CONF_MODE] not in ["DAY_SCHEDULE", "DAY_SCHEDULE_SYNCHRONIZED"]
|
|
and CONF_DAY_OF_WEEK in config
|
|
):
|
|
raise cv.Invalid(
|
|
f"{CONF_DAY_OF_WEEK} is only allowed in mode DAY_SCHEDULE or DAY_SCHEDULE_SYNCHRONIZED"
|
|
)
|
|
return config
|
|
|
|
return validator_
|
|
|
|
|
|
def check_entity_id():
|
|
def validator_(config):
|
|
if (
|
|
config[CONF_MODE] in ["DAY_SCHEDULE_SYNCHRONIZED"]
|
|
and CONF_ENTITY_ID not in config
|
|
):
|
|
raise cv.Invalid(
|
|
f"{CONF_ENTITY_ID} is required in mode DAY_SCHEDULE_SYNCHRONIZED"
|
|
)
|
|
if (
|
|
config[CONF_MODE] not in ["DAY_SCHEDULE_SYNCHRONIZED"]
|
|
and CONF_ENTITY_ID in config
|
|
):
|
|
raise cv.Invalid(
|
|
f"{CONF_ENTITY_ID} is only allowed in mode DAY_SCHEDULE_SYNCHRONIZED"
|
|
)
|
|
return config
|
|
|
|
return validator_
|
|
|
|
|
|
CONFIG_SCHEMA = cv.All(
|
|
text_sensor.TEXT_SENSOR_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(OptolinkTextSensor),
|
|
cv.Optional(CONF_MODE, default="MAP"): cv.enum(MODE, upper=True),
|
|
cv.Optional(CONF_BYTES): cv.int_range(min=1, max=9),
|
|
cv.Optional(CONF_DAY_OF_WEEK): cv.enum(DAY_OF_WEEK, upper=True),
|
|
cv.Optional(CONF_ENTITY_ID): cv.entity_id,
|
|
}
|
|
)
|
|
.extend(cv.COMPONENT_SCHEMA)
|
|
.extend(SENSOR_BASE_SCHEMA),
|
|
check_bytes(),
|
|
check_dow(),
|
|
check_entity_id(),
|
|
)
|
|
|
|
|
|
async def to_code(config):
|
|
component = await cg.get_variable(config[CONF_OPTOLINK_ID])
|
|
var = cg.new_Pvariable(config[CONF_ID], component)
|
|
|
|
await cg.register_component(var, config)
|
|
await text_sensor.register_text_sensor(var, config)
|
|
|
|
cg.add(var.set_mode(config[CONF_MODE]))
|
|
if CONF_ADDRESS in config:
|
|
cg.add(var.set_address(config[CONF_ADDRESS]))
|
|
cg.add(var.set_div_ratio(config[CONF_DIV_RATIO]))
|
|
if CONF_BYTES in config:
|
|
cg.add(var.set_bytes(config[CONF_BYTES]))
|
|
if CONF_DAY_OF_WEEK in config:
|
|
cg.add(var.set_day_of_week(config[CONF_DAY_OF_WEEK]))
|
|
if CONF_ENTITY_ID in config:
|
|
cg.add(var.set_entity_id(config[CONF_ENTITY_ID]))
|