import esphome.codegen as cg import esphome.config_validation as cv from esphome.components import sensor from esphome.const import CONF_ID, CONF_FORCE_UPDATE, CONF_RANGE_FROM, CONF_RANGE_TO from .. import ( bslamp2_ns, CODEOWNERS, CONF_FRONT_PANEL_HAL_ID, FrontPanelHAL ) DEPENDENCIES = ["xiaomi_bslamp2"] XiaomiBslamp2SliderSensor = bslamp2_ns.class_( "XiaomiBslamp2SliderSensor", sensor.Sensor, cg.Component) CONFIG_SCHEMA = sensor.SENSOR_SCHEMA.extend( { cv.GenerateID(): cv.declare_id(XiaomiBslamp2SliderSensor), cv.GenerateID(CONF_FRONT_PANEL_HAL_ID): cv.use_id(FrontPanelHAL), cv.Optional(CONF_FORCE_UPDATE, default=True): cv.boolean, cv.Optional(CONF_RANGE_FROM, default=0.01): cv.float_, cv.Optional(CONF_RANGE_TO, default=1.00): cv.float_, } ).extend(cv.COMPONENT_SCHEMA) def to_code(config): var = cg.new_Pvariable(config[CONF_ID]) yield cg.register_component(var, config) yield sensor.register_sensor(var, config) front_panel_hal_var = yield cg.get_variable(config[CONF_FRONT_PANEL_HAL_ID]) cg.add(var.set_parent(front_panel_hal_var)) cg.add(var.set_range_from(config[CONF_RANGE_FROM])) cg.add(var.set_range_to(config[CONF_RANGE_TO]))