def add_school():
if current_user.super_admin:
form = SelectForm()
form.school_list.choices = DB.list_school_form()
if form.submit2.data:
DB.add_school(form.name.data, form.surname.data)
return redirect(url_for("add_school"))
if form.submit.data:
DB.delete_school(form.school_list.data)
return redirect(url_for("add_school"))
return render_template("school.html", form=form, rows=DB.school_list())
else:
return redirect(url_for("index"))
