jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-14 00:37:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

Clean up influx component

Browse Source
This commit is contained in:
Paulus Schoutsen 2015-12-06 09:45:58 -08:00
parent a833e8b413
commit b7712ac682
1 changed files with 14 additions and 21 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

35
homeassistant/components/influx.py
View File

@ -71,29 +71,23 @@ def setup(hass, config):
_LOGGER.error("Database %s doesn't exist", dbname) _LOGGER.error("Database %s doesn't exist", dbname)
return False return False
def event_listener(event): def influx_event_listener(event):
""" Listen for new messages on the bus and sends them to Influx. """ """ Listen for new messages on the bus and sends them to Influx. """
event_data = event.as_dict()
if event_data['event_type'] is not EVENT_STATE_CHANGED: state = event.data.get('new_state')
if state is None:
return return
state = event_data['data']['new_state'] if state.state in (STATE_ON, STATE_LOCKED, STATE_ABOVE_HORIZON):
if state.state == STATE_ON or state.state == STATE_LOCKED or \
state.state == STATE_ABOVE_HORIZON:
_state = 1 _state = 1
elif state.state == STATE_OFF or state.state == STATE_UNLOCKED or \ elif state.state in (STATE_OFF, STATE_UNLOCKED, STATE_UNKNOWN,
state.state == STATE_UNKNOWN or \ STATE_BELOW_HORIZON):
state.state == STATE_BELOW_HORIZON:
_state = 0 _state = 0
else: else:
_state = state.state _state = state.state
try: measurement = state.attributes.get('unit_of_measurement', state.domain)
measurement = state.attributes['unit_of_measurement']
except KeyError:
measurement = '{}'.format(state.domain)
json_body = [ json_body = [
{ {
@ -102,19 +96,18 @@ def setup(hass, config):
'domain': state.domain, 'domain': state.domain,
'entity_id': state.object_id, 'entity_id': state.object_id,
}, },
'time': event_data['time_fired'], 'time': event.time_fired,
'fields': { 'fields': {
'value': _state, 'value': _state,
} }
} }
] ]
if json_body: try:
try: influx.write_points(json_body)
influx.write_points(json_body) except exceptions.InfluxDBClientError:
except exceptions.InfluxDBClientError: _LOGGER.exception('Error saving event to Influx')
_LOGGER.exception('Error saving event to Influx')
hass.bus.listen(EVENT_STATE_CHANGED, event_listener) hass.bus.listen(EVENT_STATE_CHANGED, influx_event_listener)
return True return True