def add_schedule(): if current_user.admin: form = SelectForm() form.class_list.choices = DB.class_list_form() if form.submit2.data: DB.add_schedule(form.class_list.data) return redirect( url_for("add_schedule2", class_id=form.class_list.data)) return render_template("add_schedule1.html", form=form) else: return redirect(url_for("index"))