def new_room():
form = RoomForm(request.form)
form.city.choices = [(city.id, city.desc)
for city in db.session.query(City).all()]
if request.method == 'POST' and form.validate():
room = form.getObj(Room())
room.user_id = current_user.id
db.session.add(room)
db.session.commit()
path = "rooms/room" + str(room.id)
os.mkdir(path)
flash('Anúncio adicionado'.decode('utf-8'), 'success')
return redirect(url_for('dashboard'))
return render_template('room/edit_room.html',
form=form,
action='Adicionar')
def edit_room(id):
room = db.session.query(Room).filter(Room.id == id).one()
form = RoomForm(request.form, room)
form.city.choices = [(city.id, city.desc)
for city in db.session.query(City).all()]
if request.method == 'GET':
city = City.query.get(room.city_id)
form.city.choices = [(city.id, city.desc)
for city in db.session.query(City).filter(
City.state == city.state).all()]
form.populateForm(room)
if request.method == 'POST' and form.validate():
room = form.getObj(room)
db.session.commit()
flash('Anúncio Atualizado'.decode('utf-8'), 'success')
return redirect(url_for('dashboard'))
return render_template('room/edit_room.html', form=form, action='Editar')
