import paho.mqtt.client as mqtt
from radarsys import settings
from radarsys.socketconfig import sio as sio
import numpy as np
import psycopg2
import os

def insert(time,data):
    sql = """INSERT INTO atrad_datas(
        datetime,nstx,status,temp_cll,nboards,tempdvr,potincdvr,potretdvr,
        temp1,potinc1,potret1,temp2,potinc2,potret2,temp3,potinc3,potret3,
        temp4,potinc4,potret4,temp5,potinc5,potret5,temp6,potinc6,potret6)
        VALUES(%s,%s,%s,%s,%s,%s,%s,%s,
        %s,%s,%s,%s,%s,%s,%s,%s,%s,
        %s,%s,%s,%s,%s,%s,%s,%s,%s);"""
    try:
        # connect to the PostgreSQL database
        conn = psycopg2.connect(database="radarsys", user='docker', password='docker', host='radarsys-postgres', port= '5432')
        # create a new cursor
        cur = conn.cursor()
        # execute the INSERT statement
        #data_tuple = [tuple(i[:]) for i in a]
        values = (time,) + tuple(data[0][:25])
        cur.execute(sql, values)

        # get the generated id back
        #vendor_id = cur.fetchone()[0]
        
        # commit the changes to the database
        conn.commit()
        # close communication with the database
        cur.close()
    except (Exception, psycopg2.DatabaseError) as error:
        print(error)
    finally:
        if conn is not None:
            conn.close()

def maxTemperature(trs):
    temps = GetTemperatures(trs)
    maxT_STX = [max(i) for i in temps]
    maxT = max(maxT_STX)
    STXnum = maxT_STX.index(maxT)
    STXloc = temps[STXnum].index(maxT)
    maxT_loc = 'Amp ' + str(STXnum+1)

    if STXloc == 0:
        maxT_loc = maxT_loc + " Controller"
    elif STXloc<7:
        maxT_loc = maxT_loc + " PA " + str(STXloc+1)
    else:
        maxT_loc = maxT_loc + " Combiners"

    return maxT,maxT_loc,temps

def dataConvert(msg):
    msgStr = str(msg.payload)
    msgClean = [i for i in msgStr[21:-1].split("*")]
    dataSTX = [[],[],[],[]]
    for trs,i in zip(msgClean,[0,1,2,3]) :
        dataSTX[i]= [int(i) for i in trs[1:-1].split(",")]
    # Data to database
    insert(msgStr[2:21],dataSTX)
    # Data to send by socket
    id_STX = dataSTX[0][0] // 4
    status = ''.join([msgClean[i][3] for i in [0,1,2,3]])
    powers = [dataSTX[0][34],dataSTX[0][36],dataSTX[2][32],dataSTX[2][34]]
    tmax,index,tempData = maxTemperature(dataSTX)
    #Json to send
    data = {'time':msgStr[2:21],'num':id_STX,'pow':powers,'tmax':[str(tmax),index],'status':status}
    data_temp = {'time':msgStr[2:21],'temp':tempData}
    return data, data_temp

def GetTemperatures(data):
    np_data = [np.array(i) for i in data]
    temps = [i[i<40] for i in np_data]
    return [i[i>15].tolist() for i in temps]

def on_connect(mqtt_client, userdata, flags, rc):
   if rc == 0:
       print('Connected successfullyasdss')
       mqtt_client.subscribe("atrad/test4")
       print("Exito")
   else:
       print('Bad connection. Code:', rc)

def on_message(mqtt_client, userdata, msg):
    print('Received message on topic: {} with payload: {}'.format(msg.topic,msg.payload), flush=True)
    mainData, tempData = dataConvert(msg)
    # print("Recibi : {}".format(msg.payload),flush=True)
    #socket fot general data 
    sio.emit('test',data = mainData)
    print(mainData)
    #socket for temperature details 
    sio.emit('temptx'+str(mainData['num'] + 1),data = tempData)

client = mqtt.Client()
client.on_connect = on_connect
client.on_message = on_message
client.username_pw_set( '','')
client.connect(
    host=os.environ.get('MQTT_SERVER', '10.10.10.200'),
    port=int(settings.os.environ.get('MQTT_PORT', 1883)),
    keepalive=int(os.environ.get('MQTT_KEEPALIVE', 60))
)