# Implementação do módulo {comando_buscar_trechos}.

import trecho
import sessao
import html_lista_de_trechos
import html_pag_generica
import html_pag_buscar_trechos
import sys

from valida_campo import ErroAtrib

def verifica_pelo_menos_um_campo(campos, dic):
  """ Garante que:
     - Todos os campos da busca são suportados pelo Objeto_Trecho.
     - Todos os campos da busca estão definidos com algo diferente de None.
     - Pelo menos um campo de busca está definido.
  """
  sys.stderr.write("dic = %s\n" % str(dic))
  for campo_busca in list(dic.keys()):
    if campo_busca not in campos:
      del dic[campo_busca]

  tem_campo = False
  for campo in campos:
    if campo in dic and dic[campo] is not None:
      if dic[campo] is None:
        del dic[campo]
      else:
        tem_campo = True
        break

  return tem_campo

def processa(ses, args):
  try:
    campos = ['origem', 'destino', 'data_min', 'data_max']
    if not verifica_pelo_menos_um_campo(campos, args):
      raise ErroAtrib("Pelo menos um dos campos da busca precisa estar preenchido")
    
    # !!! Considerar dia e hora de partida e chegada !!!
    for campo in campos:
      if campo not in args: args[campo] = None

    trcs_ids = trecho.busca_por_origem_e_destino(args['origem'], args['destino'], args['data_min'], args['data_max'])
    trcs = map(lambda id_trecho: trecho.busca_por_identificador(id_trecho), trcs_ids)
    alterar_trcs = sessao.eh_administrador(ses)
    bloco = html_lista_de_trechos.gera(trcs, alterar_trcs)
    pag = html_pag_generica.gera(ses, bloco, None)
    return pag

  except ErroAtrib as ex:
    erros = ex.args[0]
    # Repete a página com mensagem de erro:
    admin = sessao.eh_administrador(ses)
    pag = html_pag_buscar_trechos.gera(ses, args, admin, erros)
    return pag