import socket

client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
conectar = True

# USUARIO PEDINDO PERMISSAO PARA CONECTAR
while (conectar):
    varconexao = input('[ CLIENTE ]: ')
    if (varconexao == 'CONECTAR'):
        try:
            client.connect(('localhost', 1234))
        except:
            print('Servidor nao encontrado!\n')
        else:
            print(client.recv(1024).decode())
            print('')
            conectar = False
    else:
        print("Comando invalido!\n")


def BUSCAR(data):
    contador = client.send(data.encode())
    if (data == 'CONSULTAR'):
        contador = int(client.recv(1024).decode())
        print('[ SERVIDOR ]: Verificando...')
        x=0
        print(contador)
        while (x < contador):
            print('[ SERVIDOR 2 ]:', client.recv(1024).decode())
            x +=1

    else:
        print('[ SERVIDOR1]:', client.recv(1024).decode())
        return


while True:
    comando = input('[ CLIENTE ]: ')
    armazenarvalor = comando.split(' ')  # SEPARA EM LISTA OQUE FOI DIGITADO
    cont = len(armazenarvalor)  # COMTAGEM DOS ITENS DA LISTA

    # CONSULTA
    if (comando == 'CONSULTAR'):
        BUSCAR('CONSULTAR')

        print('')

    # ARMAZENAR
    elif (cont == 3):
        if (comando == 'ARMAZENAR ' + armazenarvalor[1] + ' ' + armazenarvalor[2]):
            addPT = armazenarvalor[1]
            addUS = armazenarvalor[2]
            BUSCAR('ARMAZENAR')
            client.send(addPT.encode())
            client.send(addUS.encode())
            print(client.recv(1024).decode())
        else:
            print('[ SERVIDOR ]: NÃO POSSUO TRADUÇÃO PARA ESSA PALAVRA\n')
    # SAIR
    elif (comando == 'SAIR'):
        BUSCAR('SAIR')
        client.close()
        break
    # FINALIZAR
    elif (comando == 'FINALIZAR'):
        BUSCAR('FINALIZAR')
        client.close()
        break
    # TRADUZIR PALAVRA
    else:
        if (comando == 'ARMAZENAR'):
            comando = 'incorreto'
        BUSCAR(comando)
        print('')