# -*- coding: utf-8 -*- # Copyright(C) 2010-2013 Bezleputh # # This file is part of weboob. # # weboob is free software: you can redistribute it and/or modify # it under the terms of the GNU Lesser General Public License as published by # the Free Software Foundation, either version 3 of the License, or # (at your option) any later version. # # weboob is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU Lesser General Public License for more details. # # You should have received a copy of the GNU Lesser General Public License # along with weboob. If not, see . from .base import ( BaseObject, StringField, IntField, FloatField, Field, EnumField, Enum, ) from .collection import CapCollection, CollectionNotFound, Collection from .date import DateField from datetime import time, datetime from weboob.tools.date import parse_date __all__ = ['BaseCalendarEvent', 'CapCalendarEvent'] class CATEGORIES(Enum): CONCERT = u'Concert' CINE = u'Cinema' THEATRE = u'Theatre' TELE = u'Television' CONF = u'Conference' AUTRE = u'Autre' EXPO = u'Exposition' SPECTACLE = u'Spectacle' FEST = u'Festival' SPORT = u'Sport' #the following elements deal with ICalendar stantdards #see http://fr.wikipedia.org/wiki/ICalendar#Ev.C3.A9nements_.28VEVENT.29 class TRANSP(Enum): OPAQUE = u'OPAQUE' TRANSPARENT = u'TRANSPARENT' class STATUS(Enum): TENTATIVE = u'TENTATIVE' CONFIRMED = u'CONFIRMED' CANCELLED = u'CANCELLED' class TICKET(Enum): AVAILABLE = u'Available' NOTAVAILABLE = u'Not available' CLOSED = u'Closed' class BaseCalendarEvent(BaseObject): """ Represents a calendar event """ start_date = DateField('Start date of the event') end_date = DateField('End date of the event') timezone = StringField('Timezone of the event in order to convert to utc time', default='Etc/UCT') summary = StringField('Title of the event') city = StringField('Name of the city in witch event will take place') location = StringField('Location of the event') category = EnumField('Category of the event', CATEGORIES) description = StringField('Description of the event') price = FloatField('Price of the event') booked_entries = IntField('Entry number') max_entries = IntField('Max entry number') event_planner = StringField('Name of the event planner') #the following elements deal with ICalendar stantdards #see http://fr.wikipedia.org/wiki/ICalendar#Ev.C3.A9nements_.28VEVENT.29 sequence = IntField('Number of updates, the first is number 1') # (TENTATIVE, CONFIRMED, CANCELLED) status = EnumField('Status of the event', STATUS) # (OPAQUE, TRANSPARENT) transp = EnumField('Describes if event is available', TRANSP) # (AVAILABLE, NOTAVAILABLE, CLOSED) ticket = EnumField('Describes if tickets are available', TICKET) @classmethod def id2url(cls, _id): """Overloaded in child classes provided by backends.""" raise NotImplementedError() @property def page_url(self): """ Get page URL of the announce. """ return self.id2url(self.id) class Query(BaseObject): """ Query to find events """ start_date = DateField('Start date of the event') end_date = DateField('End date of the event') city = StringField('Name of the city in witch event will take place') categories = Field('List of categories of the event', list, tuple, default=list(CATEGORIES)) ticket = Field('List of status of the tickets sale', list, tuple, default=list(TICKET)) summary = StringField('Title of the event') class CapCalendarEvent(CapCollection): """ Capability of calendar event type sites """ ASSOCIATED_CATEGORIES = 'ALL' def has_matching_categories(self, query): if self.ASSOCIATED_CATEGORIES == 'ALL': return True for category in query.categories: if category in self.ASSOCIATED_CATEGORIES: return True return False def search_events(self, query): """ Search event :param query: search query :type query: :class:`Query` :rtype: iter[:class:`BaseCalendarEvent`] """ raise NotImplementedError() def list_events(self, date_from, date_to=None): """ list coming event. :param date_from: date of beguinning of the events list :type date_from: date :param date_to: date of ending of the events list :type date_to: date :rtype: iter[:class:`BaseCalendarEvent`] """ raise NotImplementedError() def get_event(self, _id): """ Get an event from an ID. :param _id: id of the event :type _id: str :rtype: :class:`BaseCalendarEvent` or None is fot found. """ raise NotImplementedError() def attends_event(self, event, is_attending=True): """ Attends or not to an event :param event : the event :type event : BaseCalendarEvent :param is_attending : is attending to the event or not :type is_attending : bool """ raise NotImplementedError() def iter_resources(self, objs, split_path): """ Iter events by category """ if len(split_path) == 0 and self.ASSOCIATED_CATEGORIES != 'ALL': for category in self.ASSOCIATED_CATEGORIES: collection = Collection([category], category) yield collection elif len(split_path) == 1 and split_path[0] in self.ASSOCIATED_CATEGORIES: query = Query() query.categories = split_path query.start_date = datetime.combine(parse_date('today'), time.min) query.end_date = parse_date('') query.city = u'' for event in self.search_events(query): yield event def validate_collection(self, objs, collection): """ Validate Collection """ if collection.path_level == 0: return if collection.path_level == 1 and collection.split_path[0] in CATEGORIES.values: return raise CollectionNotFound(collection.split_path)