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_1,temp_cll_1,nboards_1,tempdvr_1,potincdvr_1,potretdvr_1,
        temp1_1,potinc1_1,potret1_1,temp2_1,potinc2_1,potret2_1,temp3_1,potinc3_1,potret3_1,
        temp4_1,potinc4_1,potret4_1,temp5_1,potinc5_1,potret5_1,temp6_1,potinc6_1,potret6_1,
        status_2,temp_cll_2,nboards_2,tempdvr_2,potincdvr_2,potretdvr_2,
        temp1_2,potinc1_2,potret1_2,temp2_2,potinc2_2,potret2_2,temp3_2,potinc3_2,potret3_2,
        temp4_2,potinc4_2,potret4_2,temp5_2,potinc5_2,potret5_2,temp6_2,potinc6_2,potret6_2,
        status_3,temp_cll_3,nboards_3,tempdvr_3,potincdvr_3,potretdvr_3,
        temp1_3,potinc1_3,potret1_3,temp2_3,potinc2_3,potret2_3,temp3_3,potinc3_3,potret3_3,
        temp4_3,potinc4_3,potret4_3,temp5_3,potinc5_3,potret5_3,temp6_3,potinc6_3,potret6_3,
        status_4,temp_cll_4,nboards_4,tempdvr_4,potincdvr_4,potretdvr_4,
        temp1_4,potinc1_4,potret1_4,temp2_4,potinc2_4,potret2_4,temp3_4,potinc3_4,potret3_4,
        temp4_4,potinc4_4,potret4_4,temp5_4,potinc5_4,potret5_4,temp6_4,potinc6_4,potret6_4)
        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,
        %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,%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,%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])+tuple(data[1][1:25])+tuple(data[2][1:25])+tuple(data[3][1: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 == 1:
        maxT_loc = maxT_loc + " Driver"
    elif STXloc<8:
        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],0,0,0,0]
    # alerta
    for i in range(4):
        if powers[i] < 10000 and status == '1111':
            power[4+i] = 1

    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)
    sio.emit('test',data = 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))
)