def dodaj(): """Dodawanie pytań i odpowiedzi""" form = DodajForm() if form.validate_on_submit(): odp = form.odpowiedzi.data p = Pytanie(pytanie=form.pytanie.data, odpok=odp[int(form.odpok.data)]) p.save() for o in odp: inst = Odpowiedz(pnr=p.id, odpowiedz=o) inst.save() flash("Dodano pytanie: {}".format(form.pytanie.data)) return redirect(url_for("lista")) elif request.method == 'POST': flash_errors(form) return render_template("dodaj.html", form=form, radio=list(form.odpok))
def edytuj(pid): """Edycja pytania o identyfikatorze pid i odpowiedzi""" p = get_or_404(pid) form = DodajForm() if form.validate_on_submit(): odp = form.odpowiedzi.data p.pytanie = form.pytanie.data p.odpok = odp[int(form.odpok.data)] p.save() for i, o in enumerate(p.odpowiedzi): o.odpowiedz = odp[i] o.save() flash("Zaktualizowano pytanie: {}".format(form.pytanie.data)) return redirect(url_for("lista")) elif request.method == 'POST': flash_errors(form) for i in range(3): if p.odpok == p.odpowiedzi[i].odpowiedz: p.odpok = i break form = DodajForm(obj=p) return render_template("edytuj.html", form=form, radio=list(form.odpok))