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