90 lines
2.4 KiB
Python
90 lines
2.4 KiB
Python
|
"""
|
||
|
|
homeassistant.components.statsd
|
||
|
|
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
|
||
|
|
StatsD component which allows you to send data to many backends.
|
||
|
|
|
||
|
|
For list supported backends visit this link:
|
||
|
|
https://github.com/etsy/statsd/blob/master/docs/backend.md
|
||
|
|
|
||
|
|
For more details about this component, please refer to the documentation at
|
||
|
|
https://home-assistant.io/components/statsd/
|
||
|
|
"""
|
||
|
|
import logging
|
||
|
|
import homeassistant.util as util
|
||
|
|
from homeassistant.const import (EVENT_STATE_CHANGED, STATE_ON, STATE_OFF,
|
||
|
|
STATE_UNLOCKED, STATE_LOCKED, STATE_UNKNOWN)
|
||
|
|
from homeassistant.components.sun import (STATE_ABOVE_HORIZON,
|
||
|
|
STATE_BELOW_HORIZON)
|
||
|
|
|
||
|
|
_LOGGER = logging.getLogger(__name__)
|
||
|
|
|
||
|
|
DOMAIN = "statsd"
|
||
|
|
DEPENDENCIES = []
|
||
|
|
|
||
|
|
DEFAULT_HOST = 'localhost'
|
||
|
|
DEFAULT_PORT = 8125
|
||
|
|
DEFAULT_PREFIX = 'hass'
|
||
|
|
DEFAULT_RATE = 1
|
||
|
|
|
||
|
|
REQUIREMENTS = ['python-statsd==1.7.2']
|
||
|
|
|
||
|
|
CONF_HOST = 'host'
|
||
|
|
CONF_PORT = 'port'
|
||
|
|
CONF_PREFIX = 'prefix'
|
||
|
|
CONF_RATE = 'rate'
|
||
|
|
|
||
|
|
|
||
|
|
def setup(hass, config):
|
||
|
|
""" Setup the StatsD component. """
|
||
|
|
|
||
|
|
from statsd.compat import NUM_TYPES
|
||
|
|
import statsd
|
||
|
|
|
||
|
|
conf = config[DOMAIN]
|
||
|
|
|
||
|
|
host = conf[CONF_HOST]
|
||
|
|
port = util.convert(conf.get(CONF_PORT), int, DEFAULT_PORT)
|
||
|
|
sample_rate = util.convert(conf.get(CONF_RATE), int, DEFAULT_RATE)
|
||
|
|
prefix = util.convert(conf.get(CONF_PREFIX), str, DEFAULT_PREFIX)
|
||
|
|
|
||
|
|
statsd_connection = statsd.Connection(
|
||
|
|
host=host,
|
||
|
|
port=port,
|
||
|
|
sample_rate=sample_rate,
|
||
|
|
disabled=False
|
||
|
|
)
|
||
|
|
|
||
|
|
meter = statsd.Gauge(prefix, statsd_connection)
|
||
|
|
|
||
|
|
def statsd_event_listener(event):
|
||
|
|
""" Listen for new messages on the bus and sends them to StatsD. """
|
||
|
|
|
||
|
|
state = event.data.get('new_state')
|
||
|
|
|
||
|
|
if state is None:
|
||
|
|
return
|
||
|
|
|
||
|
|
if state.state in (STATE_ON, STATE_LOCKED, STATE_ABOVE_HORIZON):
|
||
|
|
_state = 1
|
||
|
|
elif state.state in (STATE_OFF, STATE_UNLOCKED, STATE_UNKNOWN,
|
||
|
|
STATE_BELOW_HORIZON):
|
||
|
|
_state = 0
|
||
|
|
else:
|
||
|
|
_state = state.state
|
||
|
|
if _state == '':
|
||
|
|
return
|
||
|
|
try:
|
||
|
|
_state = float(_state)
|
||
|
|
except ValueError:
|
||
|
|
pass
|
||
|
|
|
||
|
|
if not isinstance(_state, NUM_TYPES):
|
||
|
|
return
|
||
|
|
|
||
|
|
_LOGGER.debug('Sending %s.%s', state.entity_id, _state)
|
||
|
|
meter.send(state.entity_id, _state)
|
||
|
|
|
||
|
|
hass.bus.listen(EVENT_STATE_CHANGED, statsd_event_listener)
|
||
|
|
|
||
|
|
return True
|