from django.contrib.sites.models import Site
from django.core.urlresolvers import reverse
+from django.http.response import HttpResponse
from django.utils.functional import lazy
+from django.db import models
from piston.handler import AnonymousBaseHandler, BaseHandler
from piston.utils import rc
from sorl.thumbnail import default
from catalogue.forms import BookImportForm
from catalogue.models import Book, Tag, BookMedia, Fragment, Collection
from catalogue.models.tag import prefetch_relations
+from catalogue.utils import is_subscribed
from picture.models import Picture
from picture.forms import PictureImportForm
from stats.utils import piwik_track
+from wolnelektury.utils import re_escape
from . import emitters # Register our emitters
book_tag_categories = ['author', 'epoch', 'kind', 'genre']
-def read_tags(tags, allowed):
+def read_tags(tags, request, allowed):
""" Reads a path of filtering tags.
:param str tags: a path of category and slug pairs, like: authors/an-author/...
:returns: list of Tag objects
:raises: ValueError when tags can't be found
"""
+
+ def process(category, slug):
+ if category == 'book':
+ try:
+ books.append(Book.objects.get(slug=slug))
+ except Book.DoesNotExist:
+ raise ValueError('Unknown book.')
+ try:
+ real_tags.append(Tag.objects.get(category=category, slug=slug))
+ except Tag.DoesNotExist:
+ raise ValueError('Tag not found')
+
if not tags:
return [], []
if category not in allowed:
raise ValueError('Category not allowed.')
-
- if category == 'book':
- try:
- books.append(Book.objects.get(slug=slug))
- except Book.DoesNotExist:
- raise ValueError('Unknown book.')
-
- try:
- real_tags.append(Tag.objects.get(category=category, slug=slug))
- except Tag.DoesNotExist:
- raise ValueError('Tag not found')
+ process(category, slug)
+
+ for key in request.GET:
+ if key in category_singular:
+ category = category_singular[key]
+ if category in allowed:
+ for slug in request.GET.getlist(key):
+ process(category, slug)
return real_tags, books
@classmethod
def url(cls, book):
""" Returns Book's URL on the site. """
-
return WL_BASE + book.get_absolute_url()
@classmethod
def children(cls, book):
""" Returns all children for a book. """
-
- return book.children.all()
+ return book.children.order_by('parent_number', 'sort_key')
@classmethod
def media(cls, book):
return MEDIA_BASE + default.backend.get_thumbnail(
book.cover, "139x193").url if book.cover else ''
+ @classmethod
+ def simple_thumb(cls, book):
+ return MEDIA_BASE + book.cover_api_thumb.url if book.cover_api_thumb else ''
+
+ @classmethod
+ def simple_cover(cls, book):
+ return MEDIA_BASE + book.simple_cover.url if book.simple_cover else ''
+
class BookDetailHandler(BaseHandler, BookDetails):
""" Main handler for Book objects.
"""
allowed_methods = ['GET']
fields = ['title', 'parent', 'children'] + Book.formats + [
- 'media', 'url', 'cover', 'cover_thumb'] + [
+ 'media', 'url', 'cover', 'cover_thumb', 'simple_thumb', 'simple_cover', 'fragment_data', 'preview'] + [
category_plural[c] for c in book_tag_categories]
@piwik_track
"""
allowed_methods = ('GET',)
model = Book
- fields = book_tag_categories + ['href', 'title', 'url', 'cover', 'cover_thumb', 'slug']
+ fields = book_tag_categories + ['href', 'title', 'url', 'cover', 'cover_thumb', 'slug', 'simple_thumb']
@classmethod
def genres(cls, book):
@piwik_track
def read(self, request, tags=None, top_level=False, audiobooks=False, daisy=False, pk=None,
- recommended=False, newest=False,
+ recommended=False, newest=False, books=None,
after=None, before=None, count=None):
""" Lists all books with given tags.
return rc.NOT_FOUND
try:
- tags, _ancestors = read_tags(tags, allowed=book_tag_categories)
+ tags, _ancestors = read_tags(tags, request, allowed=book_tag_categories)
except ValueError:
return rc.NOT_FOUND
+ if 'after' in request.GET:
+ after = request.GET['after']
+ if 'before' in request.GET:
+ before = request.GET['before']
+ if 'count' in request.GET:
+ count = request.GET['count']
+
if tags:
if top_level:
books = Book.tagged_top_level(tags)
else:
books = Book.tagged.with_all(tags)
else:
- books = Book.objects.all()
+ books = books if books is not None else Book.objects.all()
books = books.order_by('slug')
if top_level:
else:
books = books[:count]
- if books:
- return books
- else:
- return rc.NOT_FOUND
+ return books
def create(self, request, *args, **kwargs):
return rc.FORBIDDEN
return rc.NOT_FOUND
+class EpubHandler(BookDetailHandler):
+ def read(self, request, slug):
+ if not is_subscribed(request.user):
+ return rc.FORBIDDEN
+ try:
+ book = Book.objects.get(slug=slug)
+ except Book.DoesNotExist:
+ return rc.NOT_FOUND
+ response = HttpResponse(book.get_media('epub'))
+ return response
+
+
class EBooksHandler(AnonymousBooksHandler):
fields = ('author', 'href', 'title', 'cover') + tuple(Book.ebook_formats) + ('slug',)
+class BookProxy(models.Model):
+ class Meta:
+ managed = False
+
+ def __init__(self, book, key):
+ self.book = book
+ self.key = key
+
+ def __getattr__(self, item):
+ if item not in ('book', 'key'):
+ return self.book.__getattribute__(item)
+ else:
+ return self.__getattribute__(item)
+
+
+class QuerySetProxy(models.QuerySet):
+ def __init__(self, l):
+ self.list = l
+
+ def __iter__(self):
+ return iter(self.list)
+
+
+class FilterBooksHandler(AnonymousBooksHandler):
+ fields = book_tag_categories + [
+ 'href', 'title', 'url', 'cover', 'cover_thumb', 'simple_thumb', 'slug', 'key']
+
+ def parse_bool(self, s):
+ if s in ('true', 'false'):
+ return s == 'true'
+ else:
+ return None
+
+ def read(self, request):
+ key_sep = '$'
+ search_string = request.GET.get('search')
+ is_lektura = self.parse_bool(request.GET.get('lektura'))
+ is_audiobook = self.parse_bool(request.GET.get('audiobook'))
+ preview = self.parse_bool(request.GET.get('preview'))
+
+ after = request.GET.get('after')
+ count = int(request.GET.get('count', 50))
+ books = Book.objects.distinct().order_by('slug')
+ if is_lektura is not None:
+ books = books.filter(has_audience=is_lektura)
+ if is_audiobook is not None:
+ if is_audiobook:
+ books = books.filter(media__type='mp3')
+ else:
+ books = books.exclude(media__type='mp3')
+ if preview is not None:
+ books = books.filter(preview=preview)
+ for key in request.GET:
+ if key in category_singular:
+ category = category_singular[key]
+ if category in book_tag_categories:
+ slugs = request.GET[key].split(',')
+ tags = Tag.objects.filter(category=category, slug__in=slugs)
+ books = Book.tagged.with_any(tags, books)
+ if (search_string is not None) and len(search_string) < 3:
+ search_string = None
+ if search_string:
+ search_string = re_escape(search_string)
+ books_author = books.filter(cached_author__iregex='\m' + search_string)
+ books_title = books.filter(title__iregex='\m' + search_string)
+ books_title = books_title.exclude(id__in=list(books_author.values_list('id', flat=True)))
+ if after and (key_sep in after):
+ which, slug = after.split(key_sep, 1)
+ if which == 'title':
+ book_lists = [(books_title.filter(slug__gt=slug), 'title')]
+ else: # which == 'author'
+ book_lists = [(books_author.filter(slug__gt=slug), 'author'), (books_title, 'title')]
+ else:
+ book_lists = [(books_author, 'author'), (books_title, 'title')]
+ else:
+ if after and key_sep in after:
+ which, slug = after.split(key_sep, 1)
+ books = books.filter(slug__gt=slug)
+ book_lists = [(books, 'book')]
+
+ filtered_books = []
+ for book_list, label in book_lists:
+ book_list = book_list.only('slug', 'title', 'cover', 'cover_thumb')
+ for category in book_tag_categories:
+ book_list = prefetch_relations(book_list, category)
+ remaining_count = count - len(filtered_books)
+ new_books = [BookProxy(book, '%s%s%s' % (label, key_sep, book.slug))
+ for book in book_list[:remaining_count]]
+ filtered_books += new_books
+ if len(filtered_books) == count:
+ break
+
+ return QuerySetProxy(filtered_books)
+
+
# add categorized tags fields for Book
def _tags_getter(category):
@classmethod
setattr(BookDetails, plural, _tags_getter(singular))
setattr(BookDetails, singular, _tag_getter(singular))
+
add_tag_getters()
# add fields for files in Book
def _file_getter(book_format):
- field = "%s_file" % book_format
- @classmethod
- def get_file(cls, book):
- f = getattr(book, field)
- if f:
- return MEDIA_BASE + f.url
+ @staticmethod
+ def get_file(book):
+ f_url = book.media_url(book_format)
+ if f_url:
+ return MEDIA_BASE + f_url
else:
return ''
return get_file
"""
allowed_methods = ('GET',)
model = Tag
- fields = ['name', 'href', 'url']
+ fields = ['name', 'href', 'url', 'slug']
@piwik_track
def read(self, request, category=None, pk=None):
except KeyError:
return rc.NOT_FOUND
- tags = Tag.objects.filter(category=category_sng).exclude(items=None)
- if tags.exists():
- return tags
- else:
- return rc.NOT_FOUND
+ after = request.GET.get('after')
+ before = request.GET.get('before')
+ count = request.GET.get('count')
+
+ tags = Tag.objects.filter(category=category_sng).exclude(items=None).order_by('slug')
+
+ book_only = request.GET.get('book_only') == 'true'
+ picture_only = request.GET.get('picture_only') == 'true'
+ if book_only:
+ tags = tags.filter(for_books=True)
+ if picture_only:
+ tags = tags.filter(for_pictures=True)
+
+ if after:
+ tags = tags.filter(slug__gt=after)
+ if before:
+ tags = tags.filter(slug__lt=before)
+
+ if count:
+ if before:
+ tags = list(reversed(tags.order_by('-slug')[:count]))
+ else:
+ tags = tags[:count]
+
+ return tags
class FragmentDetails(object):
fnp / wolnelektury.git / blobdiff