from django.utils.html import conditional_escape as esc
from itertools import groupby
from calendar import HTMLCalendar
from datetime import date


class EventCalendar(HTMLCalendar):

    def __init__(self, pEvents):
        super(EventCalendar, self).__init__()
        self.events = self.group_by_day(pEvents)

    def formatday(self, day, weekday):
        if day != 0:
            cssclass = self.cssclasses[weekday]
            if date.today() == date(self.year, self.month, day):
                cssclass += ' today'
            if day in self.events:
                cssclass += ' filled'
                body = []
                for ev in self.events[day]:
                    body.append('<a href="/calendar/%s">' % ev.id)
                    body.append(esc(ev.nom))
                    body.append('</a><br/>')
                return self.day_cell(cssclass,
                                     '<div class="dayNumber">%d</div> %s'
                                     % (day, ''.join(body)))
            return self.day_cell(cssclass,
                                 '<div class="dayNumber">%d</div>' % day)
        return self.day_cell('noday', '&nbsp;')

    def formatmonth(self, year, month):
        self.year, self.month = year, month
        return super(EventCalendar, self).formatmonth(year, month)

    def group_by_day(self, pEvents):
        def field(ev): ev.date.day
        return dict(
            [(dat, list(items)) for dat, items in groupby(pEvents, field)]
        )

    def day_cell(self, cssclass, body):
        return '<td class="%s">%s</td>' % (cssclass, body)