esphome/esphomeyaml/components/sensor/ads1115.py

77 lines
2.4 KiB
Python
Raw Normal View History

2018-04-07 01:23:03 +02:00
import voluptuous as vol
import esphomeyaml.config_validation as cv
from esphomeyaml.components import sensor
2018-05-20 12:41:52 +02:00
from esphomeyaml.components.ads1115 import ADS1115Component
from esphomeyaml.const import CONF_ADS1115_ID, CONF_GAIN, CONF_MULTIPLEXER, CONF_NAME, \
CONF_UPDATE_INTERVAL
from esphomeyaml.helpers import get_variable
2018-04-07 01:23:03 +02:00
DEPENDENCIES = ['ads1115']
ADS1115Multiplexer = sensor.sensor_ns.enum('ADS1115Multiplexer')
2018-04-07 01:23:03 +02:00
MUX = {
'A0_A1': ADS1115Multiplexer.ADS1115_MULTIPLEXER_P0_N1,
'A0_A3': ADS1115Multiplexer.ADS1115_MULTIPLEXER_P0_N3,
'A1_A3': ADS1115Multiplexer.ADS1115_MULTIPLEXER_P1_N3,
'A2_A3': ADS1115Multiplexer.ADS1115_MULTIPLEXER_P2_N3,
'A0_GND': ADS1115Multiplexer.ADS1115_MULTIPLEXER_P0_NG,
'A1_GND': ADS1115Multiplexer.ADS1115_MULTIPLEXER_P1_NG,
'A2_GND': ADS1115Multiplexer.ADS1115_MULTIPLEXER_P2_NG,
'A3_GND': ADS1115Multiplexer.ADS1115_MULTIPLEXER_P3_NG,
2018-04-07 01:23:03 +02:00
}
ADS1115Gain = sensor.sensor_ns.enum('ADS1115Gain')
2018-04-07 01:23:03 +02:00
GAIN = {
'6.144': ADS1115Gain.ADS1115_GAIN_6P144,
'4.096': ADS1115Gain.ADS1115_GAIN_6P096,
'2.048': ADS1115Gain.ADS1115_GAIN_2P048,
'1.024': ADS1115Gain.ADS1115_GAIN_1P024,
'0.512': ADS1115Gain.ADS1115_GAIN_0P512,
'0.256': ADS1115Gain.ADS1115_GAIN_0P256,
2018-04-07 01:23:03 +02:00
}
def validate_gain(value):
if isinstance(value, float):
value = u'{:0.03f}'.format(value)
elif not isinstance(value, (str, unicode)):
raise vol.Invalid('invalid gain "{}"'.format(value))
2018-05-27 14:15:24 +02:00
return cv.one_of(*GAIN)(value)
def validate_mux(value):
value = cv.string(value).upper()
value = value.replace(' ', '_')
return cv.one_of(*MUX)(value)
2018-04-07 01:23:03 +02:00
ADS1115Sensor = sensor.sensor_ns.class_('ADS1115Sensor', sensor.EmptySensor)
PLATFORM_SCHEMA = cv.nameable(sensor.SENSOR_PLATFORM_SCHEMA.extend({
cv.GenerateID(): cv.declare_variable_id(ADS1115Sensor),
2018-05-27 14:15:24 +02:00
vol.Required(CONF_MULTIPLEXER): validate_mux,
2018-04-07 01:23:03 +02:00
vol.Required(CONF_GAIN): validate_gain,
2018-06-02 22:22:20 +02:00
cv.GenerateID(CONF_ADS1115_ID): cv.use_variable_id(ADS1115Component),
2018-08-24 22:44:15 +02:00
vol.Optional(CONF_UPDATE_INTERVAL): cv.update_interval,
}))
2018-04-07 01:23:03 +02:00
def to_code(config):
2018-06-02 22:22:20 +02:00
hub = None
for hub in get_variable(config[CONF_ADS1115_ID]):
yield
2018-04-07 01:23:03 +02:00
2018-05-20 12:41:52 +02:00
mux = MUX[config[CONF_MULTIPLEXER]]
gain = GAIN[config[CONF_GAIN]]
rhs = hub.get_sensor(config[CONF_NAME], mux, gain, config.get(CONF_UPDATE_INTERVAL))
2018-06-03 07:11:11 +02:00
sensor.register_sensor(rhs, config)
2018-05-06 15:56:12 +02:00
BUILD_FLAGS = '-DUSE_ADS1115_SENSOR'
def to_hass_config(data, config):
return sensor.core_to_hass_config(data, config)