Merge branch 'master' of https://github.com/fnp/wolnelektury
[wolnelektury.git] / apps / catalogue / feeds.py
index 0d0bf77..ddbcb1d 100644 (file)
 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
-from base64 import b64encode
-import os.path
-
+from django.contrib.sites.models import Site
 from django.contrib.syndication.views import Feed
 from django.core.urlresolvers import reverse
-from django.shortcuts import get_object_or_404
-from django.utils.feedgenerator import Atom1Feed
-from django.conf import settings
-from django.http import Http404
-from django.contrib.sites.models import Site
+from django.db.models import Q
 
-from catalogue.models import Book, Tag
-
-
-_root_feeds = (
-       {u"category": u"author", u"title": u"Autorzy", u"description": u"Utwory wg autorów"},
-       {u"category": u"kind", u"title": u"Rodzaje", u"description": u"Utwory wg rodzajów"},
-       {u"category": u"genre", u"title": u"Gatunki", u"description": u"Utwory wg gatunków"},
-       {u"category": u"epoch", u"title": u"Epoki", u"description": u"Utwory wg epok"},
-)
-
-
-class OPDSFeed(Atom1Feed):
-    link_rel = u"subsection"
-    link_type = u"application/atom+xml"
-
-    try:
-        with open(os.path.join(settings.STATIC_ROOT, "img/book-par ent.png")) as f:
-            t = f.read()
-            _book_parent_img_size = len(t)
-            _book_parent_img = b64encode(t)
-    except:
-        _book_parent_img = _book_parent_img_size = ''
-
-    try:
-        with open(os.path.join(settings.STATIC_ROOT, "img/bo ok.png")) as f:
-            t = f.read()
-            _book_img_size = len(t)
-            _book_img = b64encode(t)
-    except:
-        _book_img = _book_img_size = ''
-
-    def add_root_elements(self, handler):
-        super(OPDSFeed, self).add_root_elements(handler)
-        handler.addQuickElement(u"link", u"", {u"href": reverse("opds_authors"), u"rel": u"start", u"type": u"application/atom+xml"})
-
-
-    def add_item_elements(self, handler, item):
-        """ modified from Atom1Feed.add_item_elements """
-        handler.addQuickElement(u"title", item['title'])
-
-        # add a OPDS Navigation link if there's no enclosure
-        if item['enclosure'] is None:
-            handler.addQuickElement(u"link", u"", {u"href": item['link'], u"rel": u"subsection", u"type": u"application/atom+xml"})
-            # add a "green book" icon
-            handler.addQuickElement(u"link", '',
-                {u"rel": u"http://opds-spec.org/thumbnail",
-                 u"href": u"data:image/png;base64,%s" % self._book_parent_img,
-                 u"length": unicode(self._book_parent_img_size),
-                 u"type": u"image/png"})
-        if item['pubdate'] is not None:
-            handler.addQuickElement(u"updated", rfc3339_date(item['pubdate']).decode('utf-8'))
-
-        # Author information.
-        if item['author_name'] is not None:
-            handler.startElement(u"author", {})
-            handler.addQuickElement(u"name", item['author_name'])
-            if item['author_email'] is not None:
-                handler.addQuickElement(u"email", item['author_email'])
-            if item['author_link'] is not None:
-                handler.addQuickElement(u"uri", item['author_link'])
-            handler.endElement(u"author")
-
-        # Unique ID.
-        if item['unique_id'] is not None:
-            unique_id = item['unique_id']
-        else:
-            unique_id = get_tag_uri(item['link'], item['pubdate'])
-        handler.addQuickElement(u"id", unique_id)
-
-        # Summary.
-        # OPDS needs type=text
-        if item['description'] is not None:
-            handler.addQuickElement(u"summary", item['description'], {u"type": u"text"})
-
-        # Enclosure as OPDS Acquisition Link
-        if item['enclosure'] is not None:
-            handler.addQuickElement(u"link", '',
-                {u"rel": u"http://opds-spec.org/acquisition",
-                 u"href": item['enclosure'].url,
-                 u"length": item['enclosure'].length,
-                 u"type": item['enclosure'].mime_type})
-            # add a "red book" icon
-            handler.addQuickElement(u"link", '',
-                {u"rel": u"http://opds-spec.org/thumbnail",
-                 u"href": u"data:image/png;base64,%s" % self._book_img,
-                 u"length": unicode(self._book_img_size),
-                 u"type": u"image/png"})
-
-        # Categories.
-        for cat in item['categories']:
-            handler.addQuickElement(u"category", u"", {u"term": cat})
-
-        # Rights.
-        if item['item_copyright'] is not None:
-            handler.addQuickElement(u"rights", item['item_copyright'])
-
-
-class RootFeed(Feed):
-    feed_type = OPDSFeed
-    title = u'Wolne Lektury'
-    link = u'http://www.wolnelektury.pl/'
-    description = u"Spis utworów na stronie http://WolneLektury.pl"
-    author_name = u"Wolne Lektury"
-    author_link = u"http://www.wolnelektury.pl/"
-
-    def items(self):
-        return _root_feeds
+from catalogue import models
 
-    def item_title(self, item):
-        return item['title']
+def absolute_url(url):
+    return "http://%s%s" % (Site.objects.get_current().domain, url)
 
-    def item_link(self, item):
-        return reverse(u"opds_by_category", args=[item['category']])
 
-    def item_description(self, item):
-        return item['description']
+class AudiobookFeed(Feed):
+    description = "Audiobooki ostatnio dodane do serwisu Wolne Lektury."
 
+    mime_types = {
+        'mp3': 'audio/mpeg',
+        'ogg': 'audio/ogg',
+        'daisy': 'application/zip',
+    }
 
-class ByCategoryFeed(Feed):
-    feed_type = OPDSFeed
-    link = u'http://www.wolnelektury.pl/'
-    description = u"Spis utworów na stronie http://WolneLektury.pl"
-    author_name = u"Wolne Lektury"
-    author_link = u"http://www.wolnelektury.pl/"
+    titles = {
+        'all': 'WolneLektury.pl - audiobooki we wszystkich formatach',
+        'mp3': 'WolneLektury.pl - audiobooki w formacie MP3',
+        'ogg': 'WolneLektury.pl - audiobooki w formacie Ogg Vorbis',
+        'daisy': 'WolneLektury.pl - audiobooki w formacie DAISY',
+    }
 
-    def get_object(self, request, category):
-        feed = [feed for feed in _root_feeds if feed['category']==category]
-        if feed:
-            feed = feed[0]
-        else:
-            raise Http404
+    def get_object(self, request, type):
+        return {'type': type, 'all': 'all' in request.GET}
 
-        return feed
+    def title(self, args):
+        return self.titles[args['type']]
 
-    def title(self, feed):
-        return feed['title']
+    def link(self, args):
+        return reverse('audiobook_feed', args=(args['type'],))
 
-    def items(self, feed):
-        return (tag for tag in Tag.objects.filter(category=feed['category']) if tag.get_count() > 0)
+    def items(self, args):
+        objects = models.BookMedia.objects.order_by('-uploaded_at')
+        if type == 'all':
+            objects = objects.filter(type__in=('mp3', 'ogg', 'daisy'))
+        else:
+            objects = objects.filter(type=args['type'])
+        if not args['all']:
+            objects = objects[:50]
+        return objects
 
     def item_title(self, item):
         return item.name
 
-    def item_link(self, item):
-        return reverse("opds_by_tag", args=[item.category, item.slug])
-
-    def item_description(self):
-        return u''
+    def item_categories(self, item):
+        return sorted(set(author.name for author in 
+                            item.book.tags.filter(category='author').iterator()))
 
+    def item_description(self, item):
+        lines = []
+        artist = item.extra_info.get('artist_name', None)
+        if artist is not None:
+            lines.append(u'Czyta: %s' % artist)
+        director = item.extra_info.get('artist_name', None)
+        if director is not None:
+            lines.append(u'Reżyseruje: %s' % director)
+        return u'<br/>\n'.join(lines)
 
-class ByTagFeed(Feed):
-    feed_type = OPDSFeed
-    link = u'http://www.wolnelektury.pl/'
-    item_enclosure_mime_type = "application/epub+zip"
-    author_name = u"Wolne Lektury"
-    author_link = u"http://www.wolnelektury.pl/"
-
-    def link(self, tag):
-        return tag.get_absolute_url()
-
-    def title(self, tag):
-        return tag.name
-
-    def description(self, tag):
-        return u"Spis utworów na stronie http://WolneLektury.pl"
-
-    def get_object(self, request, category, slug):
-        return get_object_or_404(Tag, category=category, slug=slug)
-
-    def items(self, tag):
-        books = Book.tagged.with_any([tag])
-        l_tags = Tag.objects.filter(category='book', slug__in=[book.book_tag_slug() for book in books])
-        descendants_keys = [book.pk for book in Book.tagged.with_any(l_tags)]
-        if descendants_keys:
-            books = books.exclude(pk__in=descendants_keys)
-
-        return books
-
-    def item_title(self, book):
-        return book.title
-
-    def item_description(self):
-        return u''
+    def item_link(self, item):
+        return item.book.get_absolute_url()
 
-    def item_link(self, book):
-        return book.get_absolute_url()
+    def item_guid(self, item):
+        return absolute_url(item.file.url)
 
-    def item_author_name(self, book):
-        try:
-            return book.tags.filter(category='author')[0].name
-        except KeyError:
-            return u''
+    def item_enclosure_url(self, item):
+        return absolute_url(item.file.url)
 
-    def item_author_link(self, book):
-        try:
-            return book.tags.filter(category='author')[0].get_absolute_url()
-        except KeyError:
-            return u''
+    def item_enclosure_length(self, item):
+        return item.file.size
 
-    def item_enclosure_url(self, book):
-        return "http://%s%s" % (Site.objects.get_current().domain, book.epub_file.url)
+    def item_enclosure_mime_type(self, item):
+        return self.mime_types[item.type]
 
-    def item_enclosure_length(self, book):
-        return book.epub_file.size
+    def item_pubdate(self, item):
+        return item.uploaded_at