mirror of https://github.com/esphome/esphome.git
28 lines
814 B
Python
28 lines
814 B
Python
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome.components import sensor
|
|
from esphome.const import CONF_TYPE, 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).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))
|