mirror of
https://github.com/home-assistant/core.git
synced 2025-05-08 16:09:22 +00:00
2c07bfb9e0
* Remove dependencies and requirements * Revert "Remove dependencies and requirements" This reverts commit fe7171b4cd30889bad5adc9a4fd60059d05ba5a7. * Remove dependencies and requirements * Revert "Remove dependencies and requirements" This reverts commit 391355ee2cc53cbe6954f940062b18ae34b05621. * Remove dependencies and requirements * Fix flake8 complaints * Fix more flake8 complaints * Revert non-component removals
71 lines
2.0 KiB
Python
71 lines
2.0 KiB
Python
"""Support for Shiftr.io."""
|
|
import logging
|
|
|
|
import voluptuous as vol
|
|
|
|
import homeassistant.helpers.config_validation as cv
|
|
from homeassistant.const import (
|
|
CONF_PASSWORD, CONF_USERNAME, EVENT_STATE_CHANGED,
|
|
EVENT_HOMEASSISTANT_STOP)
|
|
from homeassistant.helpers import state as state_helper
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
DOMAIN = 'shiftr'
|
|
|
|
SHIFTR_BROKER = 'broker.shiftr.io'
|
|
|
|
CONFIG_SCHEMA = vol.Schema({
|
|
DOMAIN: vol.Schema({
|
|
vol.Required(CONF_USERNAME): cv.string,
|
|
vol.Required(CONF_PASSWORD): cv.string,
|
|
}),
|
|
}, extra=vol.ALLOW_EXTRA)
|
|
|
|
|
|
def setup(hass, config):
|
|
"""Initialize the Shiftr.io MQTT consumer."""
|
|
import paho.mqtt.client as mqtt
|
|
conf = config[DOMAIN]
|
|
username = conf.get(CONF_USERNAME)
|
|
password = conf.get(CONF_PASSWORD)
|
|
|
|
client_id = 'HomeAssistant'
|
|
port = 1883
|
|
keepalive = 600
|
|
|
|
mqttc = mqtt.Client(client_id, protocol=mqtt.MQTTv311)
|
|
mqttc.username_pw_set(username, password=password)
|
|
mqttc.connect(SHIFTR_BROKER, port=port, keepalive=keepalive)
|
|
|
|
def stop_shiftr(event):
|
|
"""Stop the Shiftr.io MQTT component."""
|
|
mqttc.disconnect()
|
|
|
|
hass.bus.listen_once(EVENT_HOMEASSISTANT_STOP, stop_shiftr)
|
|
|
|
def shiftr_event_listener(event):
|
|
"""Listen for new messages on the bus and sends them to Shiftr.io."""
|
|
state = event.data.get('new_state')
|
|
topic = state.entity_id.replace('.', '/')
|
|
|
|
try:
|
|
_state = state_helper.state_as_number(state)
|
|
except ValueError:
|
|
_state = state.state
|
|
|
|
try:
|
|
mqttc.publish(topic, _state, qos=0, retain=False)
|
|
|
|
if state.attributes:
|
|
for attribute, data in state.attributes.items():
|
|
mqttc.publish(
|
|
'/{}/{}'.format(topic, attribute), str(data), qos=0,
|
|
retain=False)
|
|
except RuntimeError:
|
|
pass
|
|
|
|
hass.bus.listen(EVENT_STATE_CHANGED, shiftr_event_listener)
|
|
|
|
return True
|