mirror of https://github.com/esphome/esphome.git
32 lines
878 B
Python
32 lines
878 B
Python
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome.components import sensor
|
|
from esphome.const import CONF_TYPE, DEVICE_CLASS_EMPTY, UNIT_PERCENT, ICON_LIGHTBULB
|
|
from . import APDS9960, CONF_APDS9960_ID
|
|
|
|
DEPENDENCIES = ["apds9960"]
|
|
|
|
TYPES = {
|
|
"CLEAR": "set_clear_channel",
|
|
"RED": "set_red_channel",
|
|
"GREEN": "set_green_channel",
|
|
"BLUE": "set_blue_channel",
|
|
"PROXIMITY": "set_proximity",
|
|
}
|
|
|
|
CONFIG_SCHEMA = sensor.sensor_schema(
|
|
UNIT_PERCENT, ICON_LIGHTBULB, 1, DEVICE_CLASS_EMPTY
|
|
).extend(
|
|
{
|
|
cv.Required(CONF_TYPE): cv.one_of(*TYPES, upper=True),
|
|
cv.GenerateID(CONF_APDS9960_ID): cv.use_id(APDS9960),
|
|
}
|
|
)
|
|
|
|
|
|
def to_code(config):
|
|
hub = yield cg.get_variable(config[CONF_APDS9960_ID])
|
|
var = yield sensor.new_sensor(config)
|
|
func = getattr(hub, TYPES[config[CONF_TYPE]])
|
|
cg.add(func(var))
|