MQTT component

.coveragerc

@@ -18,6 +18,8 @@ omit =
     homeassistant/components/modbus.py
     homeassistant/components/*/modbus.py
 
+    homeassistant/components/mqtt.py
+
     homeassistant/components/wink.py
     homeassistant/components/*/wink.py
 

homeassistant/components/mqtt.py

@@ -0,0 +1,198 @@
+"""
+homeassistant.components.mqtt
+~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
+MQTT component, using paho-mqtt. This component needs a MQTT broker like
+Mosquitto or Mosca. The Eclipse Foundation is running a public MQTT server
+at iot.eclipse.org. If you prefer to use that one, keep in mind to adjust
+the topic/client ID and that your messages are public.
+
+Configuration:
+
+To use MQTT you will need to add something like the following to your
+config/configuration.yaml.
+
+mqtt:
+  broker: 127.0.0.1
+  port: 1883
+  topic: home-assistant
+  keepalive: 60
+  qos: 0
+
+Variables:
+
+broker
+*Required
+This is the IP address of your MQTT broker, e.g. 192.168.1.32.
+
+port
+*Optional
+The network port to connect to. Default is 1883.
+
+topic
+*Required
+The MQTT topic to subscribe to, e.g. home-assistant.
+
+keepalive
+*Optional
+The keep alive in seconds for this client, e.g. 60.
+
+qos
+*Optional
+Quality of service level to use for the subscription.
+0, 1, or 2, defaults to 0.
+"""
+import logging
+
+from homeassistant.helpers import validate_config
+from homeassistant.const import (
+    EVENT_HOMEASSISTANT_START, EVENT_HOMEASSISTANT_STOP)
+
+try:
+    import paho.mqtt.client as mqtt
+except ImportError:
+    mqtt = None
+
+_LOGGER = logging.getLogger(__name__)
+
+DOMAIN = "mqtt"
+DEPENDENCIES = []
+MQTT_CLIENT = None
+MQTT_CLIENT_ID = 'home-assistant'
+MQTT_DEFAULT_PORT = 1883
+MQTT_DEFAULT_KEEPALIVE = 60
+MQTT_DEFAULT_QOS = 0
+MQTT_SEND = 'mqtt_send'
+EVENT_MQTT_MESSAGE_RECEIVED = 'MQTT_MESSAGE_RECEIVED'
+REQUIREMENTS = ['paho-mqtt>=1.1']
+
+ATTR_SUBTOPIC = 'subtopic'
+ATTR_PAYLOAD = 'payload'
+
+
+def send_message(hass, subtopic, payload):
+    """ Send an MQTT message. """
+    hass.services.call(DOMAIN, MQTT_SEND, {ATTR_SUBTOPIC: subtopic,
+                                           ATTR_PAYLOAD: payload})
+
+
+def setup(hass, config):
+    """ Get the MQTT protocol service. """
+
+    if not validate_config(config,
+                           {DOMAIN: ['broker',
+                                     'topic']},
+                           _LOGGER):
+        return False
+
+    if mqtt is None:
+        _LOGGER.error("Error while importing dependency 'paho-mqtt'.")
+        return False
+
+    global MQTT_CLIENT
+
+    broker = config[DOMAIN]['broker']
+    port = config[DOMAIN].get('port', MQTT_DEFAULT_PORT)
+    topic = config[DOMAIN]['topic']
+    keepalive = config[DOMAIN].get('keepalive', MQTT_DEFAULT_KEEPALIVE)
+    qos = config[DOMAIN].get('qos', MQTT_DEFAULT_QOS)
+
+    MQTT_CLIENT = MQTT(hass, broker, port, topic, keepalive, qos,
+                       MQTT_CLIENT_ID)
+
+    def stop_mqtt(event):
+        """ Stop MQTT component. """
+        MQTT_CLIENT.stop()
+
+    def start_mqtt(event):
+        """ Launch MQTT component when Home Assistant starts up. """
+        try:
+            MQTT_CLIENT.run()
+
+        except ConnectionRefusedError:
+            _LOGGER.exception("Can't connect to the broker. "
+                              "Please check your settings and the broker"
+                              "itself.")
+            return False
+
+        hass.bus.listen_once(EVENT_HOMEASSISTANT_STOP, stop_mqtt)
+
+    def mqtt_message(call):
+        """ Handle sending MQTT message service calls. """
+        subtopic = call.data.get(ATTR_SUBTOPIC)
+        complete_topic = '{}/{}'.format(str(topic), str(subtopic))
+
+        payload = call.data.get(ATTR_PAYLOAD)
+
+        MQTT_CLIENT.publish(complete_topic, payload=payload)
+
+    hass.bus.listen_once(EVENT_HOMEASSISTANT_START, start_mqtt)
+
+    hass.services.register(DOMAIN, MQTT_SEND, mqtt_message)
+
+    return True
+
+
+# This is based on one of the paho-mqtt examples:
+# http://git.eclipse.org/c/paho/org.eclipse.paho.mqtt.python.git/tree/examples/sub-class.py
+# pylint: disable=too-many-arguments, invalid-name
+class MQTT(object):
+    """ Implements messaging service for MQTT. """
+    def __init__(self, hass, broker, port, topic, keepalive, qos, clientid):
+
+        self.hass = hass
+        self._broker = broker
+        self._port = port
+        self._topic = topic
+        self._keepalive = keepalive
+        self._qos = qos
+
+        self._mqttc = mqtt.Client(clientid)
+        self._mqttc.on_message = self.mqtt_on_message
+        self._mqttc.on_connect = self.mqtt_on_connect
+        self._mqttc.on_publish = self.mqtt_on_publish
+        self._mqttc.on_subscribe = self.mqtt_on_subscribe
+
+    def mqtt_on_connect(self, mqttc, obj, flags, rc):
+        """ Connect callback """
+        _LOGGER.info('Connected to broker %s', self._broker)
+
+    def mqtt_on_publish(self, mqttc, obj, mid):
+        """ Publish callback """
+        pass
+
+    def mqtt_on_subscribe(self, mqttc, obj, mid, granted_qos):
+        """ Subscribe callback """
+        complete_topic = '{}/#'.format(self._topic)
+        _LOGGER.info('Subscribed to %s', complete_topic)
+
+    def mqtt_on_message(self, mqttc, obj, msg):
+        """ Message callback """
+        self.hass.bus.fire(EVENT_MQTT_MESSAGE_RECEIVED, {
+            'topic': msg.topic,
+            'qos': str(msg.qos),
+            'payload': msg.payload.decode('utf-8'),
+        })
+
+    def subscribe(self, topic):
+        """ Subscribe to a topic. """
+        self._mqttc.subscribe(self._topic, qos=self._qos)
+
+    def unsubscribe(self, topic):
+        """ Unsubscribe from topic. """
+        self._mqttc.unsubscribe(topic)
+
+    def stop(self):
+        """ Stop the MWTT client. """
+        self._mqttc.loop_stop()
+
+    def run(self):
+        """ Run the MQTT client. """
+        self._mqttc.connect(self._broker,
+                            int(self._port),
+                            int(self._keepalive))
+        self._mqttc.subscribe('{}/#'.format(self._topic), qos=self._qos)
+        self._mqttc.loop_start()
+
+    def publish(self, topic, payload):
+        """ Publish a MQTT message. """
+        self._mqttc.publish(topic, payload)

requirements.txt

@@ -103,3 +103,6 @@ https://github.com/rkabadi/pyedimax/archive/master.zip
 
 # RPI-GPIO platform
 RPi.GPIO >=0.5.11
+
+# PAHO MQTT Binding (protocol.mqtt)
+paho-mqtt>=1.1