mirror of https://github.com/esphome/esphome.git
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
import esphome.codegen as cg
|
|
import esphome.config_validation as cv
|
|
from esphome.components import sensor
|
|
from esphome.const import CONF_TYPE, CONF_UNIT_OF_MEASUREMENT, CONF_ACCURACY_DECIMALS, CONF_ICON, \
|
|
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 = cv.nameable(sensor.SENSOR_SCHEMA.extend({
|
|
cv.Required(CONF_TYPE): cv.one_of(*TYPES, upper=True),
|
|
cv.GenerateID(CONF_APDS9960_ID): cv.use_variable_id(APDS9960),
|
|
|
|
cv.Optional(CONF_UNIT_OF_MEASUREMENT, default=UNIT_PERCENT): sensor.unit_of_measurement,
|
|
cv.Optional(CONF_ACCURACY_DECIMALS, default=1): sensor.accuracy_decimals,
|
|
cv.Optional(CONF_ICON, default=ICON_LIGHTBULB): sensor.icon,
|
|
}))
|
|
|
|
|
|
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))
|