pub sub completed

sub.py

 import json
-import random
-import time
 from paho.mqtt import client
 
 from scripts.config.app_config import mqtt_username, mqtt_password, mqtt_host, mqtt_port, token
+from scripts.config.databases.redis.redis_connection import connection_1, connection_2, connection_3
 
 
 def connect_mqtt() -> client:
@@ -24,6 +23,17 @@ def subscribe(mqtt_client: client):
     def on_message(client, userdata, msg):
         data = msg.payload.decode()
         data_dict = json.loads(data)
+        no = data_dict["site_id"]
+        if data_dict["data_quality"] == 0:
+            r1 = connection_1.set(no, json.dumps(data_dict))
+            print("r1", r1)
+        elif data_dict["data_quality"] == 1:
+            r2 = connection_2.set(no, json.dumps(data_dict))
+            print("r1", r2)
+        elif data_dict["data_quality"] == 2:
+            r3 = connection_3.set(no, json.dumps(data_dict))
+            print("r1", r3)
+
         print(f"Received `{data_dict}` from `{msg.topic}` topic")
 
     mqtt_client.subscribe(token)