diff options
Diffstat (limited to 'app.py')
| -rw-r--r-- | app.py | 221 |
1 files changed, 221 insertions, 0 deletions
@@ -0,0 +1,221 @@ +from dotenv import load_dotenv +load_dotenv() + +import datetime +from dataclasses import dataclass +from flask import Flask, render_template, redirect, request, make_response +app = Flask(__name__) + +import db + +@dataclass +class ChoicesByVoter: + name: str + votes: list[bool] + +TITLE_MAX_LENGTH = 100 +DESCRIPTION_MAX_LENGTH = 1000 +AUTHOR_NAME_MAX_LENGTH = 100 +AUTHOR_EMAIL_MAX_LENGTH = 100 +VOTER_NAME_MAX_LENGTH = 100 + +def voter_voted_on_choice(voter_name: str, choice: db.Choice) -> bool: + for vote in choice.votes: + if vote.voter_name == voter_name: + return True + return False + +def validation_error(message: str): + return render_template("error.html.j2", error=message), 400 + +@app.route("/") +def hello_world(): + created_poll_codes = [] + for k, _ in request.cookies.items(): + if k.startswith("diddle_manage_code_"): + created_poll_codes.append(k.replace("diddle_manage_code_", "")) + + created_polls = db.get_polls_by_codes(created_poll_codes) if len(created_poll_codes) > 0 else [] + + return render_template('index.html.j2', + created_polls=created_polls) + +@app.post("/poll/create") +def create(): + form = request.form + + if "title" not in form or len(form["title"]) == 0: + return validation_error("Title is required") + if len(form["title"]) > TITLE_MAX_LENGTH: + return validation_error(f"Title must be {TITLE_MAX_LENGTH} characters or fewer") + if "description" in form and len(form["description"]) > DESCRIPTION_MAX_LENGTH: + return validation_error(f"Description must be {DESCRIPTION_MAX_LENGTH} characters or fewer") + if "author_name" not in form or len(form["author_name"]) == 0: + return validation_error("Author name is required") + if len(form["author_name"]) > AUTHOR_NAME_MAX_LENGTH: + return validation_error(f"Author name must be {AUTHOR_NAME_MAX_LENGTH} characters or fewer") + if "author_email" in form and len(form["author_email"]) > AUTHOR_EMAIL_MAX_LENGTH: + return validation_error(f"Author email must be {AUTHOR_EMAIL_MAX_LENGTH} characters or fewer") + + choices = [] + poll = db.create_poll( + form["title"], + form["description"], + form["author_name"], + form["author_email"], + choices, + ) + resp = make_response( + redirect(f"/manage/{poll.manage_code}") + ) + resp.set_cookie(f"diddle_manage_code_{poll.manage_code}", "1") + return resp + +@app.get("/poll/<id>") +def poll(id): + poll = db.get_poll(id) + if poll is None: + return "Poll not found", 404 + + display_mode = request.cookies.get("diddle_display_mode", "table") + + voter_names = [] + for choice in poll.choices: + for vote in choice.votes: + if vote.voter_name not in voter_names: + voter_names.append(vote.voter_name) + + choices_by_voter = [] + for voter_name in voter_names: + votes: list[bool] = [] + for choice in poll.choices: + voted = voter_voted_on_choice(voter_name, choice) + votes.append(voted) + + choices_by_voter.append( + ChoicesByVoter(name=voter_name, votes=votes) + ) + + choices_by_voter.sort(key=lambda x: x.name) + + return render_template('poll.html.j2', + poll=poll, + choices_by_voter=choices_by_voter, + choices=poll.choices, + now=datetime.datetime.now(), + display_mode=display_mode, + ) + +@app.post("/poll/<id>/vote") +def vote_poll(id): + form = request.form + if "voter_name" not in form or len(form["voter_name"]) == 0: + return validation_error("Voter name is required") + if len(form["voter_name"]) > VOTER_NAME_MAX_LENGTH: + return validation_error(f"Voter name must be {VOTER_NAME_MAX_LENGTH} characters or fewer") + + voter_name = form["voter_name"] + choice_ids = [] + for k in form.keys(): + if k.startswith("choice_"): + choice_id = k.replace("choice_", "") + choice_ids.append(choice_id) + + db.vote_poll(id, voter_name, choice_ids) + + return redirect(f"/poll/{id}") + +@app.post("/manage/<code>/update_info") +def update_poll_info(code): + form = request.form + + if "title" not in form or len(form["title"]) == 0: + return validation_error("Title is required") + if len(form["title"]) > TITLE_MAX_LENGTH: + return validation_error(f"Title must be {TITLE_MAX_LENGTH} characters or fewer") + if "description" in form and len(form["description"]) > DESCRIPTION_MAX_LENGTH: + return validation_error(f"Description must be {DESCRIPTION_MAX_LENGTH} characters or fewer") + if "author_name" not in form or len(form["author_name"]) == 0: + return validation_error("Author name is required") + if len(form["author_name"]) > AUTHOR_NAME_MAX_LENGTH: + return validation_error(f"Author name must be {AUTHOR_NAME_MAX_LENGTH} characters or fewer") + if "author_email" in form and len(form["author_email"]) > AUTHOR_EMAIL_MAX_LENGTH: + return validation_error(f"Author email must be {AUTHOR_EMAIL_MAX_LENGTH} characters or fewer") + + db.update_poll_info( + code, + form["title"], + form["description"], + form["author_name"], + form["author_email"], + ) + + return redirect(f"/manage/{code}") + +@app.post("/manage/<code>/add_choice") +def add_choice(code): + form = request.form + if "start_datetime" not in form or len(form["start_datetime"]) == 0: + return validation_error("Start datetime is required") + if "end_datetime" not in form or len(form["end_datetime"]) == 0: + return validation_error("End datetime is required") + if form["start_datetime"] >= form["end_datetime"]: + return validation_error("Start datetime must be before end datetime") + + db.add_choice_to_poll( + code, + form["start_datetime"], + form["end_datetime"], + ) + + return redirect(f"/manage/{code}") + +@app.post("/manage/<code>/delete_choice/<choice_id>") +def delete_choice(code, choice_id): + poll = db.get_poll_by_code(code) + if poll is None: + return "Poll not found", 404 + + db.delete_choice(choice_id) + + return redirect(f"/manage/{code}") + +@app.get("/manage/<code>") +def manage(code): + poll = db.get_poll_by_code(code) + if poll is None: + return validation_error("Poll not found") + + last_choice_id = poll.choices[-1].id if len(poll.choices) > 0 else None + resp = make_response( + render_template("manage.html.j2", + poll=poll, + last_choice_id=last_choice_id)) + resp.set_cookie(f"diddle_manage_code_{code}", "1") + return resp + +@app.post("/manage/<code>/delete") +def delete_poll(code): + poll = db.get_poll_by_code(code) + return render_template("poll_confirm_delete.html.j2", poll=poll) + +@app.post("/manage/<code>/confirm_delete") +def confirm_delete_poll(code): + db.delete_poll(code) + resp = make_response(redirect("/")) + resp.set_cookie(f"diddle_manage_code_{code}", "", expires=0) + return resp + +@app.post("/options/toggle_display_mode") +def toggle_display_mode(): + poll_id = request.form["poll_id"] + redirect_url = f"/poll/{poll_id}" + display_mode = request.cookies.get("diddle_display_mode", "table") + if display_mode == "table": + display_mode = "list" + else: + display_mode = "table" + + resp = make_response(redirect(redirect_url)) + resp.set_cookie("diddle_display_mode", display_mode) + return resp |