add: agent template

agent/src/main.py

@@ -0,0 +1,53 @@
+from paho.mqtt import client as mqtt_client
+import json
+import time
+from schema.aggregated_data_schema import AggregatedDataSchema
+from file_datasource import FileDatasource
+import config
+
+
+def connect_mqtt(broker, port):
+    """Create MQTT client"""
+    print(f"CONNECT TO {broker}:{port}")
+
+    def on_connect(client, userdata, flags, rc):
+        if rc == 0:
+            print(f"Connected to MQTT Broker ({broker}:{port})!")
+        else:
+            print("Failed to connect {broker}:{port}, return code %d\n", rc)
+            exit(rc)  # Stop execution
+
+    client = mqtt_client.Client()
+    client.on_connect = on_connect
+    client.connect(broker, port)
+    client.loop_start()
+    return client
+
+
+def publish(client, topic, datasource, delay):
+    datasource.startReading()
+    while True:
+        time.sleep(delay)
+        data = datasource.read()
+        msg = AggregatedDataSchema().dumps(data)
+        result = client.publish(topic, msg)
+        # result: [0, 1]
+        status = result[0]
+        if status == 0:
+            pass
+            # print(f"Send `{msg}` to topic `{topic}`")
+        else:
+            print(f"Failed to send message to topic {topic}")
+
+
+def run():
+    # Prepare mqtt client
+    client = connect_mqtt(config.MQTT_BROKER_HOST, config.MQTT_BROKER_PORT)
+    # Prepare datasource
+    datasource = FileDatasource("data/data.csv", "data/gps_data.csv")
+    # Infinity publish data
+    publish(client, config.MQTT_TOPIC, datasource, config.DELAY)
+
+
+if __name__ == "__main__":
+    run()