def test_pesquisa_sala():
'''Testa se a função DAO.pesquisa_sala retorna a sala a partir do
nome
'''
result = Sala(nome='sala 02', lotacao=15)
DAO.create(result)
assert result == DAO.pesquisa_sala('sala 02')
def test_pesquisa_pessoa():
'''Testa se a função DAO.pesquisa_pessoa retorna a pessoa a partir
do id
'''
pessoa = Pessoa(nome='pessoa 01', sobrenome='teste')
DAO.create(pessoa)
pessoa_id = pessoa.id
result = Pessoa.query.filter_by(nome='pessoa 01').first()
assert result == DAO.pesquisa_pessoa(pessoa_id)
def cadastro_cafe():
'''Tela de cadastro de salas de café
Página com o formulário para cadastro de salas de café.
'''
pesquisa = FormPesquisa()
form = FormCafe()
if form.validate_on_submit():
DAO.cadastrar_salacafe(form)
return redirect('/index')
return render_template('cadastroCafe.html', form=form, pesquisa=pesquisa)
def cadastro_pessoa():
'''Tela de cadastro de pessoas
Página com o formulário para cadstro de pessoas.
'''
pesquisa = FormPesquisa()
form = FormPessoa()
if form.validate_on_submit():
DAO.cadastrar_pessoa(form)
return redirect('/')
return render_template('cadastroPessoa.html', form=form, pesquisa=pesquisa)
def index():
'''Home page
Página inicial contendo a lista de todas as pessoas, botão para
cadastro de pessoas, botão para cadastro de salas, formulário para
pesquisar salas e uma lista com todas as pessoas, com um botão para
visualizar a pessoa, ao lado.
'''
pesquisa = FormPesquisa()
if pesquisa.validate_on_submit():
return redirect('verSala/' + pesquisa.nomepesquisa.data)
DAO.organizar_pessoas()
pessoas = DAO.busca_pessoas()
return render_template('index.html', pesquisa=pesquisa, pessoas=pessoas)
def ver_pessoa(id):
'''Tela que mostra as informações da pessoa
Carrega as salas que a pessoa vai ficar na etapa 1 e na etapa 2
Parrameters
-----------
id : int
número do id da pessoa
'''
pessoa = DAO.pesquisa_pessoa(id)
pesquisa = FormPesquisa()
return render_template('pessoa.html', pessoa=pessoa, pesquisa=pesquisa)
def ver_sala(sala):
'''Tela com as informações da sala
Página contendo duas tabelas, uma para cada etapa, com as pessoas
que estarão na sala em cada uma das etapas.
Parameters
----------
sala : str
nome da sala
'''
pesquisa = FormPesquisa()
sala = DAO.pesquisa_sala(sala)
pessoas1 = sala.pessoas1
pessoas2 = sala.pessoas2
return render_template('sala.html',
sala=sala,
pessoas1=pessoas1,
pessoas2=pessoas2,
pesquisa=pesquisa)
def test_organizar_pessoas():
'''Testa de a função DAO.organizar_pessoas separa as pessoas em suas
salas e altera na segunda etapa
'''
sala3 = Sala(nome="sala cafe 01")
DAO.create(sala3)
sala4 = Sala(nome="sala cafe 02")
DAO.create(sala4)
DAO.organizar_pessoas()
pessoa_result = Pessoa(nome='pessoa_result', sobrenome='test')
salas = Sala.query.filter_by(lotacao=15)
pessoa_result.pessoas1.append(salas[0])
pessoa_result.pessoas2.append(salas[1])
pessoa_result.pessoas1.append(sala3)
pessoa_result.pessoas2.append(sala4)
pessoa = Pessoa.query.first()
assert pessoa.pessoas1 == pessoa_result.pessoas1 and pessoa.pessoas2\
== pessoa_result.pessoas2
from app import app
from flask import render_template, redirect
from app.model.forms import FormCafe, FormPessoa, FormSala, FormPesquisa
from app.model.dao import DAO
from app.model.models import Pessoa, Sala
DAO = DAO()
@app.route('/', methods=['get', 'post'])
@app.route('/index', methods=['get', 'post'])
def index():
'''Home page
Página inicial contendo a lista de todas as pessoas, botão para
cadastro de pessoas, botão para cadastro de salas, formulário para
pesquisar salas e uma lista com todas as pessoas, com um botão para
visualizar a pessoa, ao lado.
'''
pesquisa = FormPesquisa()
if pesquisa.validate_on_submit():
return redirect('verSala/' + pesquisa.nomepesquisa.data)
DAO.organizar_pessoas()
pessoas = DAO.busca_pessoas()
return render_template('index.html', pesquisa=pesquisa, pessoas=pessoas)
@app.route('/cadastroPessoa', methods=['get', 'post'])
def cadastro_pessoa():
'''Tela de cadastro de pessoas
def test_create():
'''Testa se a função DAO.create() adiciona um novo objeto no banco
'''
result = Sala(nome='sala 01', lotacao=15)
DAO.create(result)
assert result == Sala.query.filter_by(nome='sala 01').first()
def test_busca_pessoas():
'''Testa se a função DAO.pesquisa_pessoa retorna todas as pessoas do
banco
'''
result = Pessoa.query.all()
assert DAO.busca_pessoas() == result