def testa_novos_valores():
id_poltrona = "A-00000001"
poltrona_antiga = poltrona.busca_por_identificador(id_poltrona)
assert poltrona_antiga is not None, "Poltrona A-00000001 não encontrada no banco"
# Valores antigos
atributos_antigos = poltrona.obtem_atributos(poltrona_antiga)
# Novos valores
nova_oferta = "off"
novo_preco = "88.00"
args = {'id_poltrona': id_poltrona,
'id_trecho': atributos_antigos['id_trecho'],
'id_compra': atributos_antigos['id_compra'],
'numero': atributos_antigos['numero'],
'oferta': nova_oferta,
'preco': novo_preco}
testa("ValoresAlterados", ses, args)
# Valores atualizados
poltrona_atualizada = poltrona.busca_por_identificador(id_poltrona)
atributos_atualizados = poltrona.obtem_atributos(poltrona_atualizada)
assert atributos_atualizados['oferta'] == False, "Oferta não atualizada"
assert atributos_atualizados['preco'] == float(novo_preco), "Preco não atualizado"
def obtem_poltronas(cpr):
global cache, nome_tb, letra_tb, colunas, diags
assert cpr != None and type(cpr) is compra.Objeto_Compra
ids_pols = poltrona.busca_por_compra(cpr)
ids_pols.sort(key=lambda id: poltrona.obtem_dia_e_hora_de_partida(
poltrona.busca_por_identificador(id)))
return ids_pols
def processa(ses, args):
id_poltronas = poltrona.busca_ofertas()
poltronas = [poltrona.busca_por_identificador(id) for id in id_poltronas]
id_trechos = list(
set([poltrona.obtem_atributo(pol, 'id_trecho') for pol in poltronas]))
trechos = [trecho.busca_por_identificador(id) for id in id_trechos]
pag = html_pag_ofertas.gera(ses, trechos, None)
return pag
def calcula_preco(cpr):
global cache, nome_tb, letra_tb, colunas, diags
assert cpr != None and type(cpr) is compra.Objeto_Compra
preco = 0
for id_pol in obtem_poltronas(cpr):
pol = poltrona.busca_por_identificador(id_pol)
preco = preco + poltrona.obtem_atributo(pol, 'preco')
return preco
def processa(ses, args):
# Obtém id da poltrona a ver
id_pol = args['id_poltrona'] if 'id_poltrona' in args else None
assert id_pol != None # Paranóia (formulário deve incluir sempre).
pol = poltrona.busca_por_identificador(id_pol)
assert pol != None # Paranóia.
pag = html_pag_poltrona.gera(ses, pol, None, None)
return pag
def testa_valores_criticos_invalidos():
id1 = "A-00000001"
ptr1 = poltrona.busca_por_identificador(id1)
assert ptr1 is not None, "Poltrona A-00000001 não encontrada no banco"
# Valores antigos
atributos_antigos = poltrona.obtem_atributos(ptr1)
ptr2 = poltrona.busca_por_identificador("A-00000002")
assert ptr2 is not None, "Poltrona A-00000002 não encontrada no banco"
# Argumentos com id de trecho E número sendo usado por outra poltrona (A-00000002)
args_mesmo_trecho = {'id_poltrona': id1,
'id_trecho': poltrona.obtem_atributo(ptr2, 'id_trecho'),
'id_compra': atributos_antigos['id_compra'],
'numero': poltrona.obtem_atributo(ptr2, 'numero'),
'oferta': 'on' if atributos_antigos['oferta'] else 'off',
'preco': atributos_antigos['preco']}
testa("MesmoTrechoENumero", ses, args_mesmo_trecho)
def gera(ids_poltronas, usr, carr):
id_trc = None # Identificador do trecho comum, ou None.
admin = False if usr == None else usuario.obtem_atributo(
usr, 'administrador')
fazer_checkin = admin # Por enquanto.
embarcar = admin #Por enquanto
if admin or usr == None:
assert carr == None # Administrador e não-logado não tem carrinho.
id_carr = None if carr == None else compra.obtem_identificador(carr)
# Gera a linha de os cabeçalhos das colunas
linha_cab = html_resumo_de_poltrona_de_trecho.gera_cabecalho(fazer_checkin)
assert type(linha_cab) is list or type(linha_cab) is tuple # Paranóia.
# Gera as linhas da tabela.
linhas = [].copy()
for id_poltrona in ids_poltronas:
pol = poltrona.busca_por_identificador(id_poltrona)
# Obtem dados da poltrona:
assert pol != None
id_trc_pol = poltrona.obtem_atributo(pol, 'id_trecho')
id_cpr_pol = poltrona.obtem_atributo(pol, 'id_compra')
cpr_pol = None if id_cpr_pol == None else compra.busca_por_identificador(
id_cpr_pol)
usr_pol = None if cpr_pol == None else compra.obtem_cliente(cpr_pol)
# Verifica se o trecho é o mesmo para todas as poltronas:
if id_trc != None: assert id_trc_pol == id_trc
id_trc = id_trc_pol
# Decide que botões vai ter:
alterar = admin
comprar = (carr != None) and poltrona.pode_comprar(usr, pol, carr)
excluir = (cpr_pol != None) and (usr_pol
== usr) and poltrona.pode_excluir(
usr, pol)
# Gera os campos da linha:
linha = html_resumo_de_poltrona_de_trecho.gera(pol, alterar, comprar,
excluir, fazer_checkin,
embarcar)
assert type(linha) is list or type(linha) is tuple # Paranóia.
linhas.append(linha)
ht_tabela = html_table.gera(linhas, linha_cab)
ht_legenda = html_resumo_de_poltrona_de_trecho.gera_legenda(fazer_checkin)
ht_res = ht_tabela + ht_legenda
return ht_res
def processa(ses, args):
admin = False if ses == None else sessao.eh_administrador(ses)
assert admin # Paranóia (cliente comum e deslogado não deveria ter acesso a este cmd).
# Obtem a poltrona a alterar:
id_pol = args['id_poltrona'] if 'id_poltrona' in args else None
assert id_pol != None # Paranóia (formulário deveria especificar).
pol = poltrona.busca_por_identificador(id_pol)
assert pol != None # Paranóia.
pag = html_pag_poltrona.gera(ses, pol, None, None)
return pag
def trecho_eh_compativel(cpr, trc):
global cache, nome_tb, letra_tb, colunas, diags
assert cpr != None and type(cpr) is compra.Objeto_Compra
for id_pol in obtem_poltronas(cpr):
pol = poltrona.busca_por_identificador(id_pol)
id_trc_pol = poltrona.obtem_atributo(pol, 'id_trecho')
# Verifca se {trc} é compatível com o trecho da polrona {pol}:
trc_pol = None if id_trc_pol == None else trecho.busca_por_identificador(
id_trc_pol)
assert trc_pol != None # Paranóia.
if not trecho.horarios_sao_compativeis(trc, trc_pol): return False
# Todos compatíveis
return preco
def processa(ses, args):
usr_ses = None if ses == None else sessao.obtem_usuario(ses)
assert sessao.eh_administrador(
ses) # O dono da sessão deve ser administrador.
try:
if not 'id' in args:
pag = html_pag_mensagem_de_erro.gera(
ses, 'É necessário adicionar um ID para pesquisar.')
return pag
id = args['id']
if len(id) != 10:
raise ErroAtrib("O identificador \"" + id + "\" é inválido")
letra = id[0]
if letra == "U":
usr = usuario.busca_por_identificador(id)
if usr == None:
raise ErroAtrib("Não existe usuário com identificador " + id)
usr_atrs = usuario.obtem_atributos(usr)
usr_atrs['id_usuario'] = usuario.obtem_identificador(usr)
pag = html_pag_usuario.gera(ses, usr, usr_atrs, None)
elif letra == "C":
cpr = compra.busca_por_identificador(id)
if cpr == None:
raise ErroAtrib(
"Não existe pedido de compra com identificador" + id)
pag = html_pag_compra.gera(ses, cpr, None, None)
elif letra == "T":
trc = trecho.busca_por_identificador(id)
if trc == None:
raise ErroAtrib(
"Não existe trecho de viagem com identificador" + id)
pag = html_pag_trecho.gera(ses, trc, None, None)
elif letra == "S":
ses_a_ver = sessao.busca_por_identificador(id)
if ses_a_ver == None:
raise ErroAtrib("Não existe sessão com identificador" + id)
pag = html_pag_sessao.gera(ses, ses_a_ver, None)
elif letra == "A":
pol = poltrona.busca_por_identificador(id)
if pol == None:
raise ErroAtrib("Não existe poltrona com identificador" + id)
pag = html_pag_poltrona.gera(ses, pol, None, None)
else:
raise ErroAtrib("Classe de objeto \"" + letra + "\" inválida")
except ErroAtrib as ex:
erros = ex.args[0]
return html_pag_mensagem_de_erro.gera(ses, erros)
return pag
def processa(ses, args):
# Usuário deve estar logado para iniciar um pedido de compra
if ses == None:
pag = html_pag_mensagem_de_erro.gera(
ses, ["Voce deve estar logado para comprar um roteiro!"])
return pag
cpr = sessao.obtem_carrinho(ses)
if cpr == None:
# Novo carrinho associada ao usuario
compra.cria(ses.cliente, ses.cliente.nome_pass, ses.cliente.doc_pass)
# Recupera os objetos trechos para verificar poltronas
ids_trechos = args['ids_trechos'].split(',')
obj_trechos = []
for id in ids_trechos:
obj_trechos.append(trecho.busca_por_identificador(id))
# Verifica e inclui poltronas na compra
for trc in obj_trechos:
# Verifica se o trecho possui pelo menos uma poltrona livre
if trecho.numero_de_poltronas_livres(trc) < 1:
pag = html_pag_mensagem_de_erro.gera(ses, [
"Nao tem mais poltronas livres no trecho {} :(".format(
trecho.obtem_atributo(trc, "codigo"))
])
return pag
# Obtem lista de poltronas livres para comparacao do melhor preco
id_pol_livres = poltrona.lista_livres(trc)
melhor_preco = math.inf
id_melhor_preco = None
for id_pol in id_pol_livres:
pol = poltrona.busca_por_identificador(id_pol)
preco_pol = poltrona.obtem_atributo(pol, "preco")
# Atualiza melhor preco, assim como seu id
if preco_pol < melhor_preco:
melhor_preco = preco_pol
id_melhor_preco = id_pol
# Adiciona a poltrona ao carrinho
comando_comprar_poltrona.processa(ses,
{'id_poltrona': id_melhor_preco})
pag = comando_ver_carrinho.processa(ses, None)
return pag
def obtem_numeros_e_precos(ids_poltronas):
global cache, nome_tb, letra_tb, colunas, diags
assert isinstance(ids_poltronas, list)
nums_e_precos = []
for id_poltrona in ids_poltronas:
assert isinstance(id_poltrona, str)
pol = poltrona.busca_por_identificador(id_poltrona)
nums_e_precos.append((
poltrona.obtem_atributo(pol, 'numero'),
poltrona.obtem_atributo(pol, 'preco')
))
return nums_e_precos
def id_para_objeto(id):
letra = id[0]
if letra == "U":
obj = usuario.busca_por_identificador(id)
elif letra == "C":
obj = compra.busca_por_identificador(id)
elif letra == "S":
obj = sessao.busca_por_identificador(id)
elif letra == "A":
obj = poltrona.busca_por_identificador(id)
elif letra == "T":
obj = trecho.busca_por_identificador(id)
else:
erro_prog("identificador '" + id + " inválido")
return obj
def processa(ses, args):
# Obtem usuário e carrinho de compras:
assert ses != None # Comando não deveria ser acessível a usuário não logado.
usr_ses = sessao.obtem_usuario(ses)
assert usr_ses != None # Paranóia.
admin = usuario.obtem_atributo(usr_ses, 'administrador')
assert not admin # Admnistrador não deveria ter acesso a este cmd.
#Obtem a poltrona a comprar:
id_pol = args['id_poltrona'] if 'id_poltrona' in args else None
assert id_pol != None # Paranóia (formulário ou botão deve fornecer)
pol = poltrona.busca_por_identificador(id_pol)
assert pol != None # Paranóia.
# Obtém carrinho do usuário:
carr = sessao.obtem_carrinho(ses)
assert carr != None # Todo cliente comum deve ter carrinho.
id_carr = compra.obtem_identificador(carr)
try:
if not poltrona.pode_comprar(usr, pol, carr):
# Não deveria acontecer, mas em todo caso:
raise ErroAtrib("Esta poltrona não está disponível")
# Muda a poltrona para comprada
poltrona.muda_atributos(pol, { 'id_compra': id_carr })
# Mostra o carrinho do usuário com a poltrona comprada:
pag = html_pag_compra.gera(ses, cpr, excluir, None)
except ErroAtrib as ex:
erros = ex.args[0]
# Se o trecho da poltrona estiver disponível, mostra o trecho para outra compra:
id_trc = poltrona.obtem_atributo(pol, 'id_trecho')
assert id_trc != None # Paranoia.
trc = trecho.busca_por_identificador(id_trc)
assert trc != None # Paranoia.
if trecho.verificar_disponibilidade(trc) and trecho_eh_compativel(cpr, trc):
# Usuário pode comprar outra poltrona deste trecho:
pag = html_pag_trecho.gera(ses, trc, None, erros)
else:
# Usuário não pode comprar nenuma poltrona neste trecho.
# Volte para a página principal.
# !!! Deveria buscar e mostrar roteiros de mesma origem e destino !!!
pag = html_pag_principal.gera(ses, erros)
return pag
def processa(ses, args):
# Determina se usuario é administrador
assert sessao.eh_administrador(ses)
# Obtem a poltrona:
id_poltrona = args['id_poltrona']
assert id_poltrona != None # Paranóia (o formulário deve incluir).
del args['id_poltrona']
pol = poltrona.busca_por_identificador(id_poltrona)
assert pol != None # Paranóia.
pol_atrs = poltrona.obtem_atributos(pol)
id_trecho = args['id_trecho']
assert id_trecho == pol_atrs['id_trecho'] # Deve ser readonly no form.
id_compra = args['id_compra']
assert id_compra == pol_atrs['id_compra'] # Deve ser readonly no form.
numero = args['numero']
assert numero == pol_atrs['numero'] # Deve ser readonly no form.
# Monta dicionário com alterações:
atrs_a_mudar = {}.copy()
oferta = args['oferta'] if 'oferta' in args else None # Alterável.
if oferta != None and type(oferta) is str: oferta = (oferta == "on")
if oferta != None: atrs_a_mudar['oferta'] = oferta
preco = args['preco'] if 'preco' in args else None # Alterável.
if preco != None and type(preco) is str: preco = float(preco)
if preco != None: atrs_a_mudar['preco'] = preco
# Editar poltrona
try:
poltrona.muda_atributos(pol, atrs_a_mudar)
# Mostra poltrona com dados alterados:
pag = html_pag_poltrona.gera(ses, pol, None, None)
except ErroAtrib as ex:
# Mostra novamente a poltrona com mesmos args e mens de erro:
erros = ex[0]
pag = html_pag_poltrona.gera(ses, pol, args, erros)
return pag
def processa(ses, args):
# Obtem o dono da sessão {ses}:
usr_ses = None if ses == None else sessao.obtem_usuario(ses)
assert usr_ses != None # Usuário não logado não deveria ter acesso a este comando.
admin = False if usr_ses == None else usuario.obtem_atributo(usr_ses, 'administrador')
carr = None if ses == None else sessao.obtem_carrinho(ses) # Carrinho de compras da sessão.
# Obtem a poltrona a excluir:
id_pol = args['id_poltrona']
pol = None if id_pol == None else poltrona.busca_por_identificador(id_pol)
assert pol != None # A poltrona deve estar identificada em {args}.
del args['id_poltrona']
# Obtem a compra de onde excluir:
id_cpr_pol = poltrona.obtem_atributo(pol, 'id_compra')
cpr_pol = None if id_cpr_pol == None else compra.busca_por_identificador(id_cpr_pol)
assert cpr_pol != None # A poltrona não pode estar livre.
# Obtem o dono da compra:
usr_pol = compra.obtem_cliente(cpr_pol)
assert usr_pol != None # Paranoia.
# Verifica permissão:
assert admin or (usr_pol == usr_ses) # Outro usuário não deveria ter acesso a este comando.
try:
# Ao definir {id_compra} como {None}, estamos excluindo a poltrona
poltrona.muda_atributos(pol, {'id_compra': None})
if (not admin) and (cpr_pol != carr):
# Mudamos o carrinho:
sessao.muda_atributos(ses, {'carrinho': cpr_pol})
aviso = ("Seu carrinho agora é a compra %s" % id_cpr_pol)
else:
aviso = None
# Gera a página de retorno:
pag = html_pag_compra.gera(ses, cpr_pol, None, aviso)
except ErroAtrib as ex:
erros = ex[0]
# Mostra a página da poltrona com mesmos args e erro:
pag = html_pag_compra.gera(ses, cpr_pol, args, aviso)
return pag
def verificar_baldeacao(cpr):
# pega poltronas de compra, ordenadas por partida
ids_poltronas = obtem_poltronas(cpr)
obj_poltronas = []
ultima_chegada = None
ultimo_destino = None
poltronas_invalidas = []
for id_poltrona in ids_poltronas:
# lendo poltrona atual
obj_poltrona = poltrona.busca_por_identificador(id_poltrona)
partida = poltrona.obtem_dia_e_hora_de_partida(obj_poltrona)
chegada = poltrona.obtem_dia_e_hora_de_chegada(obj_poltrona)
origem, destino = poltrona.obtem_origem_destino(obj_poltrona)
# convertendo string para data
formato = "%Y-%m-%d %H:%M UTC"
partida = datetime.datetime.strptime(partida, formato)
chegada = datetime.datetime.strptime(chegada, formato)
# baldeacoes
baldeacao_curta = datetime.timedelta(minutes=15)
baldeacao_longa = datetime.timedelta(hours=2)
# comparando com última poltrona
if ultima_chegada is not None and ultimo_destino is not None:
delta = partida - ultima_chegada
# mudança de aeroporto, tempo de baldeação é de 2 horas
if origem != ultimo_destino:
if delta < baldeacao_longa:
poltronas_invalidas.append(id_poltrona)
# mesmo aeroporto, tempo de baldeação é de 15 minutos
else:
if delta < baldeacao_curta:
poltronas_invalidas.append(id_poltrona)
ultima_chegada = chegada
ultimo_destino = destino
return poltronas_invalidas
def processa(ses, args):
assert ses != None and sessao.eh_administrador(ses) # Paranóia.
# Obtem a poltrona a fazer cehckin:
id_pol = args['id_poltrona'] if 'id_poltrona' in args else None
assert id_pol != None # Paranoia (formulário deve incluir este dado)
pol = poltrona.busca_por_identificador(id_pol)
# A poltrona deve estar reservada para alguma compra:
assert poltrona.obtem_atributo(pol, 'id_compra') != None
try:
poltrona.muda_atributos(pol, {'fez_checkin': True})
pag = html_pag_poltrona.gera(ses, pol, None, None)
except ErroAtrib as ex:
erros = ex[0]
# Mostra os dados da poltrona:
pag = html_pag_poltrona.gera(ses, pol, None, erros)
return pag
def poltronas_abertas(usr):
id_usr = usuario.obtem_identificador(usr)
abertas_usr = [].copy()
ids_compras = compra.busca_por_cliente(id_usr)
for id_cpr in ids_compras:
cpr = compra.busca_por_identificador(id_cpr)
assert cpr != None # Paranóia.
# Pega o id das poltronas de cada uma dessas compras
ids_poltronas = compra.obtem_poltronas(cpr)
for id_pol in ids_poltronas:
# Para cada um desses ids pega o objeto da poltrona
pol = poltrona.busca_por_identificador(id_pol)
assert pol != None # Paranóia.
assert poltrona.obtem_atributo(pol,
'id_compra') == id_cpr # Paranóia.
# Verifica se o trecho não foi encerrado
id_trc = poltrona.obtem_atributo(pol, 'id_trecho')
trc = trecho.busca_por_identificador(id_trc)
if not trecho.obtem_atributo(trc, 'encerrado'):
abertas_usr.append(pol)
return abertas_usr
def livre_mais_proxima(pol, preco_max):
global cache, nome_tb, letra_tb, colunas, diags
dist_pol_desejada = -1
id_trc = poltrona.obtem_atributo(pol, 'id_trecho')
trc = trecho.busca_por_identificador(id_trc)
ids_livres = lista_livres(trc)
id_pol = poltrona.obtem_identificador(pol)
pol_desejada = None
if len(ids_livres) != 0:
for id in ids_livres:
if (id != id_pol):
pol_aux = poltrona.busca_por_identificador(id)
dist_aux = distancia(pol, pol_aux)
preco_aux = obtem_atributo(pol_aux, 'preco')
if (dist_pol_desejada == -1 and preco_aux <= preco_max):
pol_desejada = pol_aux
dist_pol_desejada = dist_aux
else:
if (dist_aux < dist_pol_desejada and preco_aux <= preco_max):
pol_desejada = pol_aux
dist_pol_desejada = dist_aux
return pol_desejada
def resumo_de_trafego(ids_trechos):
global cache, nome_tb, letra_tb, colunas, diags
custo_total = 0
total_checkins = 0
total_poltronas = 0
total_poltronas_pagas = 0
for id in ids_trechos:
trc = busca_por_identificador(id)
total_poltronas += numero_de_poltronas(trc)
total_poltronas_pagas += (total_poltronas -
numero_de_poltronas_livres(trc))
poltronas = obtem_poltronas(trc)
for id_poltrona in poltronas:
pol = poltrona.busca_por_identificador(id_poltrona)
custo_total += poltrona.obtem_atributo(pol, 'preco')
if poltrona.obtem_atributo(pol, 'fez_checkin'):
total_checkins += 1
#O resultado é uma tupla {(ntr, npol_tot, npol_pag, renda_tot, npol_chk)}
resumo = (len(ids_trechos), total_poltronas, total_poltronas_pagas,
custo_total, total_checkins)
return resumo
def gera(ids_poltronas):
linhas = [].copy()
# Cabeçalho:
cabs_raw = [
'Trecho', 'Origem', 'Partida', 'Destino', 'Chegada', 'Poltrona',
'Preço', 'Ver Poltrona'
]
cabs_div = [].copy()
for cb in cabs_raw:
cabs_div.append(
html_div.gera(html_estilo_cabecalho_de_tabela.gera(), cb))
for id_poltrona in ids_poltronas:
pol = poltrona.busca_por_identificador(id_poltrona)
linha = html_resumo_de_poltrona_de_usuario.gera(pol)
assert type(linha) is list or type(linha) is tuple
linhas.append(linha)
ht_res = html_table.gera(linhas, cabs_div)
return ht_res
def gera(cpr, editavel, texto_bt, comando_bt):
# Obtem atributos a mostrar:
valores = {}.copy()
id_cpr = compra.obtem_identificador(cpr)
atrs_cpr = compra.obtem_atributos(cpr)
# Atributos da compra em si
valores['id_cpr'] = id_cpr
valores['status'] = atrs_cpr['status']
valores['nome_pass'] = atrs_cpr['nome_pass']
valores['doc_pass'] = atrs_cpr['doc_pass']
valores['preco_tot'] = ("%.2f" % compra.calcula_preco(cpr))
# valores['pagamento'] = atrs_cpr['pagamento']
# Cliente que está montando ou montou a compra:
usr = compra.obtem_cliente(cpr)
valores['id_usr'] = usuario.obtem_identificador(usr)
atrs_usr = usuario.obtem_atributos(usr)
valores['nome_usr'] = atrs_usr['nome']
# Bilhetes da compra:
ids_pols = compra.obtem_poltronas(cpr)
num_trechos = len(ids_pols)
valores['n_trechos'] = str(num_trechos)
if (num_trechos >= 1):
# Obtém origem, data, e hora de partida do primeiro trecho:
pol_ini = poltrona.busca_por_identificador(ids_pols[0])
id_trc_ini = poltrona.obtem_atributo(pol_ini, 'id_trecho')
trc_ini = trecho.busca_por_identificador(id_trc_ini)
origem = trecho.obtem_atributo(trc_ini, 'origem')
dh_partida = trecho.obtem_dia_e_hora_de_partida(trc_ini)
valores['partida'] = origem + " " + dh_partida
# Obtém destino, data, e hora de chegada do último trecho:
pol_fin = poltrona.busca_por_identificador(ids_pols[-1])
id_trc_fin = poltrona.obtem_atributo(pol_fin, 'id_trecho')
trc_fin = trecho.busca_por_identificador(id_trc_fin)
destino = trecho.obtem_atributo(trc_fin, 'destino')
dh_chegada = trecho.obtem_dia_e_hora_de_chegada(trc_fin)
valores['chegada'] = destino + " " + dh_chegada
# Linhas para {html_table.gera}:
linhas = (
html_cpr_campo("Compra", 'id_cpr', valores, 'text', None, False),
html_cpr_campo("Cliente", 'id_usr', valores, 'text', None, False),
html_cpr_campo("Nome do cliente", 'nome_usr', valores, 'text', None,
False),
html_cpr_campo("Nome do passageiro", 'nome_pass', valores, 'text',
"Fulano da Silva", editavel),
html_cpr_campo("Documento do passageiro", 'doc_pass', valores, 'text',
"123.456.789-10", editavel),
html_cpr_campo("Número de trechos", 'n_trechos', valores, 'text', None,
False),
html_cpr_campo("Preço total", 'preco_tot', valores, 'text', None,
False),
html_cpr_campo("Estado", 'status', valores, 'text', None, False),
html_cpr_campo("Partida", 'partida', valores, 'text', None, False),
html_cpr_campo("Chegada", 'chegada', valores, 'text', None, False),
)
ht_campos = html_table.gera(linhas, ["", ""])
# Botões:
if editavel:
args_submit = {
'id_compra': id_cpr
} # Argumentos adicionais para submissão.
ht_submit = html_botao_submit.gera(texto_bt, comando_bt, args_submit,
"#44ff44")
ht_campos += "<br/>" + ht_submit
return html_form.gera(ht_campos)
modulo = html_form_dados_de_poltrona
funcao = modulo.gera
frag = True # {True} se for apenas um fragmento HTML, {False} se for página completa.
pretty = False # Se {True}, formata HTML para legibilidate (mas introduz brancos nos textos).
utils_testes.testa_gera_html(modulo, funcao, rotulo, frag, pretty, *args)
ht_submit = ""
ht_submit += html_botao_submit.gera("Escovar", "escovar_poltrona", None,
"#ffff00")
ht_submit += html_botao_simples.gera("Repensar", "repensar_a_vida",
{'coisa': "418"}, "#00cc00")
pol1_id = "A-00000001"
pol1 = poltrona.busca_por_identificador(pol1_id)
pol1_atrs = {}
pol1_id_cpr = None
testa("pol1_alterar_vazio", pol1_id, pol1_atrs, True, ht_submit)
pol2_id = "A-00000007"
pol2 = poltrona.busca_por_identificador(pol2_id)
pol2_atrs = {'preco': 23.45, 'id_trecho': "T-00000231", 'numero': "99F"}
assert poltrona.obtem_atributo(pol2, 'id_compra') == None
pol2_id_cpr = None
testa("pol2_alterar_alguns", pol2_id, pol2_atrs, True, ht_submit)
pol3_id = "A-00000003"
pol3 = poltrona.busca_por_identificador(pol3_id)
pol3_atrs = poltrona.obtem_atributos(pol3)
pol3_id_cpr = poltrona.obtem_atributo(pol3, 'id_compra')
cpr1_id = "C-00000001"
cpr1 = compra.busca_por_identificador(cpr1_id)
assert cpr1 != None
cpr1_atrs = compra.obtem_atributos(cpr1)
compra.verifica(cpr1, cpr1_id, cpr1_atrs)
assert cpr1 == tabelas.id_para_objeto(cpr1_id)
sys.stderr.write("\n")
# ----------------------------------------------------------------------
sys.stderr.write("verificando tabela \"poltronas\" ptr = %s\n" % "ptr1")
ptr1_id = "A-00000001"
ptr1 = poltrona.busca_por_identificador(ptr1_id)
assert ptr1 != None
ptr1_atrs = poltrona.obtem_atributos(ptr1)
poltrona.verifica(ptr1, ptr1_id, ptr1_atrs)
assert ptr1 == tabelas.id_para_objeto(ptr1_id)
sys.stderr.write("\n")
# ----------------------------------------------------------------------
sys.stderr.write("verificando tabela \"trechos\" trc = %s\n" % "trc1")
trc1_id = "T-00000001"
trc1 = trecho.busca_por_identificador(trc1_id)
assert trc1 != None
trc1_atrs = trecho.obtem_atributos(trc1)
import utils_testes
import sys
sys.stderr.write("Conectando com base de dados...\n")
res = base_sql.conecta("DB", None, None)
assert res == None
sys.stderr.write("Criando alguns objetos...\n")
tabelas.cria_todos_os_testes(False)
def testa(rotulo, *args):
"""Testa {funcao(*args)}, grava resultado
em "testes/saida/{modulo}.{funcao}.{rotulo}.html"."""
modulo = html_resumo_de_poltrona_de_usuario
funcao = modulo.gera
frag = True # {True} se for apenas um fragmento HTML, {False} se for página completa.
pretty = False # Se {True}, formata HTML para legibilidate (mas introduz brancos nos textos).
utils_testes.testa_gera_html(modulo, funcao, rotulo, frag, pretty, *args)
testes = ( \
( "teste-um", "A-00000001"),
( "teste-dois", "A-00000003"),
( "teste-tres", "A-00000006"),
( "teste-quatro", "A-00000008"),
)
for rot, id_pol in testes:
pol = poltrona.busca_por_identificador(id_pol)
assert pol != None
testa(rot, pol)
# Sessão do cliente 1:
sesC = sessao.busca_por_identificador("S-00000001")
assert sessao.obtem_usuario(sesC) == usrC
# Sessão de outro cliente comum:
sesO = sessao.busca_por_identificador("S-00000003")
assert sessao.obtem_usuario(sesO) != None
assert sessao.obtem_usuario(sesO) != usrC
# Sessão de administrador:
sesA = sessao.busca_por_identificador("S-00000004")
assert sessao.eh_administrador(sesA)
# Poltrona reservada pelo usuario 1:
polR_id = "A-00000001"
polR = poltrona.busca_por_identificador(polR_id)
assert poltrona.obtem_atributo(polR, 'id_compra') == "C-00000001"
# Poltrona livre:
polL_id = "A-00000005"
polL = poltrona.busca_por_identificador(polL_id)
assert poltrona.obtem_atributo(polL, 'id_compra') == None
# Atributos a alterar:
atrs_m = {}.copy()
for ses_tag, ses in (("sesN", None), ("sesC", sesC), ("sesO", sesO), ("sesA",
sesA)):
for pol_tag, pol in (("polL", polL), ("polR", polR)):
for err_tag, err in (("err0", None), ("err1",
"Das schibunga is kaput")):
funcao = modulo.processa
frag = False # {True} se for apenas um fragmento HTML, {False} se for página completa.
pretty = False # Se {True}, formata HTML para legibilidate (mas introduz brancos nos textos).
testa_gera_html(modulo, funcao, rotulo, frag, pretty, *args)
# Sessao de usuário comum e poltrona desse usuário:
ses_com1 = sessao.busca_por_identificador("S-00000003")
usr_com1 = sessao.obtem_usuario(ses_com1)
assert not usuario.obtem_atributo(usr_com1, 'administrador')
id_cpr_com1 = "C-00000003"
cpr_com1 = compra.busca_por_identificador(id_cpr_com1)
assert compra.obtem_cliente(cpr_com1) == usr_com1
id_pol_com1 = "A-00000008"
pol_com1 = poltrona.busca_por_identificador(id_pol_com1)
assert poltrona.obtem_atributo(pol_com1, "id_compra") == id_cpr_com1
testa("com1-com1", ses_com1, {'id_poltrona': id_pol_com1})
# Sessao de administrador e poltrona de outro usuário:
ses_adm3 = sessao.busca_por_identificador("S-00000004")
usr_adm3 = sessao.obtem_usuario(ses_adm3)
assert usuario.obtem_atributo(usr_adm3, 'administrador')
id_cpr_com2 = "C-00000001"
cpr_com2 = compra.busca_por_identificador(id_cpr_com2)
usr_com2 = compra.obtem_cliente(cpr_com2)
assert not usuario.obtem_atributo(usr_com2, 'administrador')
id_pol_com2 = "A-00000006"
pol_com2 = poltrona.busca_por_identificador(id_pol_com2)
def gera(ses, trc, atrs, erros):
# Obtem usuário da sessão, determina privilégios:
usr_ses = None if ses == None else sessao.obtem_usuario(ses)
admin = sessao.eh_administrador(ses)
if atrs == None: atrs = {}.copy() # Por via das dúvidas.
# Botões de acão:
ht_submit = ""
if trc == None:
# Acrescentando novo trecho:
id_trc = None
novo = True
# O submit é "Acrescentar":
ht_submit += html_botao_submit.gera("Acrescentar",
"acrescentar_trecho", None,
"#ff0000")
else:
# Visualização/alteração de trecho existente:
assert type(trc) is trecho.Objeto_Trecho
novo = False
id_trc = trecho.obtem_identificador(trc)
assert id_trc != None # Paranóia.
# Completa {atrs} com atributos de {trc}:
atrs_trc = trecho.obtem_atributos(trc)
assert atrs_trc != None # Paranóia
for ch, val in atrs_trc.items():
if not ch in atrs: atrs[ch] = val
# Botoes de ação:
if admin:
# O submit é "Alterar":
ht_submit += html_botao_submit.gera("Alterar", "alterar_trecho",
None, "#ff0000")
# Tem tamém botão "Clonar":
ht_submit += html_botao_simples.gera("Clonar", "clonar_trecho",
{'id_trecho': id_trc},
"#ff0000")
# Constrói formulário com dados do trecho:
ht_dados_trecho = html_form_dados_de_trecho.gera(id_trc, atrs, admin,
ht_submit)
# Constrói a lista de poltronas do trecho:
if novo:
ht_dados_poltronas = ""
else:
# Formata a lista de poltronas:
ids_pols_todas = poltrona.busca_por_trecho(trc)
if admin:
# Mostra todas as poltronas do trecho:
ids_pols_ver = ids_pols_todas
else:
# Mostra só as poltronas disponíveis ou reservadas para compras do usuário {usr}:
trc_encerrado = atrs_trc['encerrado']
ids_pols_ver = [].copy()
for id_pol in ids_pols_todas:
assert id_pol != None # Paranóia.
pol = poltrona.busca_por_identificador(id_pol)
assert pol != None # Paranóia.
id_cpr_pol = poltrona.obtem_atributo(pol, 'id_compra')
cpr_pol = None if id_cpr_pol == None else compra.busca_por_identificador(
id_cpr_pol)
usr_pol = None if cpr_pol == None else compra.obtem_cliente(
cpr_pol)
# Verifica relevância de {pol} para o usuário:
livre = (cpr_pol == None)
if (livre and (not trc_encerrado)) or ((usr_ses != None) and
(usr_pol == usr_ses)):
ids_pols_ver.append(id_pol)
if len(ids_pols_ver) == 0:
# Não há poltronas a mostrar.
estilo_aviso = "color:red;font-size:20;"
ht_dados_poltronas = "<br/>" + html_span.gera(
estilo_aviso, "Não há poltronas disponíveis")
else:
# Formata a lista de poltronas:
fazer_checkin = admin # Deve ter botão de "Checkin" em cada poltrona?
# sys.stderr.write(" ids_pols = %s\n" % str(ids_pols))
# sys.stderr.write(" id_trc = %s\n" % str(id_trc))
carr = None if ses == None else sessao.obtem_carrinho(ses)
ht_dados_poltronas = html_lista_de_poltronas_de_trecho.gera(
ids_pols_ver, usr_ses, carr)
# Botão "Voltar":
ht_bt_cancel = html_botao_simples.gera("Voltar", 'principal', None,
"#ff2200")
# Monta a página:
ht_conteudo = \
ht_dados_trecho + "<br/>" + \
ht_dados_poltronas + "<br/>" + \
ht_bt_cancel
pagina = html_pag_generica.gera(ses, ht_conteudo, erros)
return pagina
