refactor: linting to follow Pylint recommendations

TestEnvironment/pub_client_1.py

@@ -1,30 +1,32 @@
-# simulator device 1 for mqtt message publishing
+"""
+Simulator device 1 for MQTT message publishing.
+"""
+
 import random
 import time
-
 import paho.mqtt.client as paho
 
-# brokersettings
-BROKER = "localhost"  # mqtt broker url + port exposed to local
+# Broker settings
+BROKER = "localhost"  # MQTT broker URL
 PORT = 1883
 
+def on_publish(client, userdata, result): # pylint: disable=unused-argument
+    """Callback function for when a message is published."""
+    print("Device 1: Data published.")
 
-def on_publish(client, userdata, result):
-    print("Device 1 : Data published.")
-    pass
-
-
-client = paho.Client(client_id="admin")
-client.on_publish = on_publish
-client.connect(host=BROKER, port=PORT)
-for i in range(20):
-    d = random.randint(1, 5)
+def main():
+    """Main function to publish MQTT messages."""
+    client = paho.Client(client_id="admin")
+    client.on_publish = on_publish
+    client.connect(host=BROKER, port=PORT)
 
-    # telemetry to send
-    MESSAGE = "Device 1 : Data " + str(i)
-    time.sleep(d)
+    for i in range(20):
+        delay = random.randint(1, 5)
+        message = f"Device 1: Data {i}"
+        time.sleep(delay)
+        client.publish(topic="data", payload=message)
 
-    # publish message
-    ret = client.publish(topic="data", payload=MESSAGE)
+    print("Stopped...")
 
-print("Stopped...")
+if __name__ == "__main__":
+    main()

TestEnvironment/sub_client_1.py

@@ -1,24 +1,32 @@
+"""
+MQTT Subscriber 1 for receiving messages.
+"""
+
 import paho.mqtt.client as mqtt
 
-# broker settings
-BROKER = "localhost"  # mqtt broker url + port exposed to local
+# Broker settings
+BROKER = "localhost"  # MQTT broker URL
 PORT = 1883
 
-# time for Subscriber to live
+# Time for Subscriber to live
 TIMELIVE = 60
 
-
-def on_connect(client, userdata, flags, rc):
+def on_connect(client, userdata, flags, rc): # pylint: disable=unused-argument
+    """Callback function for when the client connects to the broker."""
     print("Connected with result code " + str(rc))
     client.subscribe(topic="data")
 
-
-def on_message(client, userdata, msg):
+def on_message(client, userdata, msg): # pylint: disable=unused-argument
+    """Callback function for when a message is received."""
     print(msg.payload.decode())
 
+def main():
+    """Main function to set up the MQTT client and start the loop."""
+    sub_client = mqtt.Client()
+    sub_client.connect(host=BROKER, port=PORT, keepalive=TIMELIVE)
+    sub_client.on_connect = on_connect
+    sub_client.on_message = on_message
+    sub_client.loop_forever()
 
-sub_client = mqtt.Client()
-sub_client.connect(host=BROKER, port=PORT, keepalive=TIMELIVE)
-sub_client.on_connect = on_connect
-sub_client.on_message = on_message
-sub_client.loop_forever()
+if __name__ == "__main__":
+    main()

TestEnvironment/sub_client_2.py

@@ -1,24 +1,32 @@
+"""
+MQTT Subscriber 2 for receiving messages.
+"""
+
 import paho.mqtt.client as mqtt
 
-# broker settings
-BROKER = "localhost"  # mqtt broker url + port exposed to local
+# Broker settings
+BROKER = "localhost"  # MQTT broker URL
 PORT = 1883
 
-# time for Subscriber to live
+# Time for Subscriber to live
 TIMELIVE = 60
 
-
-def on_connect(client, userdata, flags, rc):
+def on_connect(client, userdata, flags, rc): # pylint: disable=unused-argument
+    """Callback function for when the client connects to the broker."""
     print("Connected with result code " + str(rc))
     client.subscribe(topic="data")
 
-
-def on_message(client, userdata, msg):
+def on_message(client, userdata, msg): # pylint: disable=unused-argument
+    """Callback function for when a message is received."""
     print(msg.payload.decode())
 
+def main():
+    """Main function to set up the MQTT client and start the loop."""
+    sub_client = mqtt.Client()
+    sub_client.connect(host=BROKER, port=PORT, keepalive=TIMELIVE)
+    sub_client.on_connect = on_connect
+    sub_client.on_message = on_message
+    sub_client.loop_forever()
 
-sub_client = mqtt.Client()
-sub_client.connect(host=BROKER, port=PORT, keepalive=TIMELIVE)
-sub_client.on_connect = on_connect
-sub_client.on_message = on_message
-sub_client.loop_forever()
+if __name__ == "__main__":
+    main()