joeni

""" This is a module dedicated to collecting all functionality regarding the

building of the roster. """

from django.shortcuts import render

from collections import OrderedDict

import datetime

from django.urls import reverse # Why?

from django.utils.translation import gettext as _

from .models import *

import administration

def same_day(datetime_a, datetime_b):

    """True if both a and b are on the same day, false otherwise."""

    return (

        datetime_a.day == datetime_b.day and

        datetime_a.month == datetime_b.month and

        datetime_a.year == datetime_b.year)

def make_quarter_buckets(events):

    """Returns a dict with all days that the given events take place.

    Every quarter between the first and last event will get a bucket in the dict,

    with the keys in format "dd-mm-yyyy". Also quarters without events will be

    included, but will simply have empty buckets."""

    quarters = OrderedDict()  # This is the first time I ever use an OrderedDict and I intend it to be my last one as well.

    current_quarter = datetime.datetime.now().replace(hour=8, minute=0)

    while current_quarter.hour != 20 or current_quarter.minute != 00:

        quarters[current_quarter.strftime("%H:%M")] = list()

        current_quarter += datetime.timedelta(minutes=15)

    for event in events:

        quarters[event.begin_time.strftime("%H:%M")].append(event)

    return quarters  # Yay! ^.^

def create_roster_rows(events):

    """Creates the rows for use in the roster table.

    None of the times in the given events may overlap, and all must start and

    end at a quarter of the hour (so :00, :15, :30, or :45). If you think you're above this,

    I'll raise you a ValueError, kind sir.

    Events must be of administration.models.Event type."""

    if len(events) == 0:

        return []

    for event in events:

        for other_event in events:

            if other_event is not event and (

                    (event.begin_time >= other_event.begin_time and event.begin_time < other_event.end_time)

                    or (event.end_time > other_event.begin_time and event.end_time <= other_event.end_time)

                    ):

                raise ValueError("One of the events overlaps with another event.")

        if event.begin_time.minute not in [0, 15, 30, 45] or event.end_time.minute not in [0, 15, 30, 45]:

            raise ValueError("One of the events did not begin or end on a quarter.")

    # XXX: I assume here the given queryset is sorted:

    first_day = events[0].begin_time

    last_day = events.last().begin_time

    # All events validated

    quarter_buckets = make_quarter_buckets(events)

    skip_row = dict()

    table_code = list()

    for quarter_bucket in quarter_buckets.keys():

        # Preparing time column

        quarter_line = "<tr><td style='font-size: xx-small;'>"

        if quarter_bucket[3:] == "00":

            quarter_line += quarter_bucket

        quarter_line += "</td>"

        current_day = first_day

        while not same_day(current_day, last_day+datetime.timedelta(days=1)):

            for event in quarter_buckets[quarter_bucket]:

                if same_day(current_day, event.begin_time):

                    quarters = (event.end_time - event.begin_time).seconds // 60 // 15

                    skip_row[quarter_bucket] = quarters

                    event_line = "<td "

                    if isinstance(event, CourseEvent) and (datetime.datetime.now(datetime.timezone.utc) - event.last_update).days > 1:

                        event_line += "style='background-color: #"+event.course.course.color+"; color: white;' "

                    elif (datetime.datetime.now(datetime.timezone.utc) - event.created).days <= 5 and (event.last_update - event.created).days < 1:

                        event_line += "class='event-new' "

                    elif (datetime.datetime.now(datetime.timezone.utc) - event.last_update).days <= 5:

                        event_line += "class='event-update' "

                    if event.note:

                        event_line +="class='event-note' title='" +event.note+ "' "

                    # FIXME: From here, I just assume the events are all CourseEvents, because

                    # that is the most important one to implement for the prototype.

                    if quarters > 1:

                        event_line += "rowspan='"+str(quarters)+"' "

                    event_line +=">"

                    event_line += str(

                        #"<a href=" + reverse("courses-course-index", args="")+"> "  # FIXME so that this links to the course's index page

                        #"<a href=""> "

                        str(event.course.course.name)

                        #+ str(event.course.course.name)

                        + "<br />"

                        + str(event.docent)

                        + "<br />"

                        + event.begin_time.strftime("%H:%M") + " - " + event.end_time.strftime("%H:%M")

                        + "<br />"

                        #+ str(event.room) + " (" + str(event.subject) + ")</a></td>")

                        + str(event.room) + " (" + str(event.subject) + ")</td>")

                    quarter_line += event_line

                else:

                    """if quarter_bucket in skip_row:

                        skip_row[quarter_bucket] -= 1

                        if skip_row[quarter_bucket] == 0:

                            del skip_row[quarter_bucket]

                            quarter_line += "<td></td>"

                    else:

                        quarter_line += "<td></td>"""

                    pass

            current_day += datetime.timedelta(days=1)

        quarter_line += "</tr>"

        table_code.append(quarter_line)

    return table_code