120 lines
3.8 KiB
Python
120 lines
3.8 KiB
Python
"""
|
|
A component which allows you to send data to an Influx database.
|
|
|
|
For more details about this component, please refer to the documentation at
|
|
https://home-assistant.io/components/influxdb/
|
|
"""
|
|
import logging
|
|
|
|
import homeassistant.util as util
|
|
from homeassistant.const import (EVENT_STATE_CHANGED, STATE_UNAVAILABLE,
|
|
STATE_UNKNOWN)
|
|
from homeassistant.helpers import state as state_helper
|
|
from homeassistant.helpers import validate_config
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
DOMAIN = "influxdb"
|
|
DEPENDENCIES = []
|
|
|
|
DEFAULT_HOST = 'localhost'
|
|
DEFAULT_PORT = 8086
|
|
DEFAULT_DATABASE = 'home_assistant'
|
|
DEFAULT_SSL = False
|
|
DEFAULT_VERIFY_SSL = False
|
|
|
|
REQUIREMENTS = ['influxdb==3.0.0']
|
|
|
|
CONF_HOST = 'host'
|
|
CONF_PORT = 'port'
|
|
CONF_DB_NAME = 'database'
|
|
CONF_USERNAME = 'username'
|
|
CONF_PASSWORD = 'password'
|
|
CONF_SSL = 'ssl'
|
|
CONF_VERIFY_SSL = 'verify_ssl'
|
|
CONF_BLACKLIST = 'blacklist'
|
|
CONF_WHITELIST = 'whitelist'
|
|
CONF_TAGS = 'tags'
|
|
|
|
|
|
# pylint: disable=too-many-locals
|
|
def setup(hass, config):
|
|
"""Setup the InfluxDB component."""
|
|
from influxdb import InfluxDBClient, exceptions
|
|
|
|
if not validate_config(config, {DOMAIN: ['host',
|
|
CONF_USERNAME,
|
|
CONF_PASSWORD]}, _LOGGER):
|
|
return False
|
|
|
|
conf = config[DOMAIN]
|
|
|
|
host = conf[CONF_HOST]
|
|
port = util.convert(conf.get(CONF_PORT), int, DEFAULT_PORT)
|
|
database = util.convert(conf.get(CONF_DB_NAME), str, DEFAULT_DATABASE)
|
|
username = util.convert(conf.get(CONF_USERNAME), str)
|
|
password = util.convert(conf.get(CONF_PASSWORD), str)
|
|
ssl = util.convert(conf.get(CONF_SSL), bool, DEFAULT_SSL)
|
|
verify_ssl = util.convert(conf.get(CONF_VERIFY_SSL), bool,
|
|
DEFAULT_VERIFY_SSL)
|
|
blacklist = conf.get(CONF_BLACKLIST, [])
|
|
whitelist = conf.get(CONF_WHITELIST, [])
|
|
tags = conf.get(CONF_TAGS, {})
|
|
|
|
try:
|
|
influx = InfluxDBClient(host=host, port=port, username=username,
|
|
password=password, database=database,
|
|
ssl=ssl, verify_ssl=verify_ssl)
|
|
influx.query("select * from /.*/ LIMIT 1;")
|
|
except exceptions.InfluxDBClientError as exc:
|
|
_LOGGER.error("Database host is not accessible due to '%s', please "
|
|
"check your entries in the configuration file and that "
|
|
"the database exists and is READ/WRITE.", exc)
|
|
return False
|
|
|
|
def influx_event_listener(event):
|
|
"""Listen for new messages on the bus and sends them to Influx."""
|
|
state = event.data.get('new_state')
|
|
if state is None or state.state in (
|
|
STATE_UNKNOWN, '', STATE_UNAVAILABLE) or \
|
|
state.entity_id in blacklist:
|
|
return
|
|
|
|
try:
|
|
if len(whitelist) > 0 and state.entity_id not in whitelist:
|
|
return
|
|
|
|
_state = state_helper.state_as_number(state)
|
|
except ValueError:
|
|
_state = state.state
|
|
|
|
measurement = state.attributes.get('unit_of_measurement')
|
|
if measurement in (None, ''):
|
|
measurement = state.entity_id
|
|
|
|
json_body = [
|
|
{
|
|
'measurement': measurement,
|
|
'tags': {
|
|
'domain': state.domain,
|
|
'entity_id': state.object_id,
|
|
},
|
|
'time': event.time_fired,
|
|
'fields': {
|
|
'value': _state,
|
|
}
|
|
}
|
|
]
|
|
|
|
for tag in tags:
|
|
json_body[0]['tags'][tag] = tags[tag]
|
|
|
|
try:
|
|
influx.write_points(json_body)
|
|
except exceptions.InfluxDBClientError:
|
|
_LOGGER.exception('Error saving event "%s" to InfluxDB', json_body)
|
|
|
|
hass.bus.listen(EVENT_STATE_CHANGED, influx_event_listener)
|
|
|
|
return True
|