Produits Entr'ouvert » w.c.s. » Au quotidien

import time

import datetime

from quixote.directory import Directory

from quixote import get_publisher, get_request, redirect, get_session, get_response

from qommon import misc, template, errors

from events import Event, RemoteCalendar

class TagDirectory(Directory):

    def _q_lookup(self, component):

        events = Event.select()

        for remote_calendar in RemoteCalendar.select():

            if remote_calendar.events:

                events.extend(remote_calendar.events)

        self.events = [x for x in events if component in (x.keywords or [])]

        self.events.sort(lambda x,y: cmp(x.date_start, y.date_start))

        self.tag = component

        return self.display_events()

    def display_events [html] (self):

        template.html_top(_('Agenda'))

        if len(self.events) > 1:

            '<p id="nb-events">'

            _('%(nb)d events with %(keyword)s keyword') % {

                    'nb': len(self.events),

                    'keyword': self.tag

            }

            '</p>'

        if self.events:

            '<dl id="events">'

            for ev in self.events:

                ev.as_html_dt_dd()

            '</dl>'

        else:

            '<p id="nb-events">'

            _('No event registered with the %s keyword.') % self.tag

            '</p>'

class AgendaDirectory(Directory):

    _q_exports = ['', 'icalendar', 'tag']

    year = None

    month = None

    tag = TagDirectory()

    def _q_traverse(self, path):

        get_response().breadcrumb.append(('agenda/', _('Agenda')))

        self.year, self.month = time.localtime()[:2]

        if len(path) >= 1 and path[0].isdigit():

            self.year, self.month = (None, None)

            self.year = int(path[0])

            get_response().breadcrumb.append(('%s/' % self.year, self.year))

            path = path[1:]

            if len(path) >= 1 and path[0] in [str(x) for x in range(1, 13)]:

                self.month = int(path[0])

                get_response().breadcrumb.append(('%s/' % self.month,

                            misc.get_month_name(self.month)))

                path = path[1:]

            if len(path) == 0:

                return redirect(get_request().get_path() + '/')

        return Directory._q_traverse(self, path)

    def _q_index [html] (self):

        if self.month:

            self.display_month_links()

            self.display_month()

        else:

            return redirect('..')

    def display_month [html] (self):

        template.html_top(_('Agenda'))

        events = Event.select()

        remote_cal = get_request().form.get('cal')

        if remote_cal != 'local':

            if remote_cal:

                events = RemoteCalendar.get(remote_cal).events

                if not events:

                    events = []

            else:

                for remote_calendar in RemoteCalendar.select():

                    if remote_calendar.events:

                        events.extend(remote_calendar.events)

        events = [x for x in events if x.in_month(self.year, self.month)]

        events.sort(lambda x,y: cmp(x.date_start, y.date_start))

        if events:

            if len(events) > 1:

                '<p id="nb-events">'

                _('%(nb)d events for %(month_name)s %(year)s') % {

                        'nb': len(events),

                        'month_name': misc.get_month_name(self.month),

                        'year': self.year}

                '</p>'

            '<dl id="events">'

            for ev in events:

                ev.as_html_dt_dd()

            '</dl>'

        else:

            '<p id="nb-events">'

            _('No event registered for the month of %s.') % '%s %s' % (

                    misc.get_month_name(self.month), self.year)

            '</p>'

    def display_month_links [html] (self):

        today = datetime.date(*(time.localtime()[:2] + (1,)))

        '<ul id="month-links">'

        for i in range(12):

            '<li>'

            if (today.year, today.month) == (self.year, self.month):

                '<strong>'

                '%s %s' % (misc.get_month_name(today.month), today.year)

                '</strong>'

            else:

                root_url = get_publisher().get_root_url()

                '<a href="%sagenda/%s/%s/">' % (root_url, today.year, today.month)

                '%s %s' % (misc.get_month_name(today.month), today.year)

                '</a>'

            '</li>'

            today += datetime.timedelta(31)

        '</ul>'

    def display_remote_calendars [html] (self):

        remote_calendars = [x for x in RemoteCalendar.select() if x.label]

        if not remote_calendars:

            return

        remote_calendars.sort(lambda x,y: cmp(x.label, y.label))

        '<p class="tags">'

        remote_cal = get_request().form.get('cal')

        if remote_cal:

            '<a href=".">%s</a> ' % _('All')

        else:

            '<strong><a href=".">%s</a></strong> ' % _('All')

        if remote_cal != 'local':

            '<a href="./?cal=local">%s</a> ' % _('Local')

        else:

            '<strong><a href="./?cal=local">%s</a></strong> ' % _('Local')

        for cal in remote_calendars:

            if remote_cal == str(cal.id):

                '<strong><a href="./?cal=%s">%s</a></strong> ' % (cal.id, cal.label)

            else:

                '<a href="./?cal=%s">%s</a> ' % (cal.id, cal.label)

        '</p>'

    def icalendar(self):

        if not Event.keys():

            raise errors.TraversalError()

        response = get_response()

        response.set_content_type('text/calendar', 'utf-8')

        return Event.as_vcalendar().encode('utf-8')