mqtt_handler.py

#!/usr/bin/python
#
# Created by Matthijs Visser

import logging
import paho.mqtt.client as paho
import ssl
from logging.handlers import TimedRotatingFileHandler

log = logging.getLogger("log")
log.setLevel(logging.INFO)

class MqqtHandler (object):
    
	def __init__(self, broker_ip, broker_port, client_id, topic_prefix, retain=False, qos=0,
			authentication=False, user="", password="", tls_enabled=False,
			tls_ca_cert="", tls_cert="", tls_key="",tls_insecure=True):
		self.broker = broker_ip
		self.port = broker_port
		self.client_id = client_id
		self.topic_prefix = topic_prefix
		self.retain = retain
		self.qos = qos
		self.authentication = authentication
		self.user = user
		self.password = password
		self.tls_enabled = tls_enabled
                self.tls_ca_cert = tls_ca_cert
                self.tls_cert = tls_cert
                self.tls_key = tls_key
                self.tls_insecure = tls_insecure
	
	def connect(self):
		settings_message = ""
		self.mqtt_client = paho.Client(self.client_id, True)

		if self.authentication:
			self.mqtt_client.username_pw_set(self.user, self.password)
			settings_message = 'with username {}, '.format(self.user)
		if self.tls_enabled:
                        self.mqtt_client.tls_set(self.tls_ca_cert,tls_version=ssl.PROTOCOL_TLSv1_2)
                        self.mqtt_client.tls_insecure_set(self.tls_insecure)

		self.mqtt_client.connect(self.broker, self.port, 60)
		self.mqtt_client.loop_start()
		log.info('Connected to MQTT at: {}:{}'.format(self.broker, self.port))
		settings_message += 'QoS level = {} and retain = {}'.format(self.qos, self.retain)
		log.info(settings_message)
		
	def disconnect(self):
		self.mqtt_client.disconnect()

	def publish(self, topic, message):
		full_topic = self.create_topic(topic.lower())
		try:
			log.info('Publishing \'{}\'\t\'{}\'\tto {}:{}'.format(full_topic, message,
														   	self.broker, 
															self.port))
			mqtt_info = self.mqtt_client.publish(full_topic, message, self.qos, self.retain)
			mqtt_info.wait_for_publish()

		except ValueError as e:
			logging.error('Value error: {}'.format(e))
		except TypeError as e:
			logging.error('Type error: {}'.format(e))

	def subscribe(self, topic):
		if self.mqtt_client.subscribe(topic) == 0:
			log.info('Subscribed to topic: {}'.format(self.topic_prefix))
			return True
		else:
			return False

	def create_topic(self, data):
		return "{}/{}".format(self.topic_prefix, data)

	def loop_start(self):
		self.mqtt_client.loop_start()

	def loop_stop(self):
		self.mqtt_client.loop_stop()