src/opds/views.py

   1 # -*- coding: utf-8 -*-
   2 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
   3 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   4 #
   5 import os.path
   6 from urlparse import urljoin
   7
   8 from django.contrib.syndication.views import Feed
   9 from django.core.urlresolvers import reverse
  10 from django.shortcuts import get_object_or_404
  11 from django.utils.feedgenerator import Atom1Feed
  12 from django.conf import settings
  13 from django.http import Http404
  14 from django.contrib.sites.models import Site
  15 from django.utils.functional import lazy
  16
  17 from basicauth import logged_in_or_basicauth, factory_decorator
  18 from catalogue.models import Book, Tag
  19
  20 from search.views import Search
  21 import operator
  22 import logging
  23 import re
  24
  25 from stats.utils import piwik_track
  26
  27 log = logging.getLogger('opds')
  28
  29 _root_feeds = (
  30     {
  31         u"category": u"",
  32         u"link": u"opds_user",
  33         u"link_args": [],
  34         u"title": u"Moje półki",
  35         u"description": u"Półki użytkownika dostępne po zalogowaniu"
  36     },
  37     {
  38         u"category": u"author",
  39         u"link": u"opds_by_category",
  40         u"link_args": [u"author"],
  41         u"title": u"Autorzy",
  42         u"description": u"Utwory wg autorów"
  43     },
  44     {
  45         u"category": u"kind",
  46         u"link": u"opds_by_category",
  47         u"link_args": [u"kind"],
  48         u"title": u"Rodzaje",
  49         u"description": u"Utwory wg rodzajów"
  50     },
  51     {
  52         u"category": u"genre",
  53         u"link": u"opds_by_category",
  54         u"link_args": [u"genre"],
  55         u"title": u"Gatunki",
  56         u"description": u"Utwory wg gatunków"
  57     },
  58     {
  59         u"category": u"epoch",
  60         u"link": u"opds_by_category",
  61         u"link_args": [u"epoch"],
  62         u"title": u"Epoki",
  63         u"description": u"Utwory wg epok"
  64     },
  65 )
  66
  67
  68 current_domain = lazy(lambda: Site.objects.get_current().domain, str)()
  69
  70
  71 def full_url(url):
  72     return urljoin("http://%s" % current_domain, url)
  73
  74
  75 class OPDSFeed(Atom1Feed):
  76     link_rel = u"subsection"
  77     link_type = u"application/atom+xml"
  78
  79     _book_parent_img = lazy(lambda: full_url(os.path.join(settings.STATIC_URL, "img/book-parent.png")), str)()
  80     try:
  81         _book_parent_img_size = unicode(os.path.getsize(os.path.join(settings.STATIC_ROOT, "img/book-parent.png")))
  82     except IOError:
  83         _book_parent_img_size = ''
  84
  85     _book_img = lazy(lambda: full_url(os.path.join(settings.STATIC_URL, "img/book.png")), str)()
  86     try:
  87         _book_img_size = unicode(os.path.getsize(os.path.join(settings.STATIC_ROOT, "img/book.png")))
  88     except IOError:
  89         _book_img_size = ''
  90
  91     def add_root_elements(self, handler):
  92         super(OPDSFeed, self).add_root_elements(handler)
  93         handler.addQuickElement(u"link", None,
  94                                 {u"href": reverse("opds_authors"),
  95                                  u"rel": u"start",
  96                                  u"type": u"application/atom+xml"})
  97         handler.addQuickElement(u"link", None,
  98                                 {u"href": full_url(os.path.join(settings.STATIC_URL, "opensearch.xml")),
  99                                  u"rel": u"search",
 100                                  u"type": u"application/opensearchdescription+xml"})
 101
 102     def add_item_elements(self, handler, item):
 103         """ modified from Atom1Feed.add_item_elements """
 104         handler.addQuickElement(u"title", item['title'])
 105
 106         # add a OPDS Navigation link if there's no enclosure
 107         if not item.get('enclosures') is None:
 108             handler.addQuickElement(
 109                 u"link", u"", {u"href": item['link'], u"rel": u"subsection", u"type": u"application/atom+xml"})
 110             # add a "green book" icon
 111             handler.addQuickElement(
 112                 u"link", '',
 113                 {
 114                     u"rel": u"http://opds-spec.org/thumbnail",
 115                     u"href": self._book_parent_img,
 116                     u"length": self._book_parent_img_size,
 117                     u"type": u"image/png",
 118                 })
 119         if item['pubdate'] is not None:
 120             # FIXME: rfc3339_date is undefined, is this ever run?
 121             handler.addQuickElement(u"updated", rfc3339_date(item['pubdate']).decode('utf-8'))
 122
 123         # Author information.
 124         if item['author_name'] is not None:
 125             handler.startElement(u"author", {})
 126             handler.addQuickElement(u"name", item['author_name'])
 127             if item['author_email'] is not None:
 128                 handler.addQuickElement(u"email", item['author_email'])
 129             if item['author_link'] is not None:
 130                 handler.addQuickElement(u"uri", item['author_link'])
 131             handler.endElement(u"author")
 132
 133         # Unique ID.
 134         if item['unique_id'] is not None:
 135             unique_id = item['unique_id']
 136         else:
 137             # FIXME: get_tag_uri is undefined, is this ever run?
 138             unique_id = get_tag_uri(item['link'], item['pubdate'])
 139         handler.addQuickElement(u"id", unique_id)
 140
 141         # Summary.
 142         # OPDS needs type=text
 143         if item['description'] is not None:
 144             handler.addQuickElement(u"summary", item['description'], {u"type": u"text"})
 145
 146         # Enclosure as OPDS Acquisition Link
 147         for enc in item.get('enclosures', []):
 148             handler.addQuickElement(
 149                 u"link", '',
 150                 {
 151                     u"rel": u"http://opds-spec.org/acquisition",
 152                     u"href": enc.url,
 153                     u"length": enc.length,
 154                     u"type": enc.mime_type,
 155                 })
 156             # add a "red book" icon
 157             handler.addQuickElement(
 158                 u"link", '',
 159                 {
 160                     u"rel": u"http://opds-spec.org/thumbnail",
 161                     u"href": self._book_img,
 162                     u"length": self._book_img_size,
 163                     u"type": u"image/png",
 164                 })
 165
 166         # Categories.
 167         for cat in item['categories']:
 168             handler.addQuickElement(u"category", u"", {u"term": cat})
 169
 170         # Rights.
 171         if item['item_copyright'] is not None:
 172             handler.addQuickElement(u"rights", item['item_copyright'])
 173
 174
 175 class AcquisitionFeed(Feed):
 176     feed_type = OPDSFeed
 177     link = u'http://www.wolnelektury.pl/'
 178     item_enclosure_mime_type = "application/epub+zip"
 179     author_name = u"Wolne Lektury"
 180     author_link = u"http://www.wolnelektury.pl/"
 181
 182     def item_title(self, book):
 183         return book.title
 184
 185     def item_description(self):
 186         return u''
 187
 188     def item_link(self, book):
 189         return book.get_absolute_url()
 190
 191     def item_author_name(self, book):
 192         try:
 193             return book.authors().first().name
 194         except AttributeError:
 195             return u''
 196
 197     def item_author_link(self, book):
 198         try:
 199             return book.authors().first().get_absolute_url()
 200         except AttributeError:
 201             return u''
 202
 203     def item_enclosure_url(self, book):
 204         return full_url(book.epub_url()) if book.epub_file else None
 205
 206     def item_enclosure_length(self, book):
 207         return book.epub_file.size if book.epub_file else None
 208
 209
 210 @piwik_track
 211 class RootFeed(Feed):
 212     feed_type = OPDSFeed
 213     title = u'Wolne Lektury'
 214     link = u'http://wolnelektury.pl/'
 215     description = u"Spis utworów na stronie http://WolneLektury.pl"
 216     author_name = u"Wolne Lektury"
 217     author_link = u"http://wolnelektury.pl/"
 218
 219     def items(self):
 220         return _root_feeds
 221
 222     def item_title(self, item):
 223         return item['title']
 224
 225     def item_link(self, item):
 226         return reverse(item['link'], args=item['link_args'])
 227
 228     def item_description(self, item):
 229         return item['description']
 230
 231
 232 @piwik_track
 233 class ByCategoryFeed(Feed):
 234     feed_type = OPDSFeed
 235     link = u'http://wolnelektury.pl/'
 236     description = u"Spis utworów na stronie http://WolneLektury.pl"
 237     author_name = u"Wolne Lektury"
 238     author_link = u"http://wolnelektury.pl/"
 239
 240     def get_object(self, request, category):
 241         feed = [feed for feed in _root_feeds if feed['category'] == category]
 242         if feed:
 243             feed = feed[0]
 244         else:
 245             raise Http404
 246
 247         return feed
 248
 249     def title(self, feed):
 250         return feed['title']
 251
 252     def items(self, feed):
 253         return Tag.objects.filter(category=feed['category']).exclude(items=None)
 254
 255     def item_title(self, item):
 256         return item.name
 257
 258     def item_link(self, item):
 259         return reverse("opds_by_tag", args=[item.category, item.slug])
 260
 261     def item_description(self):
 262         return u''
 263
 264
 265 @piwik_track
 266 class ByTagFeed(AcquisitionFeed):
 267     def link(self, tag):
 268         return tag.get_absolute_url()
 269
 270     def title(self, tag):
 271         return tag.name
 272
 273     def description(self, tag):
 274         return u"Spis utworów na stronie http://WolneLektury.pl"
 275
 276     def get_object(self, request, category, slug):
 277         return get_object_or_404(Tag, category=category, slug=slug)
 278
 279     def items(self, tag):
 280         return Book.tagged_top_level([tag])
 281
 282
 283 @factory_decorator(logged_in_or_basicauth())
 284 @piwik_track
 285 class UserFeed(Feed):
 286     feed_type = OPDSFeed
 287     link = u'http://www.wolnelektury.pl/'
 288     description = u"Półki użytkownika na stronie http://WolneLektury.pl"
 289     author_name = u"Wolne Lektury"
 290     author_link = u"http://wolnelektury.pl/"
 291
 292     def get_object(self, request):
 293         return request.user
 294
 295     def title(self, user):
 296         return u"Półki użytkownika %s" % user.username
 297
 298     def items(self, user):
 299         return Tag.objects.filter(category='set', user=user).exclude(items=None)
 300
 301     def item_title(self, item):
 302         return item.name
 303
 304     def item_link(self, item):
 305         return reverse("opds_user_set", args=[item.slug])
 306
 307     def item_description(self):
 308         return u''
 309
 310
 311 @factory_decorator(logged_in_or_basicauth())
 312 @piwik_track
 313 class UserSetFeed(AcquisitionFeed):
 314     def link(self, tag):
 315         return tag.get_absolute_url()
 316
 317     def title(self, tag):
 318         return tag.name
 319
 320     def description(self, tag):
 321         return u"Spis utworów na stronie http://WolneLektury.pl"
 322
 323     def get_object(self, request, slug):
 324         return get_object_or_404(Tag, category='set', slug=slug, user=request.user)
 325
 326     def items(self, tag):
 327         return Book.tagged.with_any([tag])
 328
 329
 330 @piwik_track
 331 class SearchFeed(AcquisitionFeed):
 332     description = u"Wyniki wyszukiwania na stronie WolneLektury.pl"
 333     title = u"Wyniki wyszukiwania"
 334
 335     QUOTE_OR_NOT = r'(?:(?=["])"([^"]+)"|([^ ]+))'
 336     INLINE_QUERY_RE = re.compile(
 337         r"author:" + QUOTE_OR_NOT +
 338         "|translator:" + QUOTE_OR_NOT +
 339         "|title:" + QUOTE_OR_NOT +
 340         "|categories:" + QUOTE_OR_NOT +
 341         "|description:" + QUOTE_OR_NOT +
 342         "|text:" + QUOTE_OR_NOT
 343         )
 344     MATCHES = {
 345         'author': (0, 1),
 346         'translator': (2, 3),
 347         'title': (4, 5),
 348         'categories': (6, 7),
 349         'description': (8, 9),
 350         'text': (10, 11),
 351         }
 352
 353     PARAMS_TO_FIELDS = {
 354         'author': 'authors',
 355         'translator': 'translators',
 356         #        'title': 'title',
 357         'categories': 'tag_name_pl',
 358         'description': 'text',
 359         #        'text': 'text',
 360         }
 361
 362     ATOM_PLACEHOLDER = re.compile(r"^{(atom|opds):\w+}$")
 363
 364     def get_object(self, request):
 365         """
 366         For OPDS 1.1 We should handle a query for search terms
 367         and criteria provided either as opensearch or 'inline' query.
 368         OpenSearch defines fields: atom:author, atom:contributor (treated as translator),
 369         atom:title. Inline query provides author, title, categories (treated as book tags),
 370         description (treated as content search terms).
 371
 372         if search terms are provided, we shall search for books
 373         according to Hint information (from author & contributror & title).
 374
 375         but if search terms are empty, we should do a different search
 376         (perhaps for is_book=True)
 377
 378         """
 379
 380         query = request.GET.get('q', '')
 381
 382         inline_criteria = re.findall(self.INLINE_QUERY_RE, query)
 383         if inline_criteria:
 384             remains = re.sub(self.INLINE_QUERY_RE, '', query)
 385             remains = re.sub(r'[ \t]+', ' ', remains)
 386
 387             def get_criteria(criteria, name):
 388                 for c in criteria:
 389                     for p in self.MATCHES[name]:
 390                         if c[p]:
 391                             if p % 2 == 0:
 392                                 return c[p].replace('+', ' ')
 393                             return c[p]
 394                 return None
 395
 396             criteria = dict(map(
 397                 lambda cn: (cn, get_criteria(inline_criteria, cn)),
 398                 ['author', 'translator', 'title', 'categories',
 399                  'description', 'text']))
 400             query = remains
 401             # empty query and text set case?
 402             log.debug("Inline query = [%s], criteria: %s" % (query, criteria))
 403         else:
 404             def remove_dump_data(val):
 405                 """Some clients don't get opds placeholders and just send them."""
 406                 if self.ATOM_PLACEHOLDER.match(val):
 407                     return ''
 408                 return val
 409
 410             criteria = dict(
 411                 (cn, remove_dump_data(request.GET.get(cn, '')))
 412                 for cn in self.MATCHES.keys())
 413             # query is set above.
 414             log.debug("Inline query = [%s], criteria: %s" % (query, criteria))
 415
 416         srch = Search()
 417
 418         book_hit_filter = srch.index.Q(book_id__any=True)
 419         filters = [book_hit_filter] + [srch.index.Q(
 420             **{self.PARAMS_TO_FIELDS.get(cn, cn): criteria[cn]}
 421             ) for cn in self.MATCHES.keys() if cn in criteria
 422             if criteria[cn]]
 423
 424         if query:
 425             q = srch.index.query(
 426                 reduce(
 427                     operator.or_,
 428                     [srch.index.Q(**{self.PARAMS_TO_FIELDS.get(cn, cn): query}) for cn in self.MATCHES.keys()],
 429                     srch.index.Q()))
 430         else:
 431             q = srch.index.query(srch.index.Q())
 432
 433         q = srch.apply_filters(q, filters).field_limit(score=True, fields=['book_id'])
 434         results = q.execute()
 435
 436         book_scores = dict([(r['book_id'], r['score']) for r in results])
 437         books = Book.objects.filter(id__in=set([r['book_id'] for r in results]))
 438         books = list(books)
 439         books.sort(reverse=True, key=lambda book: book_scores[book.id])
 440         return books
 441
 442     def get_link(self, query):
 443         return "%s?q=%s" % (reverse('search'), query)
 444
 445     def items(self, books):
 446         try:
 447             return books
 448         except ValueError:
 449             # too short a query
 450             return []