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