diff options
Diffstat (limited to 'pysite/views/main/jams/join.py')
| -rw-r--r-- | pysite/views/main/jams/join.py | 180 |
1 files changed, 180 insertions, 0 deletions
diff --git a/pysite/views/main/jams/join.py b/pysite/views/main/jams/join.py new file mode 100644 index 00000000..87a2c4ca --- /dev/null +++ b/pysite/views/main/jams/join.py @@ -0,0 +1,180 @@ +from email.utils import parseaddr + +from flask import request, redirect, url_for +from werkzeug.exceptions import NotFound, BadRequest + +from pysite.base_route import RouteView +from pysite.decorators import csrf +from pysite.mixins import DBMixin, OauthMixin + + +class JamsJoinView(RouteView, DBMixin, OauthMixin): + path = "/jams/join/<int:jam>" + name = "jams.join" + + table_name = "code_jams" + forms_table = "code_jam_forms" + questions_table = "code_jam_questions" + responses_table = "code_jam_responses" + participants_table = "code_jam_participants" + infractions_table = "code_jam_infractions" + + def get(self, jam): + jam_obj = self.db.get(self.table_name, jam) + + if not jam_obj: + return NotFound() + + if not self.user_data: + return redirect(url_for("discord.login")) + + if self.get_response(jam, self.user_data["user_id"]): + return self.render("main/jams/already.html", jam=jam_obj) + + infractions = self.get_infractions(self.user_data["user_id"]) + + for infraction in infractions: + if infraction["number"] == -1: # Indefinite ban + return self.render("main/jams/banned.html", infraction=infraction, jam=jam_obj) + + if infraction["number"]: # Got some jams left + if jam not in infraction["decremented_for"]: + # Make sure they haven't already tried to apply for this jam + infraction["number"] -= 1 + infraction["decremented_for"].append(jam) + + self.db.insert(self.infractions_table, infraction, conflict="replace") + + return self.render("main/jams/banned.html", infraction=infraction, jam=jam_obj) + + if jam in infraction["decremented_for"]: + # They already tried to apply for this jam + return self.render("main/jams/banned.html", infraction=infraction, jam=jam_obj) + + form_obj = self.db.get(self.forms_table, jam) + questions = [] + + if form_obj: + for question in form_obj["questions"]: + questions.append(self.db.get(self.questions_table, question)) + + return self.render( + "main/jams/join.html", jam=jam_obj, form=form_obj, + questions=questions, question_ids=[q["id"] for q in questions] + ) + + @csrf + def post(self, jam): + jam_obj = self.db.get(self.table_name, jam) + + if not jam_obj: + return NotFound() + + if not self.user_data: + return redirect(url_for("discord.login")) + + if self.get_response(jam, self.user_data["user_id"]): + return self.render("main/jams/already.html", jam=jam_obj) + + infractions = self.get_infractions(self.user_data["user_id"]) + + for infraction in infractions: + if infraction["number"] == -1: # Indefinite ban + return self.render("main/jams/banned.html", infraction=infraction) + + if infraction["number"]: # Got some jams left + if jam not in infraction["decremented_for"]: + # Make sure they haven't already tried to apply for this jam + infraction["number"] -= 1 + infraction["decremented_for"].append(jam) + + self.db.insert(self.infractions_table, infraction, conflict="replace") + + return self.render("main/jams/banned.html", infraction=infraction, jam=jam_obj) + + if jam in infraction["decremented_for"]: + # They already tried to apply for this jam + return self.render("main/jams/banned.html", infraction=infraction, jam=jam_obj) + + form_obj = self.db.get(self.forms_table, jam) + + if not form_obj: + return NotFound() + + questions = [] + + for question in form_obj["questions"]: + questions.append(self.db.get(self.questions_table, question)) + + answers = [] + + for question in questions: + value = request.form.get(question["id"]) + answer = {"question": question["id"]} + + if not question["optional"] and value is None: + return BadRequest() + + if question["type"] == "checkbox": + if value == "on": + answer["value"] = True + elif not question["optional"]: + return BadRequest() + else: + answer["value"] = False + + elif question["type"] == "email": + if value: + address = parseaddr(value) + + if address == ("", ""): + return BadRequest() + + answer["value"] = value + + elif question["type"] in ["number", "range", "slider"]: + if value is not None: + value = int(value) + + if value > int(question["data"]["max"]) or value < int(question["data"]["min"]): + return BadRequest() + + answer["value"] = value + + elif question["type"] == "radio": + if value: + if value not in question["data"]["options"]: + return BadRequest() + + answer["value"] = value + + elif question["type"] in ["text", "textarea"]: + answer["value"] = value + + answers.append(answer) + + user_id = self.user_data["user_id"] + + response = { + "snowflake": user_id, + "jam": jam, + "approved": False, + "answers": answers + } + + print(response) + self.db.insert(self.responses_table, response) + return self.render("main/jams/thanks.html", jam=jam_obj) + + def get_response(self, jam, user_id): + query = self.db.query(self.responses_table).filter({"jam": jam, "snowflake": user_id}) + result = self.db.run(query, coerce=list) + print(result) + + if result: + return result[0] + return None + + def get_infractions(self, user_id): + query = self.db.query(self.infractions_table).filter({"participant": user_id}) + return self.db.run(query, coerce=list) |