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')