1d70005b01
fix check for already created entities remove hepa and carbon filter add AQI attribute initial commit fix check for already created entities remove hepa and carbon filter add AQI attribute add air quality component tests fix method call tests fix line lengths fix pylint issues fix docstrings revert fan related changes remove whitespace change fix fan update state test add for loop for platform initialization add requested changes to aiq platform change string concatenation to new style string formatting update air quality tests update air quality tests refactor sensor component changes fix pylint issues fix debug string in the air quality component replace failing tests for older devices fix line length fan tests remove dependencies const and move imports move back imports to methods remove whitespace from blank line
126 lines
4 KiB
Python
126 lines
4 KiB
Python
"""Support for Dyson Pure Cool Air Quality Sensors."""
|
|
import logging
|
|
|
|
from homeassistant.components.air_quality import AirQualityEntity, DOMAIN
|
|
from . import DYSON_DEVICES
|
|
|
|
ATTRIBUTION = 'Dyson purifier air quality sensor'
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
DYSON_AIQ_DEVICES = 'dyson_aiq_devices'
|
|
|
|
ATTR_VOC = 'volatile_organic_compounds'
|
|
|
|
|
|
def setup_platform(hass, config, add_entities, discovery_info=None):
|
|
"""Set up the Dyson Sensors."""
|
|
from libpurecool.dyson_pure_cool import DysonPureCool
|
|
|
|
if discovery_info is None:
|
|
return
|
|
|
|
hass.data.setdefault(DYSON_AIQ_DEVICES, [])
|
|
|
|
# Get Dyson Devices from parent component
|
|
device_ids = [device.unique_id for device in hass.data[DYSON_AIQ_DEVICES]]
|
|
for device in hass.data[DYSON_DEVICES]:
|
|
if isinstance(device, DysonPureCool) and \
|
|
device.serial not in device_ids:
|
|
hass.data[DYSON_AIQ_DEVICES].append(DysonAirSensor(device))
|
|
add_entities(hass.data[DYSON_AIQ_DEVICES])
|
|
|
|
|
|
class DysonAirSensor(AirQualityEntity):
|
|
"""Representation of a generic Dyson air quality sensor."""
|
|
|
|
def __init__(self, device):
|
|
"""Create a new generic air quality Dyson sensor."""
|
|
self._device = device
|
|
self._old_value = None
|
|
self._name = device.name
|
|
|
|
async def async_added_to_hass(self):
|
|
"""Call when entity is added to hass."""
|
|
self.hass.async_add_executor_job(
|
|
self._device.add_message_listener, self.on_message)
|
|
|
|
def on_message(self, message):
|
|
"""Handle new messages which are received from the fan."""
|
|
from libpurecool.dyson_pure_state_v2 import \
|
|
DysonEnvironmentalSensorV2State
|
|
|
|
_LOGGER.debug('%s: Message received for %s device: %s',
|
|
DOMAIN, self.name, message)
|
|
if (self._old_value is None or
|
|
self._old_value != self._device.environmental_state) and \
|
|
isinstance(message, DysonEnvironmentalSensorV2State):
|
|
self._old_value = self._device.environmental_state
|
|
self.schedule_update_ha_state()
|
|
|
|
@property
|
|
def should_poll(self):
|
|
"""No polling needed."""
|
|
return False
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the Dyson sensor."""
|
|
return self._name
|
|
|
|
@property
|
|
def attribution(self):
|
|
"""Return the attribution."""
|
|
return ATTRIBUTION
|
|
|
|
@property
|
|
def air_quality_index(self):
|
|
"""Return the Air Quality Index (AQI)."""
|
|
return max(self.particulate_matter_2_5,
|
|
self.particulate_matter_10,
|
|
self.nitrogen_dioxide,
|
|
self.volatile_organic_compounds)
|
|
|
|
@property
|
|
def particulate_matter_2_5(self):
|
|
"""Return the particulate matter 2.5 level."""
|
|
if self._device.environmental_state:
|
|
return int(self._device.environmental_state.particulate_matter_25)
|
|
return None
|
|
|
|
@property
|
|
def particulate_matter_10(self):
|
|
"""Return the particulate matter 10 level."""
|
|
if self._device.environmental_state:
|
|
return int(self._device.environmental_state.particulate_matter_10)
|
|
return None
|
|
|
|
@property
|
|
def nitrogen_dioxide(self):
|
|
"""Return the NO2 (nitrogen dioxide) level."""
|
|
if self._device.environmental_state:
|
|
return int(self._device.environmental_state.nitrogen_dioxide)
|
|
return None
|
|
|
|
@property
|
|
def volatile_organic_compounds(self):
|
|
"""Return the VOC (Volatile Organic Compounds) level."""
|
|
if self._device.environmental_state:
|
|
return int(self._device.
|
|
environmental_state.volatile_organic_compounds)
|
|
return None
|
|
|
|
@property
|
|
def unique_id(self):
|
|
"""Return the sensor's unique id."""
|
|
return self._device.serial
|
|
|
|
@property
|
|
def device_state_attributes(self):
|
|
"""Return the device state attributes."""
|
|
data = {}
|
|
|
|
voc = self.volatile_organic_compounds
|
|
if voc is not None:
|
|
data[ATTR_VOC] = voc
|
|
return data
|