1 files changed, 100 insertions, 0 deletions
diff --git a/routes/answer.py b/routes/answer.py
new file mode 100644
index 0000000..fa3d138
--- /dev/null
+++ b/routes/answer.py
@@ -0,0 +1,100 @@
+from app import app
+from flask import render_template, session, request, redirect
+import db_actions as D
+from routes.tools import rows2dicts, get_alert, get_nick, red
+
+
+@app.route("/pages/new_answer.html")
+def new_answer():
+    if "id" not in session:
+        return red["nick"]
+    return render_template("new_answer.html", 
+            alert=get_alert(),
+            nick=get_nick()
+        )
+
+@app.route("/kys/<link>")
+def kys_link(link):
+    if aid := D.find_quiz_by_link( link ):
+        session["answer_id"] = aid
+        return redirect("/#answer")
+    return redirect("/")
+
+@app.route("/set/answer_id",methods=["POST"])
+def answer_id():
+    if "id" not in session:
+        session["alert"] = "Nimimerkkiä ei ole asetettu."
+        return redirect("/#nick")
+    else:
+        sid = session["id"]
+
+    if "next" not in request.form:
+        next = "/#answer"
+    else:
+        next = "/#"+request.form["next"]
+    
+    if "link" not in request.form or request.form["link"]=="":
+        session["alert"] = "Kyselmän nimeä ei ole annettu."
+        return redirect(next)
+        
+    if aid := D.find_quiz_by_link( request.form["link"] ):
+        session["answer_id"] = aid
+    else:
+        session["alert"] = "Koodilla ei löytynyt kyselmää"
+        return redirect(next)
+        
+    if next == "/#analyse" and not D.is_user_answered( aid, sid ):
+        session["alert"] = "Et ole vielä vastannut tähän kyselmään. \
+                            Voit tutkia vastaksia vastattuasi."
+        return redirect("/#answer")
+
+    return redirect( next )
+
+@app.route("/pages/answer.html")
+def answer():
+    if "id" in session:
+        sid = session["id"]
+    else:
+        return red["nick"]
+        
+    if "answer_id" in session:
+        aid = session["answer_id"]
+    else:
+        return red["new_answer"]
+        
+    if D.is_user_answered(aid, sid):
+        return red["new_answer"]
+        
+    return render_template("answer.html",
+            alert = get_alert(),
+            nick = get_nick(),
+            questions = rows2dicts( D.get_questions(aid), ['i','q','n','p'] ),
+            link = D.get_quiz_link( aid )
+        )
+
+@app.route("/set/answers",methods=["POST"])
+def set_answers():
+    if "id" not in session:
+        session["alert"]="Nimimerkkiä ei ole vielä valittu!"
+        return redirect( "/#nick" )
+    if "answer_id" not in session:
+        session["alert"]="Kyselyä ei ole valittu vastaamista varten!"
+        return redirect( "/#answer" )
+
+    sid = session["id"]
+    for qid, answer in request.form.items():
+        try: 
+            if int(answer) < 0 or int(answer) > 999:
+                session["alert"]="Luvattoman pieniä tai suuria lukuja!"
+                return redirect( "/#answer" )
+            elif D.get_user_answer(sid, int(qid) ):
+                session["alert"]="Kyselyyn olikin jo saatu vastauksia."
+                return redirect( "/#answer" )
+        except ValueError:
+            session["alert"] = "Vastaukset ei ole lukuja!"
+            return redirect( "/#answer" )
+
+    for qid, answer in request.form.items():
+        D.answer_new(sid, int(qid), int(answer))
+
+    return redirect("/#analyse")