new calendar module: sueurdemetal

modules/sueurdemetal/backend.py

@@ -0,0 +1,107 @@
+# -*- coding: utf-8 -*-
+
+# Copyright(C) 2013      Vincent A
+#
+# This file is part of weboob.
+#
+# weboob is free software: you can redistribute it and/or modify
+# it under the terms of the GNU Affero 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 Affero General Public License for more details.
+#
+# You should have received a copy of the GNU Affero General Public License
+# along with weboob. If not, see <http://www.gnu.org/licenses/>.
+
+
+from weboob.tools.backend import BaseBackend
+from weboob.capabilities.calendar import ICapCalendarEvent, BaseCalendarEvent, CATEGORIES, TRANSP, STATUS
+import datetime
+
+from .browser import SueurDeMetalBrowser
+
+
+__all__ = ['SueurDeMetalBackend']
+
+
+class Concert(BaseCalendarEvent):
+    @classmethod
+    def id2url(cls, _id):
+        return 'http://www.sueurdemetal.com/detail-concert-metal.php?c=%s' % _id
+
+
+class SueurDeMetalBackend(BaseBackend, ICapCalendarEvent):
+    NAME = 'sueurdemetal'
+    DESCRIPTION = u'SueurDeMetal French concerts list website'
+    MAINTAINER = u'Vincent A'
+    EMAIL = 'dev@indigo.re'
+    LICENSE = 'AGPLv3+'
+    VERSION = '0.h'
+
+    BROWSER = SueurDeMetalBrowser
+
+    ASSOCIATED_CATEGORIES = [CATEGORIES.CONCERT]
+
+    def __init__(self, *a, **kw):
+        super(SueurDeMetalBackend, self).__init__(*a, **kw)
+        self.cities = {}
+
+    def search_events(self, query):
+        city_id = self.find_city_id(query.city)
+        for d in self.browser.get_concerts_city(city_id):
+            yield self._make_event(d)
+
+    def get_event(self, _id):
+        d = self.browser.get_concert(_id)
+        return self._make_event(d)
+
+    def _make_event(self, d):
+        event = Concert(d['id'])
+        event.category = CATEGORIES.CONCERT
+        event.start_date = d['date']
+        event.end_date = datetime.datetime.combine(event.start_date.date(), datetime.time.max)
+        event.summary = d['summary']
+        event.url = d['url']
+
+        if 'price' in d:
+            event.price = d['price']
+
+        if d['active']:
+            event.status = STATUS.CONFIRMED
+        else:
+            event.status = STATUS.CANCELLED
+
+        if 'city' in d:
+            event.city = d['city']
+        else:
+            event.city = self.find_city_name(d['city_id'])
+        event.transp = TRANSP.OPAQUE
+
+        # "room, address" or "room" or "address" or ""
+        location = ', '.join(filter(None, (d.get('room', ''), d.get('address', ''))))
+        if location:
+            event.location = location
+
+        return event
+
+    def _fetch_cities(self):
+        if self.cities:
+            return
+        self.cities = self.browser.get_cities()
+
+    def find_city_id(self, name):
+        self._fetch_cities()
+        name = name.lower()
+        for c in self.cities:
+            if c.lower() == name:
+                return self.cities[c]['id']
+
+    def find_city_name(self, _id):
+        self._fetch_cities()
+        for c in self.cities.values():
+            if c['id'] == _id:
+                return c['name']