import calendar
from datetime import datetime
from pprint import pprint

from flask import Blueprint, render_template

# import pytz
from .dm import dmbp
from .get_period_times import (
    get_period_totals,
    read_and_extract,
    task_file,
    task_or_none,
    timezone,
    get_work_project_tasks,
    get_work_period_totals,
)



class DMHTMLCalendar(calendar.HTMLCalendar):
    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.use_work_projects = False
        self.task = None

    def setcalmonth(self, month):
        self.dmmonth = month

    def setcalyear(self, year):
        self.dmyear = year

    def settask(self, task):
        self.task = task

    def set_work_projects_mode(self, enabled=True):
        self.use_work_projects = enabled

    def oneday(self, month, day):
        current_year = datetime.today().year
        start = datetime(self.dmyear, month, day).replace(
            hour=0, minute=0, second=0, tzinfo=timezone
        )
        end = datetime(self.dmyear, month, day).replace(
            hour=23, minute=59, second=59, tzinfo=timezone
        )

        if self.use_work_projects:
            rows = get_work_period_totals(start, end)
        else:
            rows = get_period_totals(start, end, self.task)

        returnstr = "<table class='totaltable'>"
        for row in rows:
            returnstr += "<tr><td>{}</td><td>{}</td></tr>".format(
                row["ws"], row["total"]
            )

        returnstr += "</table>"
        return returnstr

    def oneweek(self, month, week):
        start_day = None
        end_day = None

        for d, wd in week:
            if d == 0:
                continue
            else:
                start_day = d
                break

        for d, wd in reversed(week):
            if d == 0:
                continue
            else:
                end_day = d
                break

        start = datetime(self.dmyear, month, start_day).replace(
            hour=0, minute=0, second=0, tzinfo=timezone
        )

        end = datetime(self.dmyear, month, end_day).replace(
            hour=23, minute=59, second=59, tzinfo=timezone
        )

        if self.use_work_projects:
            rows = get_work_period_totals(start, end)
        else:
            rows = get_period_totals(start, end, self.task)

        print(rows)

        returnstr = "<table class='totaltable'>"
        for row in rows:
            returnstr += "<tr><td>{}</td><td>{}</td></tr>".format(
                row["ws"], row["total"]
            )

        returnstr += "</table>"
        return returnstr

    def formatweekheader(self):
        """
        Return a header for a week as a table row.
        """
        s = "".join(self.formatweekday(i) for i in self.iterweekdays())
        s += "<td>Week Totals</td>"
        return "<tr>%s</tr>" % s

    def formatweek(self, theweek):
        """
        Return a complete week as a table row.
        """
        s = "".join(self.formatday(d, wd) for (d, wd) in theweek)
        s += "<td>{}</td>".format(self.oneweek(self.dmmonth, theweek))
        return "<tr>%s</tr>" % s

    def formatday(self, day, weekday):
        """
        Return a day as a table cell.
        """
        if day == 0:
            return '<td class="noday">&nbsp;</td>'  # day outside month
        else:
            return '<td class="%s">%s</td>' % (
                self.cssclasses[weekday],
                self.oneday(self.dmmonth, day),
            )


@dmbp.route("/workmonth")
@dmbp.route("/workmonth/<int:month>")
@dmbp.route("/workmonth/<int:month>/<int:year>")
def workmonth(month=None, year=None):
    usemonth = datetime.today().month
    useyear = datetime.today().year

    if month:
        usemonth = month

    if year:
        useyear = year

    cal = DMHTMLCalendar(calendar.SATURDAY)

    cal.set_work_projects_mode(True)
    cal.setcalmonth(usemonth)
    cal.setcalyear(useyear)

    return render_template("calendar.html", content=cal.formatmonth(useyear, usemonth))


@dmbp.route("/month")
@dmbp.route("/month/<int:month>")
@dmbp.route("/month/<int:month>/<int:year>")
@dmbp.route("/month/<string:task>")
@dmbp.route("/month/<string:task>/<int:month>")
@dmbp.route("/month/<string:task>/<int:month>/<int:year>")
def month(month=None, year=None, task=None):
    usemonth = datetime.today().month
    useyear = datetime.today().year

    if month:
        usemonth = month

    if year:
        useyear = year

    cal = DMHTMLCalendar(calendar.SATURDAY)

    cal.settask(task_or_none(task))

    cal.setcalmonth(usemonth)
    cal.setcalyear(useyear)

    return render_template("calendar.html", content=cal.formatmonth(useyear, usemonth))