import datetime

from flask import redirect, request, url_for
from werkzeug.exceptions import BadRequest

from pysite.base_route import RouteView
from pysite.decorators import csrf
from pysite.mixins import DBMixin, OauthMixin


class JamsProfileView(RouteView, DBMixin, OauthMixin):
    path = "/jams/profile"
    name = "jams.profile"

    table_name = "code_jam_participants"

    def get(self):
        if not self.user_data:
            return redirect(url_for("discord.login"))

        participant = self.db.get(self.table_name, self.user_data["user_id"])

        if not participant:
            participant = {"id": self.user_data["user_id"]}

        return self.render(
            "main/jams/profile.html", participant=participant
        )

    @csrf
    def post(self):
        if not self.user_data:
            return redirect(url_for("discord.login"))

        participant = self.db.get(self.table_name, self.user_data["user_id"])

        if not participant:
            participant = {"id": self.user_data["user_id"]}

        dob = request.form.get("dob")
        github_username = request.form.get("github_username")
        timezone = request.form.get("timezone")

        if not dob or not github_username or not timezone:
            return BadRequest()

        # Convert given datetime strings into actual objects, adding timezones to keep rethinkdb happy
        dob = datetime.datetime.strptime(dob, "%Y-%m-%d")
        dob = dob.replace(tzinfo=datetime.timezone.utc)

        participant["dob"] = dob
        participant["github_username"] = github_username
        participant["timezone"] = timezone

        self.db.insert(self.table_name, participant, conflict="replace")

        return self.render(
            "main/jams/profile.html", participant=participant, done=True
        )