Black
This commit is contained in:
Paulus Schoutsen
parent
da05dfe708
commit
4de97abc3a
2676 changed files with 163166 additions and 140084 deletions
|
|
@ -7,8 +7,12 @@ import math
|
|||
import voluptuous as vol
|
||||
|
||||
from homeassistant.const import (
|
||||
CONF_ACCESS_TOKEN, CONF_DEVICES, DEVICE_CLASS_HUMIDITY,
|
||||
DEVICE_CLASS_TEMPERATURE, TEMP_CELSIUS)
|
||||
CONF_ACCESS_TOKEN,
|
||||
CONF_DEVICES,
|
||||
DEVICE_CLASS_HUMIDITY,
|
||||
DEVICE_CLASS_TEMPERATURE,
|
||||
TEMP_CELSIUS,
|
||||
)
|
||||
from homeassistant.exceptions import PlatformNotReady
|
||||
from homeassistant.helpers.aiohttp_client import async_get_clientsession
|
||||
import homeassistant.helpers.config_validation as cv
|
||||
|
|
@ -17,48 +21,64 @@ from homeassistant.util import Throttle, dt
|
|||
|
||||
_LOGGER = logging.getLogger(__name__)
|
||||
|
||||
ATTR_SCORE = 'score'
|
||||
ATTR_TIMESTAMP = 'timestamp'
|
||||
ATTR_LAST_API_UPDATE = 'last_api_update'
|
||||
ATTR_COMPONENT = 'component'
|
||||
ATTR_VALUE = 'value'
|
||||
ATTR_SENSORS = 'sensors'
|
||||
ATTR_SCORE = "score"
|
||||
ATTR_TIMESTAMP = "timestamp"
|
||||
ATTR_LAST_API_UPDATE = "last_api_update"
|
||||
ATTR_COMPONENT = "component"
|
||||
ATTR_VALUE = "value"
|
||||
ATTR_SENSORS = "sensors"
|
||||
|
||||
CONF_UUID = 'uuid'
|
||||
CONF_UUID = "uuid"
|
||||
|
||||
DEVICE_CLASS_PM2_5 = 'PM2.5'
|
||||
DEVICE_CLASS_PM10 = 'PM10'
|
||||
DEVICE_CLASS_CARBON_DIOXIDE = 'CO2'
|
||||
DEVICE_CLASS_VOLATILE_ORGANIC_COMPOUNDS = 'VOC'
|
||||
DEVICE_CLASS_SCORE = 'score'
|
||||
DEVICE_CLASS_PM2_5 = "PM2.5"
|
||||
DEVICE_CLASS_PM10 = "PM10"
|
||||
DEVICE_CLASS_CARBON_DIOXIDE = "CO2"
|
||||
DEVICE_CLASS_VOLATILE_ORGANIC_COMPOUNDS = "VOC"
|
||||
DEVICE_CLASS_SCORE = "score"
|
||||
|
||||
SENSOR_TYPES = {
|
||||
'TEMP': {'device_class': DEVICE_CLASS_TEMPERATURE,
|
||||
'unit_of_measurement': TEMP_CELSIUS,
|
||||
'icon': 'mdi:thermometer'},
|
||||
'HUMID': {'device_class': DEVICE_CLASS_HUMIDITY,
|
||||
'unit_of_measurement': '%',
|
||||
'icon': 'mdi:water-percent'},
|
||||
'CO2': {'device_class': DEVICE_CLASS_CARBON_DIOXIDE,
|
||||
'unit_of_measurement': 'ppm',
|
||||
'icon': 'mdi:periodic-table-co2'},
|
||||
'VOC': {'device_class': DEVICE_CLASS_VOLATILE_ORGANIC_COMPOUNDS,
|
||||
'unit_of_measurement': 'ppb',
|
||||
'icon': 'mdi:cloud'},
|
||||
"TEMP": {
|
||||
"device_class": DEVICE_CLASS_TEMPERATURE,
|
||||
"unit_of_measurement": TEMP_CELSIUS,
|
||||
"icon": "mdi:thermometer",
|
||||
},
|
||||
"HUMID": {
|
||||
"device_class": DEVICE_CLASS_HUMIDITY,
|
||||
"unit_of_measurement": "%",
|
||||
"icon": "mdi:water-percent",
|
||||
},
|
||||
"CO2": {
|
||||
"device_class": DEVICE_CLASS_CARBON_DIOXIDE,
|
||||
"unit_of_measurement": "ppm",
|
||||
"icon": "mdi:periodic-table-co2",
|
||||
},
|
||||
"VOC": {
|
||||
"device_class": DEVICE_CLASS_VOLATILE_ORGANIC_COMPOUNDS,
|
||||
"unit_of_measurement": "ppb",
|
||||
"icon": "mdi:cloud",
|
||||
},
|
||||
# Awair docs don't actually specify the size they measure for 'dust',
|
||||
# but 2.5 allows the sensor to show up in HomeKit
|
||||
'DUST': {'device_class': DEVICE_CLASS_PM2_5,
|
||||
'unit_of_measurement': 'µg/m3',
|
||||
'icon': 'mdi:cloud'},
|
||||
'PM25': {'device_class': DEVICE_CLASS_PM2_5,
|
||||
'unit_of_measurement': 'µg/m3',
|
||||
'icon': 'mdi:cloud'},
|
||||
'PM10': {'device_class': DEVICE_CLASS_PM10,
|
||||
'unit_of_measurement': 'µg/m3',
|
||||
'icon': 'mdi:cloud'},
|
||||
'score': {'device_class': DEVICE_CLASS_SCORE,
|
||||
'unit_of_measurement': '%',
|
||||
'icon': 'mdi:percent'},
|
||||
"DUST": {
|
||||
"device_class": DEVICE_CLASS_PM2_5,
|
||||
"unit_of_measurement": "µg/m3",
|
||||
"icon": "mdi:cloud",
|
||||
},
|
||||
"PM25": {
|
||||
"device_class": DEVICE_CLASS_PM2_5,
|
||||
"unit_of_measurement": "µg/m3",
|
||||
"icon": "mdi:cloud",
|
||||
},
|
||||
"PM10": {
|
||||
"device_class": DEVICE_CLASS_PM10,
|
||||
"unit_of_measurement": "µg/m3",
|
||||
"icon": "mdi:cloud",
|
||||
},
|
||||
"score": {
|
||||
"device_class": DEVICE_CLASS_SCORE,
|
||||
"unit_of_measurement": "%",
|
||||
"icon": "mdi:percent",
|
||||
},
|
||||
}
|
||||
|
||||
AWAIR_QUOTA = 300
|
||||
|
|
@ -67,15 +87,14 @@ AWAIR_QUOTA = 300
|
|||
# Don't bother asking us for state more often than that.
|
||||
SCAN_INTERVAL = timedelta(minutes=5)
|
||||
|
||||
AWAIR_DEVICE_SCHEMA = vol.Schema({
|
||||
vol.Required(CONF_UUID): cv.string,
|
||||
})
|
||||
AWAIR_DEVICE_SCHEMA = vol.Schema({vol.Required(CONF_UUID): cv.string})
|
||||
|
||||
PLATFORM_SCHEMA = cv.PLATFORM_SCHEMA.extend({
|
||||
vol.Required(CONF_ACCESS_TOKEN): cv.string,
|
||||
vol.Optional(CONF_DEVICES): vol.All(
|
||||
cv.ensure_list, [AWAIR_DEVICE_SCHEMA]),
|
||||
})
|
||||
PLATFORM_SCHEMA = cv.PLATFORM_SCHEMA.extend(
|
||||
{
|
||||
vol.Required(CONF_ACCESS_TOKEN): cv.string,
|
||||
vol.Optional(CONF_DEVICES): vol.All(cv.ensure_list, [AWAIR_DEVICE_SCHEMA]),
|
||||
}
|
||||
)
|
||||
|
||||
|
||||
# Awair *heavily* throttles calls that get user information,
|
||||
|
|
@ -84,8 +103,7 @@ PLATFORM_SCHEMA = cv.PLATFORM_SCHEMA.extend({
|
|||
# list of devices, and they may provide the same set of information
|
||||
# that the devices() call would return. However, the only thing
|
||||
# used at this time is the `uuid` value.
|
||||
async def async_setup_platform(hass, config, async_add_entities,
|
||||
discovery_info=None):
|
||||
async def async_setup_platform(hass, config, async_add_entities, discovery_info=None):
|
||||
"""Connect to the Awair API and find devices."""
|
||||
from python_awair import AwairClient
|
||||
|
||||
|
|
@ -106,8 +124,7 @@ async def async_setup_platform(hass, config, async_add_entities,
|
|||
await awair_data.async_update()
|
||||
for sensor in SENSOR_TYPES:
|
||||
if sensor in awair_data.data:
|
||||
awair_sensor = AwairSensor(awair_data, device,
|
||||
sensor, throttle)
|
||||
awair_sensor = AwairSensor(awair_data, device, sensor, throttle)
|
||||
all_devices.append(awair_sensor)
|
||||
|
||||
async_add_entities(all_devices, True)
|
||||
|
|
@ -116,8 +133,11 @@ async def async_setup_platform(hass, config, async_add_entities,
|
|||
_LOGGER.error("Awair API access_token invalid")
|
||||
except AwairClient.RatelimitError:
|
||||
_LOGGER.error("Awair API ratelimit exceeded.")
|
||||
except (AwairClient.QueryError, AwairClient.NotFoundError,
|
||||
AwairClient.GenericError) as error:
|
||||
except (
|
||||
AwairClient.QueryError,
|
||||
AwairClient.NotFoundError,
|
||||
AwairClient.GenericError,
|
||||
) as error:
|
||||
_LOGGER.error("Unexpected Awair API error: %s", error)
|
||||
|
||||
raise PlatformNotReady
|
||||
|
|
@ -129,9 +149,9 @@ class AwairSensor(Entity):
|
|||
def __init__(self, data, device, sensor_type, throttle):
|
||||
"""Initialize the sensor."""
|
||||
self._uuid = device[CONF_UUID]
|
||||
self._device_class = SENSOR_TYPES[sensor_type]['device_class']
|
||||
self._name = 'Awair {}'.format(self._device_class)
|
||||
unit = SENSOR_TYPES[sensor_type]['unit_of_measurement']
|
||||
self._device_class = SENSOR_TYPES[sensor_type]["device_class"]
|
||||
self._name = "Awair {}".format(self._device_class)
|
||||
unit = SENSOR_TYPES[sensor_type]["unit_of_measurement"]
|
||||
self._unit_of_measurement = unit
|
||||
self._data = data
|
||||
self._type = sensor_type
|
||||
|
|
@ -150,7 +170,7 @@ class AwairSensor(Entity):
|
|||
@property
|
||||
def icon(self):
|
||||
"""Icon to use in the frontend."""
|
||||
return SENSOR_TYPES[self._type]['icon']
|
||||
return SENSOR_TYPES[self._type]["icon"]
|
||||
|
||||
@property
|
||||
def state(self):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue