def editPSet(pset_id): pset = db_session.query(ProblemSet).get(pset_id) course = db_session.query(Course).get(pset.course_id) if None in [pset,course]: abort(404) form = ProblemSetForm(request.form) if request.method == "POST" and form.validate(): pset.title = form.title.data pset.opens = form.opens.data pset.due = form.due.data db_session.commit() return redirect(url_for('showPSet',course_id=course.id,pset_id=pset.id)) return render_template("editpset.html",course=course,pset=pset)
def newPSet(course_id): course = db_session.query(Course).get(course_id) if course is None: abort(404) form = ProblemSetForm(request.form) if request.method == "POST" and form.validate(): pset = ProblemSet() pset.title = form.title.data pset.opens = form.opens.data pset.due = form.due.data pset.course_id=course.id course.problemsets.append(pset) db_session.commit() return redirect(url_for("showCourse", course_id=course.id)) return render_template("newpset.html", course=course, form=form)