A more robust way to handle JSON serialization/deserialization on MQTT backend

platypush/backend/mqtt.py

@@ -5,7 +5,6 @@ import paho.mqtt.publish as publisher
 
 from platypush.backend import Backend
 from platypush.message import Message
-from platypush.context import get_backend
 
 
 class MqttBackend(Backend):
@@ -27,15 +26,10 @@ class MqttBackend(Backend):
         self.port = port
         self.topic = '{}/{}'.format(topic, self.device_id)
 
-        try:
-            self.redis = get_backend('redis')
-        except:
-            self.redis = None
-
 
     def send_message(self, msg):
         try:
-            msg = json.loads(msg)
+            msg = Message.build(json.loads(msg))
         except:
             pass
 
@@ -48,17 +42,11 @@ class MqttBackend(Backend):
 
         def on_message(client, userdata, msg):
             msg = msg.payload.decode('utf-8')
-            try:
+            try: msg = Message.build(json.loads(msg))
-                msg = Message.build(json.loads(msg))
+            except: pass
-            except:
-                pass
 
             self.logger.info('Received message on the MQTT backend: {}'.format(msg))
-
+            self.bus.post(msg)
-            if self.redis:
-                self.redis.send_message(msg)
-            else:
-                self.bus.post(msg)
 
         super().run()
         client = mqtt.Client()

platypush/plugins/mqtt.py

@@ -1,11 +1,19 @@
+import json
 import paho.mqtt.publish as publisher
 
+from platypush.message import Message
 from platypush.message.response import Response
 from platypush.plugins import Plugin
 
 
 class MqttPlugin(Plugin):
     def send_message(self, topic, msg, host, port=1883, *args, **kwargs):
+        try: msg = json.dumps(msg)
+        except: pass
+
+        try: msg = Message.build(json.loads(msg))
+        except: pass
+
         publisher.single(topic, str(msg), hostname=host, port=port)
         return Response(output={'state': 'ok'})