mirror of
https://github.com/home-assistant/core
synced 2024-08-28 03:36:46 +02:00
128 lines
3.6 KiB
Python
128 lines
3.6 KiB
Python
"""Support for Kaiterra Air Quality Sensors."""
|
|
from __future__ import annotations
|
|
|
|
from homeassistant.components.air_quality import AirQualityEntity
|
|
from homeassistant.const import CONF_DEVICE_ID, CONF_NAME
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers.dispatcher import async_dispatcher_connect
|
|
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
|
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
|
|
|
|
from .const import (
|
|
ATTR_AQI_LEVEL,
|
|
ATTR_AQI_POLLUTANT,
|
|
ATTR_VOC,
|
|
DISPATCHER_KAITERRA,
|
|
DOMAIN,
|
|
)
|
|
|
|
|
|
async def async_setup_platform(
|
|
hass: HomeAssistant,
|
|
config: ConfigType,
|
|
async_add_entities: AddEntitiesCallback,
|
|
discovery_info: DiscoveryInfoType | None = None,
|
|
) -> None:
|
|
"""Set up the air_quality kaiterra sensor."""
|
|
if discovery_info is None:
|
|
return
|
|
|
|
api = hass.data[DOMAIN]
|
|
name = discovery_info[CONF_NAME]
|
|
device_id = discovery_info[CONF_DEVICE_ID]
|
|
|
|
async_add_entities([KaiterraAirQuality(api, name, device_id)])
|
|
|
|
|
|
class KaiterraAirQuality(AirQualityEntity):
|
|
"""Implementation of a Kaittera air quality sensor."""
|
|
|
|
_attr_should_poll = False
|
|
|
|
def __init__(self, api, name, device_id):
|
|
"""Initialize the sensor."""
|
|
self._api = api
|
|
self._name = f"{name} Air Quality"
|
|
self._device_id = device_id
|
|
|
|
def _data(self, key):
|
|
return self._device.get(key, {}).get("value")
|
|
|
|
@property
|
|
def _device(self):
|
|
return self._api.data.get(self._device_id, {})
|
|
|
|
@property
|
|
def available(self):
|
|
"""Return the availability of the sensor."""
|
|
return self._api.data.get(self._device_id) is not None
|
|
|
|
@property
|
|
def name(self):
|
|
"""Return the name of the sensor."""
|
|
return self._name
|
|
|
|
@property
|
|
def air_quality_index(self):
|
|
"""Return the Air Quality Index (AQI)."""
|
|
return self._data("aqi")
|
|
|
|
@property
|
|
def air_quality_index_level(self):
|
|
"""Return the Air Quality Index level."""
|
|
return self._data("aqi_level")
|
|
|
|
@property
|
|
def air_quality_index_pollutant(self):
|
|
"""Return the Air Quality Index level."""
|
|
return self._data("aqi_pollutant")
|
|
|
|
@property
|
|
def particulate_matter_2_5(self):
|
|
"""Return the particulate matter 2.5 level."""
|
|
return self._data("rpm25c")
|
|
|
|
@property
|
|
def particulate_matter_10(self):
|
|
"""Return the particulate matter 10 level."""
|
|
return self._data("rpm10c")
|
|
|
|
@property
|
|
def carbon_dioxide(self):
|
|
"""Return the CO2 (carbon dioxide) level."""
|
|
return self._data("rco2")
|
|
|
|
@property
|
|
def volatile_organic_compounds(self):
|
|
"""Return the VOC (Volatile Organic Compounds) level."""
|
|
return self._data("rtvoc")
|
|
|
|
@property
|
|
def unique_id(self):
|
|
"""Return the sensor's unique id."""
|
|
return f"{self._device_id}_air_quality"
|
|
|
|
@property
|
|
def extra_state_attributes(self):
|
|
"""Return the device state attributes."""
|
|
data = {}
|
|
attributes = [
|
|
(ATTR_VOC, self.volatile_organic_compounds),
|
|
(ATTR_AQI_LEVEL, self.air_quality_index_level),
|
|
(ATTR_AQI_POLLUTANT, self.air_quality_index_pollutant),
|
|
]
|
|
|
|
for attr, value in attributes:
|
|
if value is not None:
|
|
data[attr] = value
|
|
|
|
return data
|
|
|
|
async def async_added_to_hass(self):
|
|
"""Register callback."""
|
|
self.async_on_remove(
|
|
async_dispatcher_connect(
|
|
self.hass, DISPATCHER_KAITERRA, self.async_write_ha_state
|
|
)
|
|
)
|