mirror of
https://github.com/esphome/esphome.git
synced 2025-03-20 01:28:16 +00:00
81 lines
2.6 KiB
Python
81 lines
2.6 KiB
Python
import voluptuous as vol
|
|
|
|
from esphome import pins
|
|
from esphome.components import sensor
|
|
import esphome.config_validation as cv
|
|
from esphome.const import CONF_ID, CONF_NAME, CONF_PM_10_0, \
|
|
CONF_PM_2_5, CONF_TYPE, CONF_UPDATE_INTERVAL
|
|
from esphome.cpp_generator import Pvariable
|
|
from esphome.cpp_helpers import gpio_input_pin_expression, setup_component
|
|
from esphome.cpp_types import App, Component
|
|
|
|
|
|
PPD42XComponent = sensor.sensor_ns.class_('PPD42XComponent', Component)
|
|
PPD42XSensor = sensor.sensor_ns.class_('PPD42XSensor', sensor.Sensor)
|
|
|
|
CONF_PPD42 = 'PPD42'
|
|
CONF_PPD42NJ = 'PPD42NJ'
|
|
CONF_PPD42NS = 'PPD42NS'
|
|
|
|
CONF_PIN_02_5 = 'pin_02_5'
|
|
CONF_PIN_10_0 = 'pin_10_0'
|
|
|
|
|
|
PPD42XType = sensor.sensor_ns.enum('PPD42XType')
|
|
PPD42X_TYPES = {
|
|
CONF_PPD42: PPD42XType.PPD42X_TYPE,
|
|
CONF_PPD42NJ: PPD42XType.PPD42X_TYPE_NJ,
|
|
CONF_PPD42NS: PPD42XType.PPD42X_TYPE_NS,
|
|
}
|
|
|
|
SENSORS_TO_TYPE = {
|
|
CONF_PM_2_5: [CONF_PPD42, CONF_PPD42NJ, CONF_PPD42NS],
|
|
CONF_PM_10_0: [CONF_PPD42, CONF_PPD42NJ, CONF_PPD42NS],
|
|
}
|
|
|
|
|
|
def validate_ppd42x_sensors(value):
|
|
for key, types in SENSORS_TO_TYPE.items():
|
|
if key in value and value[CONF_TYPE] not in types:
|
|
raise vol.Invalid(u"{} does not have {} sensor!".format(value[CONF_TYPE], key))
|
|
return value
|
|
|
|
|
|
PPD42X_SENSOR_SCHEMA = sensor.SENSOR_SCHEMA.extend({
|
|
cv.GenerateID(): cv.declare_variable_id(PPD42XSensor),
|
|
})
|
|
|
|
PLATFORM_SCHEMA = vol.All(sensor.PLATFORM_SCHEMA.extend({
|
|
cv.GenerateID(): cv.declare_variable_id(PPD42XComponent),
|
|
vol.Required(CONF_TYPE): cv.one_of(*PPD42X_TYPES, upper=True),
|
|
vol.Optional(CONF_PIN_02_5): pins.gpio_input_pin_schema,
|
|
vol.Optional(CONF_PIN_10_0): pins.gpio_input_pin_schema,
|
|
|
|
vol.Optional(CONF_PM_2_5): cv.nameable(PPD42X_SENSOR_SCHEMA),
|
|
vol.Optional(CONF_PM_10_0): cv.nameable(PPD42X_SENSOR_SCHEMA),
|
|
vol.Optional(CONF_UPDATE_INTERVAL): cv.update_interval,
|
|
}).extend(cv.COMPONENT_SCHEMA.schema), cv.has_at_least_one_key(*SENSORS_TO_TYPE))
|
|
|
|
|
|
def to_code(config):
|
|
for pl_02_5_ in gpio_input_pin_expression(config[CONF_PIN_02_5]):
|
|
yield
|
|
for pl_10_0_ in gpio_input_pin_expression(config[CONF_PIN_10_0]):
|
|
yield
|
|
|
|
rhs = App.make_ppd42x(PPD42X_TYPES[config[CONF_TYPE]], config.get(CONF_UPDATE_INTERVAL))
|
|
ppd = Pvariable(config[CONF_ID], rhs)
|
|
|
|
if CONF_PM_2_5 in config:
|
|
conf = config[CONF_PM_2_5]
|
|
sensor.register_sensor(ppd.make_pl_02_5_sensor(conf[CONF_NAME], pl_02_5_.copy()), conf)
|
|
|
|
if CONF_PM_10_0 in config:
|
|
conf = config[CONF_PM_10_0]
|
|
sensor.register_sensor(ppd.make_pl_10_0_sensor(conf[CONF_NAME], pl_10_0_.copy()), conf)
|
|
|
|
setup_component(ppd, config)
|
|
|
|
|
|
BUILD_FLAGS = '-DUSE_PPD42X'
|