+# -*- coding: utf-8 -*-
+# 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.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 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
+
+ def item_title(self, item):
+ return item['title']
+
+ def item_link(self, item):
+ return reverse(u"opds_by_category", args=[item['category']])
+
+ def item_description(self, item):
+ return item['description']
+
+
+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/"
+
+ 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
+
+ return feed
+
+ def title(self, feed):
+ return feed['title']
+
+ def items(self, feed):
+ return (tag for tag in Tag.objects.filter(category=feed['category']) if tag.get_count() > 0)
+
+ 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''
+
+
+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, book):
+ return book.get_absolute_url()
+
+ def item_author_name(self, book):
+ try:
+ return book.tags.filter(category='author')[0].name
+ except KeyError:
+ return u''
+
+ def item_author_link(self, book):
+ try:
+ return book.tags.filter(category='author')[0].get_absolute_url()
+ except KeyError:
+ return u''
+
+ def item_enclosure_url(self, book):
+ return "http://%s%s" % (Site.objects.get_current().domain, book.epub_file.url)
+
+ def item_enclosure_length(self, book):
+ return book.epub_file.size
fnp / wolnelektury.git / commitdiff