ice_breaking_challenge/quiz.py

from flask import Blueprint, render_template, session
from flask import redirect, url_for, session

from ice_breaking_challenge import quiz_generated, quiz_results

bp = Blueprint("quiz", __name__, url_prefix="/quiz")

@bp.route("/", methods=["GET", "POST"])
def quiz():
    """quiz"""
    quizzes = quiz_results[session.sid]
    if 'question_number' not in session:
        session['question_number'] = 0
    else:
        session['question_number'] += 1

    if session['question_number'] == 2: # TODO:
        return redirect(url_for("quiz.finish"))

    print(session['question_number'])
    quiz = quizzes[session['question_number']]

    return render_template("quiz.html", quiz=quiz)


@bp.route("/finish", methods=["GET"])
def finish():
    global quiz_generated, quiz_results
    session_id = session.sid
    del quiz_generated[session_id]
    del quiz_results[session_id]
    return render_template("finish.html")