import mysql.connector
from mysql.connector import errorcode

from hl7apy.core import Message
import socket

host = "172.26.83.74"
port = 5678

try:
    cnx = mysql.connector.connect(user='root',
                                      password='root',
                                      database='BDhospital')
except mysql.connector.Error as err:
        if err.errno == errorcode.ER_ACCESS_DENIED_ERROR:
          print("Something is wrong with your user name or password")
        elif err.errno == errorcode.ER_BAD_DB_ERROR:
          print("Database does not exist")
        else:
          print(err)
else:
    flag = True
    i = 0
    while(flag):
        cursor = cnx.cursor()

        query = ("SELECT * FROM worklist WHERE estado_envio=0")

        cursor.execute(query)

        for row in cursor.fetchall():
            i += 1
            m = Message("ORM_O01", version="2.5")

            #dataPedido
            m.msh.msh_7 = str(row[5])
            #numPedido
            m.msh.msh_10 = str(row[0])
            #idPaciente
            m.pid.pid_3 = str(row[1])
            #numProcesso
            #m.pid.pid_18 = str(row[4])
            #moradaPaciente
            m.pid.pid_11 = row[2]
            #telefonePaciente
            m.pid.pid_13 = str(row[3])
            #orc
            if (str(row[7]) == 0):
                m.orc.orc_1 = "NW"
            elif (str(row[7]) == 2):
                m.orc.orc_1 = "SC"
            else:
                m.orc.orc_1 = "CA"
            #episodioMedico
            m.orc.orc_2 = str(row[8])
            m.orc.orc_3 = str(row[8])
            m.obr.obr_2 = str(row[8])
            m.obr.obr_3 = str(row[8])
            #descricao
            m.obr.obr_4 = row[9]

            pacote = m.to_mllp()

            sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

            try:
                print("...")
                sock.connect((host, port))
                print(pacote)
                sock.sendall(pacote.encode())

                #atualizar worklist na BD
                cursor2 = cnx.cursor()
                cursor2.execute ("""
                    UPDATE WORKLIST
                    SET estado = %s
                    WHERE id = %s
                    """, (1, row[0]))
                cnx.commit()
                cursor.close()
                print("Enviado...")


            finally:
                sock.close()

        x = input("Foram enviadas "+ str(i)+ " msgs! 1 - Continuar, Outro - Sair: ")
        x = int(x)
        if (x != 1):
            flag = False
        else:
            i = 0


        """
        0 -- Ativo
        1 -- Cancelado
        2 -- Alterado
        4 -- terminado

        0 -- por enviar
        1 -- enviado

        """

        #flag = False