95 lines
3.1 KiB
Python
95 lines
3.1 KiB
Python
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome.components import text_sensor
|
|
from esphome.const import (
|
|
CONF_ID,
|
|
)
|
|
from . import Dsmr, CONF_DSMR_ID
|
|
|
|
AUTO_LOAD = ["dsmr"]
|
|
|
|
CONFIG_SCHEMA = cv.Schema(
|
|
{
|
|
cv.GenerateID(CONF_DSMR_ID): cv.use_id(Dsmr),
|
|
cv.Optional("identification"): text_sensor.TEXT_SENSOR_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(text_sensor.TextSensor),
|
|
}
|
|
),
|
|
cv.Optional("p1_version"): text_sensor.TEXT_SENSOR_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(text_sensor.TextSensor),
|
|
}
|
|
),
|
|
cv.Optional("p1_version_be"): text_sensor.TEXT_SENSOR_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(text_sensor.TextSensor),
|
|
}
|
|
),
|
|
cv.Optional("timestamp"): text_sensor.TEXT_SENSOR_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(text_sensor.TextSensor),
|
|
}
|
|
),
|
|
cv.Optional("electricity_tariff"): text_sensor.TEXT_SENSOR_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(text_sensor.TextSensor),
|
|
}
|
|
),
|
|
cv.Optional("electricity_failure_log"): text_sensor.TEXT_SENSOR_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(text_sensor.TextSensor),
|
|
}
|
|
),
|
|
cv.Optional("message_short"): text_sensor.TEXT_SENSOR_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(text_sensor.TextSensor),
|
|
}
|
|
),
|
|
cv.Optional("message_long"): text_sensor.TEXT_SENSOR_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(text_sensor.TextSensor),
|
|
}
|
|
),
|
|
cv.Optional("gas_equipment_id"): text_sensor.TEXT_SENSOR_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(text_sensor.TextSensor),
|
|
}
|
|
),
|
|
cv.Optional("thermal_equipment_id"): text_sensor.TEXT_SENSOR_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(text_sensor.TextSensor),
|
|
}
|
|
),
|
|
cv.Optional("water_equipment_id"): text_sensor.TEXT_SENSOR_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(text_sensor.TextSensor),
|
|
}
|
|
),
|
|
cv.Optional("sub_equipment_id"): text_sensor.TEXT_SENSOR_SCHEMA.extend(
|
|
{
|
|
cv.GenerateID(): cv.declare_id(text_sensor.TextSensor),
|
|
}
|
|
),
|
|
}
|
|
).extend(cv.COMPONENT_SCHEMA)
|
|
|
|
|
|
async def to_code(config):
|
|
hub = await cg.get_variable(config[CONF_DSMR_ID])
|
|
|
|
text_sensors = []
|
|
for key, conf in config.items():
|
|
if not isinstance(conf, dict):
|
|
continue
|
|
id = conf.get("id")
|
|
if id and id.type == text_sensor.TextSensor:
|
|
var = cg.new_Pvariable(conf[CONF_ID])
|
|
await text_sensor.register_text_sensor(var, conf)
|
|
cg.add(getattr(hub, f"set_{key}")(var))
|
|
text_sensors.append(f"F({key})")
|
|
|
|
cg.add_define(
|
|
"DSMR_TEXT_SENSOR_LIST(F, sep)", cg.RawExpression(" sep ".join(text_sensors))
|
|
)
|