def cafe_edit(cafe_id): """ show form for editting cafe and handle form""" cafe = Cafe.query.get(cafe_id) form = AddEditCafeForm(obj=cafe) #get city_code choice from current cities in db form.city_code.choices = City.get_all_cities() if form.validate_on_submit(): cafe.name = form.name.data cafe.description = form.description.data cafe.url = form.url.data cafe.address = form.address.data cafe.city_code = form.city_code.data cafe.image_url = form.image_url.data db.session.commit() flash(f'{cafe.name} edited') return redirect(f"/cafes/{cafe.id}") return render_template("cafe/edit-form.html", form=form, cafe=cafe) ####################################### # users @app.route('/signup', methods=["GET", "POST"]) def user_signup(): """return sign up form and handle sign up """ form = SignupForm() if form.validate_on_submit(): new_user = User.register(username=form.username.data, email=form.email.data, first_name=form.first_name.data, last_name=form.last_name.data, description=form.description.data, image_url=form.image_url.data, password=form.password.data) db.session.add(new_user) db.session.commit() do_login(new_user) flash(f"You are signed up and logged in") return redirect("/cafes") return render_template("auth/signup-form.html", form=form)
def cafe_add(): """ show form for adding cafe and handle form""" form = AddEditCafeForm() #get city_code choice from current cities in db form.city_code.choices = City.get_all_cities() if form.validate_on_submit(): new_cafe = Cafe(name=form.name.data, description=form.description.data, url=form.url.data, address=form.address.data, city_code=form.city_code.data, image_url=form.image_url.data or None) db.session.add(new_cafe) db.session.commit() flash(f'{new_cafe.name} added') return redirect(f"/cafes/{new_cafe.id}") return render_template("cafe/add-form.html", form=form)