def add_class(): if current_user.admin: form = SelectForm() form.school_list.choices = DB.list_school_form() form.class_list.choices = DB.class_list_form() if form.submit.data: DB.add_class(form.school_list.data, form.name.data) return redirect(url_for("add_class")) if form.submit2.data: DB.delete_class(form.class_list.data) return redirect(url_for("add_class")) return render_template("add_class.html", form=form, rows=DB.class_list()) else: return redirect(url_for("index"))