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

import time

import datetime

from sets import Set

from quixote.directory import Directory

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

from quixote.html import htmltext, TemplateIO

from qommon import misc, template, errors, get_cfg

from qommon.form import *

from events import Event, RemoteCalendar, get_default_event_tags

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(self):

        template.html_top(_('Agenda'))

        r = TemplateIO(html=True)

        if len(self.events) > 1:

            r += htmltext('<p id="nb-events">')

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

                    'nb': len(self.events),

                    'keyword': self.tag

            }

            r += htmltext('</p>')

        if self.events:

            r += htmltext('<dl id="events">')

            for ev in self.events:

                r += htmltext(ev.as_html_dt_dd())

            r += htmltext('</dl>')

        else:

            r += htmltext('<p id="nb-events">')

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

            r += htmltext('</p>')

        return r.getvalue()

class AgendaDirectory(Directory):

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

    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(self):

        if self.month:

            r = TemplateIO(html=True)

            r += htmltext(self.display_month_links())

            r += htmltext(self.display_month())

            return r.getvalue()

        else:

            return redirect('..')

    def display_month(self):

        template.html_top(_('Agenda'))

        events = Event.select()

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

        if remote_cal != 'local':

            if remote_cal:

                try:

                    events = RemoteCalendar.get(remote_cal).events

                except KeyError:

                    raise errors.TraversalError()

                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))

        r = TemplateIO(html=True)

        if events:

            if len(events) > 1:

                r += htmltrxt('<p id="nb-events">')

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

                        'nb': len(events),

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

                        'year': self.year}

                r += htmltext('</p>')

            r += htmltext('<dl id="events">')

            for ev in events:

                r += htmltext(ev.as_html_dt_dd())

            r += htmltext('</dl>')

        else:

            r += htmltext('<p id="nb-events">')

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

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

            r += htmltext('</p>')

        root_url = get_publisher().get_root_url()

        r += htmltext('<div id="agenda-subs">')

        r += htmltext('<p>')

        r += _('You can subscribe to this calendar:')

        r += htmltext('</p>')

        r += htmltext('<ul>')

        r += htmltext('  <li><a href="%sagenda/icalendar" id="par_ical">%s</a></li>') % (

                root_url, _('iCalendar'))

        r += htmltext('  <li><a href="%sagenda/atom" id="par_rss">%s</a></li>') % (

                root_url, _('Feed'))

        r += htmltext('</ul>')

        r += htmltext('</div>')

        return r.getvalue()

    def display_month_links(self):

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

        r = TemplateIO(html=True)

        r += htmltext('<ul id="month-links">')

        for i in range(12):

            r += htmltext('<li>')

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

                r += htmltext('<strong>')

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

                r += htmltext('</strong>')

            else:

                root_url = get_publisher().get_root_url()

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

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

                r += htmltext('</a>')

            r += htmltext('</li>')

            today += datetime.timedelta(31)

        r += htmltext('</ul>')

        return r.getvalue()

    def display_remote_calendars(self):

        r = TemplateIO(html=True)

        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))

        r += htmltext('<p class="tags">')

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

        agenda_root_url = get_publisher().get_root_url() + 'agenda/'

        if remote_cal:

            r += htmltext('<a href="%s">%s</a> ') % (agenda_root_url, _('All'))

        else:

            r += htmltext('<strong><a href="%s">%s</a></strong> ') % (agenda_root_url, _('All'))

        if remote_cal != 'local':

            r += htmltext('<a href="%s?cal=local">%s</a> ') % (agenda_root_url, _('Local'))

        else:

            r += htmltext('<strong><a href="%s?cal=local">%s</a></strong> ') % (agenda_root_url, _('Local'))

        for cal in remote_calendars:

            if remote_cal == str(cal.id):

                r += htmltext('<strong><a href="%s?cal=%s">%s</a></strong> ') % (

                        agenda_root_url, cal.id, cal.label)

            else:

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

        r += htmltext('</p>')

        return r.getvalue()

    def icalendar(self):

        if not Event.keys():

            raise errors.TraversalError()

        response = get_response()

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

        vcal = Event.as_vcalendar()

        if type(vcal) is unicode:

            return vcal.encode('utf-8')

        else:

            return vcal

    def atom(self):

        response = get_response()

        response.set_content_type('application/atom+xml')

        from pyatom import pyatom

        xmldoc = pyatom.XMLDoc()

        feed = pyatom.Feed()

        xmldoc.root_element = feed

        feed.title = get_cfg('misc', {}).get('sitename', 'Au Quotidien') + ' - ' + _('Agenda')

        feed.id = get_request().get_url()

        author_email = get_cfg('emails', {}).get('reply_to')

        if not author_email:

            author_email = get_cfg('emails', {}).get('from')

        if author_email:

            feed.authors.append(pyatom.Author(author_email))

        feed.links.append(pyatom.Link(get_request().get_url(1) + '/'))

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

        nyear, nmonth = year, month+1

        if nmonth > 12:

            nyear, nmonth = nyear+1, 1

        events = [x for x in Event.select() if x.in_month(year, month) or x.in_month(nyear, nmonth)]

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

        events.reverse()

        for item in events:

            entry = item.get_atom_entry()

            if entry is not None:

                feed.entries.append(entry)

        return str(feed)

    def filter(self, no_event=False):

        template.html_top(_('Agenda'))

        tags = get_cfg('misc', {}).get('event_tags')

        if not tags:

            tags = get_default_event_tags()

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

        form = Form(enctype='multipart/form-data')

        if tags and remote_calendars:

            form.widgets.append(HtmlWidget('<table id="agenda-filter"><tr><td>'))

        if tags:

            form.add(CheckboxesWidget, 'tags', title = _('Tags'),

                    elements = [(x,x) for x in tags],

                    inline = False)

        if tags and remote_calendars:

            form.widgets.append(HtmlWidget('</td><td>'))

        if remote_calendars:

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

            form.add(CheckboxesWidget, 'calendars', title = _('Calendars'),

                    elements = [('local', _('Local'))] + [(x.id, x.label) for x in remote_calendars],

                    inline = False)

        if tags and remote_calendars:

            form.widgets.append(HtmlWidget('</td></tr></table>'))

        form.add_submit('submit', _('Submit'))

        form.add_submit('cancel', _('Cancel'))

        if form.get_widget('cancel').parse():

            return redirect('.')

        if no_event or not form.is_submitted():

            r = TemplateIO(html=True)

            if no_event:

                r += htmltext('<p id="nb-events">')

                r += _('No events matching the filter.')

                r += htmltext('</p>')

            r += form.render()

            return r.getvalue()

        else:

            return self.filter_submitted(form, tags, remote_calendars)

    def filter_submitted(self, form, tags, remote_calendars):

        if remote_calendars:

            selected_remote_calendars = form.get_widget('calendars').parse()

            events = []

            for remote_calendar in selected_remote_calendars:

                if remote_calendar == 'local':

                    events.extend(Event.select())

                else:

                    try:

                        events.extend(RemoteCalendar.get(remote_calendar).events)

                    except KeyError:

                        pass

        else:

            events = Event.select()

        events = [x for x in events if x.after_today()]

        if tags:

            selected_tags = Set(form.get_widget('tags').parse())

            if selected_tags and len(selected_tags) != len(tags):

                events = [x for x in events if Set(x.keywords).intersection(selected_tags)]

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

        r = TemplateIO(html=True)

        if len(events) > 1:

            r += htmltext('<p id="nb-events">')

            r += htmltext(_('%(nb)d events')) % {'nb': len(events)}

            r += htmltext('</p>')

        if events:

            r += htmltext('<dl id="events">')

            for ev in events:

                r += htmltext(ev.as_html_dt_dd())

            r += htmltext('</dl>')

            return r.getvalue()

        else:

            return self.filter(no_event=True)