import psycopg2


def main():

    # ----------- conectar -----------
    conn = None

    try:
      conn = psycopg2.connect("dbname='econexoes' user='postgres' host='localhost' password='eco123'")
    except:
      print("I am unable to connect to the database")

    try:
      cur = conn.cursor()

      cur.execute(
        "select id, path from diretorio where cliente_id is null and path != '/'")
      diretorios = cur.fetchall()
      clientes_nao_econtrados = []

      for diretorio in diretorios:
        clienteNome = diretorio[1].split('/')[2]

        cur.execute("select id from cliente where nome like '{}' or nome like '{}'".format(clienteNome, clienteNome.upper()))
        clientes = cur.fetchall()

        if len(clientes) == 0 and clienteNome not in clientes_nao_econtrados:
          clientes_nao_econtrados.append(clienteNome)
          print(clienteNome)
        else:
          for cliente in clientes:
            print("{} - success".format(cliente[0]))
            cur.execute("update diretorio set cliente_id = {} where id = {}". format(cliente[0], diretorio[0]))
            conn.commit()

      print(len(clientes_nao_econtrados))
      cur.close()
      conn.close()
    except (Exception, psycopg2.DatabaseError) as error:
      print(error)


print("main")
main()