#!/usr/bin/python3

# Pérenisation de données arrivant sur un MQTT
# vers une base de données MariaDb/MySQL
# Auteur : Arofarn
# v0.1

#################
# Configuration #
#################

#MQTT Broker
mqtt_host = "aro-yuno"
mqtt_port = 1883
mqtt_client_id = "mqtt2sql"
#mqtt_auth = { 'username' : 'arofarn', 'password' : 'xxx' }
qos=1
topic = "huzzah0/#"

#MySQL database

sql_host = "localhost"
#sql_port =  
sql_base = "iot_test"
sql_user = "arofarn"
#sql_pass = "password"

########
# MAIN #
########

import paho.mqtt.client as mqtt
from datetime import datetime
import json
import pymysql.cursors

def on_connect(client, userdata, flags, rc):
    print("Connected with result code "+str(rc))
    client.subscribe(topic)

def on_message(client, userdata, msg):
    top=msg.topic[len(topic)-1:].strip()

    subtopics = top.split("/")
    payload = msg.payload.decode()
    
    val = json.loads(payload)
    print("%s : %s %s (%s)" % (val['type'], val['value'], val['unit'], subtopics[0]))
    
    with connection.cursor() as cursor:
        sql = "INSERT INTO `data` (`valdate`, `value`, `type_id`, `sensor_id`) VALUES (%s, %s, %s, %s)"
        cursor.execute(sql, (datetime.strptime(val['date'], "%d/%m/%Y %H:%M:%S"), float(val['value']), val['type'], subtopics[0]))

    connection.commit()

def on_message_date(client, userdata, msg):
    payload = msg.payload.decode()
    try:
        d = datetime.strptime(payload, "%d/%m/%Y %H:%M:%S")
        print("Date : %s" % d)
    except:
        print("Date mal formatée : %s" % payload)
    

def on_disconnect(client, userdata, msg):
    if msg != 0:
        print("Déconnexion imprévu : %s" % msg)
    connection.close()

connection = pymysql.connect(host=sql_host,
                             user=sql_user,
                             #password=sql_pass,
                             db=sql_base,
                             charset='utf8mb4',
                             cursorclass=pymysql.cursors.DictCursor)

########
# Main #
########

#Connect to MQTT broker and loop...
mqtt_client = mqtt.Client(mqtt_client_id, clean_session=False)
#mqtt_client.username_pw_set(mqtt_auth['username'], mqtt_auth['password'])
mqtt_client.on_connect = on_connect
mqtt_client.on_message = on_message
mqtt_client.on_disconnect = on_disconnect

mqtt_client.message_callback_add("huzzah0/NTP/date", on_message_date)
    
mqtt_client.connect(mqtt_host, mqtt_port, 60)

mqtt_client.loop_forever()