def new_guest():
form = GuestForm(request.form)
form.city.choices = [(city.id, city.desc)
for city in db.session.query(City).all()]
if request.method == 'POST' and form.validate():
guest = form.getObj(Guest())
guest.user_id = current_user.id
guest.picture = 'picna.jpg'
db.session.add(guest)
db.session.commit()
flash('Anúncio adicionado'.decode('utf-8'), 'success')
return redirect(url_for('dashboard'))
return render_template('guest/edit_guest.html',
form=form,
action='Adicionar')
def edit_guest(id):
guest = db.session.query(Guest).filter(Guest.id == id).one()
form = GuestForm(request.form, guest)
form.city.choices = [(city.id, city.desc)
for city in db.session.query(City).all()]
if request.method == 'GET':
city = City.query.get(guest.city_id)
form.city.choices = [(city.id, city.desc)
for city in db.session.query(City).filter(
City.state == city.state).all()]
form.populateForm(guest)
if request.method == 'POST' and form.validate():
guest = form.getObj(guest)
db.session.commit()
flash('Anúncio Atualizado'.decode('utf-8'), 'success')
return redirect(url_for('dashboard'))
return render_template('guest/edit_guest.html', form=form, action='Editar')
