ha-core/homeassistant/components/envirophat/sensor.py

195 lines
6.6 KiB
Python

"""Support for Enviro pHAT sensors."""
import importlib
import logging
from datetime import timedelta
import voluptuous as vol
from homeassistant.components.sensor import PLATFORM_SCHEMA
from homeassistant.const import TEMP_CELSIUS, CONF_DISPLAY_OPTIONS, CONF_NAME
import homeassistant.helpers.config_validation as cv
from homeassistant.helpers.entity import Entity
from homeassistant.util import Throttle
_LOGGER = logging.getLogger(__name__)
DEFAULT_NAME = "envirophat"
CONF_USE_LEDS = "use_leds"
MIN_TIME_BETWEEN_UPDATES = timedelta(seconds=60)
SENSOR_TYPES = {
"light": ["light", " ", "mdi:weather-sunny"],
"light_red": ["light_red", " ", "mdi:invert-colors"],
"light_green": ["light_green", " ", "mdi:invert-colors"],
"light_blue": ["light_blue", " ", "mdi:invert-colors"],
"accelerometer_x": ["accelerometer_x", "G", "mdi:earth"],
"accelerometer_y": ["accelerometer_y", "G", "mdi:earth"],
"accelerometer_z": ["accelerometer_z", "G", "mdi:earth"],
"magnetometer_x": ["magnetometer_x", " ", "mdi:magnet"],
"magnetometer_y": ["magnetometer_y", " ", "mdi:magnet"],
"magnetometer_z": ["magnetometer_z", " ", "mdi:magnet"],
"temperature": ["temperature", TEMP_CELSIUS, "mdi:thermometer"],
"pressure": ["pressure", "hPa", "mdi:gauge"],
"voltage_0": ["voltage_0", "V", "mdi:flash"],
"voltage_1": ["voltage_1", "V", "mdi:flash"],
"voltage_2": ["voltage_2", "V", "mdi:flash"],
"voltage_3": ["voltage_3", "V", "mdi:flash"],
}
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
{
vol.Required(CONF_DISPLAY_OPTIONS, default=list(SENSOR_TYPES)): [
vol.In(SENSOR_TYPES)
],
vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string,
vol.Optional(CONF_USE_LEDS, default=False): cv.boolean,
}
)
def setup_platform(hass, config, add_entities, discovery_info=None):
"""Set up the Sense HAT sensor platform."""
try:
envirophat = importlib.import_module("envirophat")
except OSError:
_LOGGER.error("No Enviro pHAT was found.")
return False
data = EnvirophatData(envirophat, config.get(CONF_USE_LEDS))
dev = []
for variable in config[CONF_DISPLAY_OPTIONS]:
dev.append(EnvirophatSensor(data, variable))
add_entities(dev, True)
class EnvirophatSensor(Entity):
"""Representation of an Enviro pHAT sensor."""
def __init__(self, data, sensor_types):
"""Initialize the sensor."""
self.data = data
self._name = SENSOR_TYPES[sensor_types][0]
self._unit_of_measurement = SENSOR_TYPES[sensor_types][1]
self.type = sensor_types
self._state = None
@property
def name(self):
"""Return the name of the sensor."""
return self._name
@property
def state(self):
"""Return the state of the sensor."""
return self._state
@property
def icon(self):
"""Icon to use in the frontend, if any."""
return SENSOR_TYPES[self.type][2]
@property
def unit_of_measurement(self):
"""Return the unit the value is expressed in."""
return self._unit_of_measurement
def update(self):
"""Get the latest data and updates the states."""
self.data.update()
if self.type == "light":
self._state = self.data.light
if self.type == "light_red":
self._state = self.data.light_red
if self.type == "light_green":
self._state = self.data.light_green
if self.type == "light_blue":
self._state = self.data.light_blue
if self.type == "accelerometer_x":
self._state = self.data.accelerometer_x
if self.type == "accelerometer_y":
self._state = self.data.accelerometer_y
if self.type == "accelerometer_z":
self._state = self.data.accelerometer_z
if self.type == "magnetometer_x":
self._state = self.data.magnetometer_x
if self.type == "magnetometer_y":
self._state = self.data.magnetometer_y
if self.type == "magnetometer_z":
self._state = self.data.magnetometer_z
if self.type == "temperature":
self._state = self.data.temperature
if self.type == "pressure":
self._state = self.data.pressure
if self.type == "voltage_0":
self._state = self.data.voltage_0
if self.type == "voltage_1":
self._state = self.data.voltage_1
if self.type == "voltage_2":
self._state = self.data.voltage_2
if self.type == "voltage_3":
self._state = self.data.voltage_3
class EnvirophatData:
"""Get the latest data and update."""
def __init__(self, envirophat, use_leds):
"""Initialize the data object."""
self.envirophat = envirophat
self.use_leds = use_leds
# sensors readings
self.light = None
self.light_red = None
self.light_green = None
self.light_blue = None
self.accelerometer_x = None
self.accelerometer_y = None
self.accelerometer_z = None
self.magnetometer_x = None
self.magnetometer_y = None
self.magnetometer_z = None
self.temperature = None
self.pressure = None
self.voltage_0 = None
self.voltage_1 = None
self.voltage_2 = None
self.voltage_3 = None
@Throttle(MIN_TIME_BETWEEN_UPDATES)
def update(self):
"""Get the latest data from Enviro pHAT."""
# Light sensor reading: 16-bit integer
self.light = self.envirophat.light.light()
if self.use_leds:
self.envirophat.leds.on()
# the three color values scaled against the overall light, 0-255
self.light_red, self.light_green, self.light_blue = self.envirophat.light.rgb()
if self.use_leds:
self.envirophat.leds.off()
# accelerometer readings in G
self.accelerometer_x, self.accelerometer_y, self.accelerometer_z = (
self.envirophat.motion.accelerometer()
)
# raw magnetometer reading
self.magnetometer_x, self.magnetometer_y, self.magnetometer_z = (
self.envirophat.motion.magnetometer()
)
# temperature resolution of BMP280 sensor: 0.01°C
self.temperature = round(self.envirophat.weather.temperature(), 2)
# pressure resolution of BMP280 sensor: 0.16 Pa, rounding to 0.1 Pa
# with conversion to 100 Pa = 1 hPa
self.pressure = round(self.envirophat.weather.pressure() / 100.0, 3)
# Voltage sensor, reading between 0-3.3V
self.voltage_0, self.voltage_1, self.voltage_2, self.voltage_3 = (
self.envirophat.analog.read_all()
)