+++ /dev/null
-# -*- 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 collections import OrderedDict
-from random import randint
-import re
-from django.conf import settings
-from django.db import connection, models, transaction
-from django.db.models import permalink
-import django.dispatch
-from django.contrib.contenttypes.fields import GenericRelation
-from django.core.urlresolvers import reverse
-from django.utils.translation import ugettext_lazy as _
-import jsonfield
-from fnpdjango.storage import BofhFileSystemStorage
-from ssify import flush_ssi_includes
-from newtagging import managers
-from catalogue import constants
-from catalogue.fields import EbookField
-from catalogue.models import Tag, Fragment, BookMedia
-from catalogue.utils import create_zip
-from catalogue import app_settings
-from catalogue import tasks
-
-bofh_storage = BofhFileSystemStorage()
-
-
-def _cover_upload_to(i, n):
- return 'book/cover/%s.jpg' % i.slug
-
-def _cover_thumb_upload_to(i, n):
- return 'book/cover_thumb/%s.jpg' % i.slug
-
-def _ebook_upload_to(upload_path):
- def _upload_to(i, n):
- return upload_path % i.slug
- return _upload_to
-
-
-class Book(models.Model):
- """Represents a book imported from WL-XML."""
- title = models.CharField(_('title'), max_length=120)
- sort_key = models.CharField(_('sort key'), max_length=120, db_index=True, editable=False)
- sort_key_author = models.CharField(_('sort key by author'), max_length=120, db_index=True, editable=False, default=u'')
- slug = models.SlugField(_('slug'), max_length=120, db_index=True,
- unique=True)
- common_slug = models.SlugField(_('slug'), max_length=120, db_index=True)
- language = models.CharField(_('language code'), max_length=3, db_index=True,
- default=app_settings.DEFAULT_LANGUAGE)
- description = models.TextField(_('description'), blank=True)
- created_at = models.DateTimeField(_('creation date'), auto_now_add=True, db_index=True)
- changed_at = models.DateTimeField(_('creation date'), auto_now=True, db_index=True)
- parent_number = models.IntegerField(_('parent number'), default=0)
- extra_info = jsonfield.JSONField(_('extra information'), default={})
- gazeta_link = models.CharField(blank=True, max_length=240)
- wiki_link = models.CharField(blank=True, max_length=240)
- # files generated during publication
-
- cover = EbookField('cover', _('cover'),
- null=True, blank=True,
- upload_to=_cover_upload_to,
- storage=bofh_storage, max_length=255)
- # Cleaner version of cover for thumbs
- cover_thumb = EbookField('cover_thumb', _('cover thumbnail'),
- null=True, blank=True,
- upload_to=_cover_thumb_upload_to,
- max_length=255)
- ebook_formats = constants.EBOOK_FORMATS
- formats = ebook_formats + ['html', 'xml']
-
- parent = models.ForeignKey('self', blank=True, null=True,
- related_name='children')
- ancestor = models.ManyToManyField('self', blank=True, null=True,
- editable=False, related_name='descendant', symmetrical=False)
-
- objects = models.Manager()
- tagged = managers.ModelTaggedItemManager(Tag)
- tags = managers.TagDescriptor(Tag)
- tag_relations = GenericRelation(Tag.intermediary_table_model)
-
- html_built = django.dispatch.Signal()
- published = django.dispatch.Signal()
-
- short_html_url_name = 'catalogue_book_short'
-
- class AlreadyExists(Exception):
- pass
-
- class Meta:
- ordering = ('sort_key',)
- verbose_name = _('book')
- verbose_name_plural = _('books')
- app_label = 'catalogue'
-
- def __unicode__(self):
- return self.title
-
- def save(self, force_insert=False, force_update=False, **kwargs):
- from sortify import sortify
-
- self.sort_key = sortify(self.title)
- self.title = unicode(self.title) # ???
-
- try:
- author = self.tags.filter(category='author')[0].sort_key
- except IndexError:
- author = u''
- self.sort_key_author = author
-
- ret = super(Book, self).save(force_insert, force_update, **kwargs)
-
- return ret
-
- @permalink
- def get_absolute_url(self):
- return ('catalogue.views.book_detail', [self.slug])
-
- @staticmethod
- @permalink
- def create_url(slug):
- return ('catalogue.views.book_detail', [slug])
-
- @property
- def name(self):
- return self.title
-
- def language_code(self):
- return constants.LANGUAGES_3TO2.get(self.language, self.language)
-
- def language_name(self):
- return dict(settings.LANGUAGES).get(self.language_code(), "")
-
- def has_media(self, type_):
- if type_ in Book.formats:
- return bool(getattr(self, "%s_file" % type_))
- else:
- return self.media.filter(type=type_).exists()
-
- def get_media(self, type_):
- if self.has_media(type_):
- if type_ in Book.formats:
- return getattr(self, "%s_file" % type_)
- else:
- return self.media.filter(type=type_)
- else:
- return None
-
- def get_mp3(self):
- return self.get_media("mp3")
- def get_odt(self):
- return self.get_media("odt")
- def get_ogg(self):
- return self.get_media("ogg")
- def get_daisy(self):
- return self.get_media("daisy")
-
- def has_description(self):
- return len(self.description) > 0
- has_description.short_description = _('description')
- has_description.boolean = True
-
- # ugly ugly ugly
- def has_mp3_file(self):
- return bool(self.has_media("mp3"))
- has_mp3_file.short_description = 'MP3'
- has_mp3_file.boolean = True
-
- def has_ogg_file(self):
- return bool(self.has_media("ogg"))
- has_ogg_file.short_description = 'OGG'
- has_ogg_file.boolean = True
-
- def has_daisy_file(self):
- return bool(self.has_media("daisy"))
- has_daisy_file.short_description = 'DAISY'
- has_daisy_file.boolean = True
-
- def wldocument(self, parse_dublincore=True, inherit=True):
- from catalogue.import_utils import ORMDocProvider
- from librarian.parser import WLDocument
-
- if inherit and self.parent:
- meta_fallbacks = self.parent.cover_info()
- else:
- meta_fallbacks = None
-
- return WLDocument.from_file(self.xml_file.path,
- provider=ORMDocProvider(self),
- parse_dublincore=parse_dublincore,
- meta_fallbacks=meta_fallbacks)
-
- @staticmethod
- def zip_format(format_):
- def pretty_file_name(book):
- return "%s/%s.%s" % (
- book.extra_info['author'],
- book.slug,
- format_)
-
- field_name = "%s_file" % format_
- books = Book.objects.filter(parent=None).exclude(**{field_name: ""})
- paths = [(pretty_file_name(b), getattr(b, field_name).path)
- for b in books.iterator()]
- return create_zip(paths, app_settings.FORMAT_ZIPS[format_])
-
- def zip_audiobooks(self, format_):
- bm = BookMedia.objects.filter(book=self, type=format_)
- paths = map(lambda bm: (None, bm.file.path), bm)
- return create_zip(paths, "%s_%s" % (self.slug, format_))
-
- def search_index(self, book_info=None, index=None, index_tags=True, commit=True):
- if index is None:
- from search.index import Index
- index = Index()
- try:
- index.index_book(self, book_info)
- if index_tags:
- index.index_tags()
- if commit:
- index.index.commit()
- except Exception, e:
- index.index.rollback()
- raise e
-
-
- @classmethod
- def from_xml_file(cls, xml_file, **kwargs):
- from django.core.files import File
- from librarian import dcparser
-
- # use librarian to parse meta-data
- book_info = dcparser.parse(xml_file)
-
- if not isinstance(xml_file, File):
- xml_file = File(open(xml_file))
-
- try:
- return cls.from_text_and_meta(xml_file, book_info, **kwargs)
- finally:
- xml_file.close()
-
- @classmethod
- def from_text_and_meta(cls, raw_file, book_info, overwrite=False,
- dont_build=None, search_index=True,
- search_index_tags=True):
- if dont_build is None:
- dont_build = set()
- dont_build = set.union(set(dont_build), set(app_settings.DONT_BUILD))
-
- # check for parts before we do anything
- children = []
- if hasattr(book_info, 'parts'):
- for part_url in book_info.parts:
- try:
- children.append(Book.objects.get(slug=part_url.slug))
- except Book.DoesNotExist:
- raise Book.DoesNotExist(_('Book "%s" does not exist.') %
- part_url.slug)
-
- # Read book metadata
- book_slug = book_info.url.slug
- if re.search(r'[^a-z0-9-]', book_slug):
- raise ValueError('Invalid characters in slug')
- book, created = Book.objects.get_or_create(slug=book_slug)
-
- if created:
- book_shelves = []
- old_cover = None
- else:
- if not overwrite:
- raise Book.AlreadyExists(_('Book %s already exists') % (
- book_slug))
- # Save shelves for this book
- book_shelves = list(book.tags.filter(category='set'))
- old_cover = book.cover_info()
-
- # Save XML file
- book.xml_file.save('%s.xml' % book.slug, raw_file, save=False)
-
- book.language = book_info.language
- book.title = book_info.title
- if book_info.variant_of:
- book.common_slug = book_info.variant_of.slug
- else:
- book.common_slug = book.slug
- book.extra_info = book_info.to_dict()
- book.save()
-
- meta_tags = Tag.tags_from_info(book_info)
-
- book.tags = set(meta_tags + book_shelves)
-
- cover_changed = old_cover != book.cover_info()
- obsolete_children = set(b for b in book.children.all()
- if b not in children)
- notify_cover_changed = []
- for n, child_book in enumerate(children):
- new_child = child_book.parent != book
- child_book.parent = book
- child_book.parent_number = n
- child_book.save()
- if new_child or cover_changed:
- notify_cover_changed.append(child_book)
- # Disown unfaithful children and let them cope on their own.
- for child in obsolete_children:
- child.parent = None
- child.parent_number = 0
- child.save()
- if old_cover:
- notify_cover_changed.append(child)
-
- cls.repopulate_ancestors()
-
- # No saves beyond this point.
-
- # Build cover.
- if 'cover' not in dont_build:
- book.cover.build_delay()
- book.cover_thumb.build_delay()
-
- # Build HTML and ebooks.
- book.html_file.build_delay()
- if not children:
- for format_ in constants.EBOOK_FORMATS_WITHOUT_CHILDREN:
- if format_ not in dont_build:
- getattr(book, '%s_file' % format_).build_delay()
- for format_ in constants.EBOOK_FORMATS_WITH_CHILDREN:
- if format_ not in dont_build:
- getattr(book, '%s_file' % format_).build_delay()
-
- if not settings.NO_SEARCH_INDEX and search_index:
- tasks.index_book.delay(book.id, book_info=book_info, index_tags=search_index_tags)
-
- for child in notify_cover_changed:
- child.parent_cover_changed()
-
- cls.published.send(sender=cls, instance=book)
- return book
-
- @classmethod
- def repopulate_ancestors(cls):
- """Fixes the ancestry cache."""
- # TODO: table names
- with transaction.atomic():
- cursor = connection.cursor()
- if connection.vendor == 'postgres':
- cursor.execute("TRUNCATE catalogue_book_ancestor")
- cursor.execute("""
- WITH RECURSIVE ancestry AS (
- SELECT book.id, book.parent_id
- FROM catalogue_book AS book
- WHERE book.parent_id IS NOT NULL
- UNION
- SELECT ancestor.id, book.parent_id
- FROM ancestry AS ancestor, catalogue_book AS book
- WHERE ancestor.parent_id = book.id
- AND book.parent_id IS NOT NULL
- )
- INSERT INTO catalogue_book_ancestor
- (from_book_id, to_book_id)
- SELECT id, parent_id
- FROM ancestry
- ORDER BY id;
- """)
- else:
- cursor.execute("DELETE FROM catalogue_book_ancestor")
- for b in cls.objects.exclude(parent=None):
- parent = b.parent
- while parent is not None:
- b.ancestor.add(parent)
- parent = parent.parent
-
- def flush_includes(self, languages=True):
- if not languages:
- return
- if languages is True:
- languages = [lc for (lc, _ln) in settings.LANGUAGES]
- flush_ssi_includes([
- template % (self.pk, lang)
- for template in [
- '/katalog/b/%d/mini.%s.html',
- '/katalog/b/%d/mini_nolink.%s.html',
- '/katalog/b/%d/short.%s.html',
- '/katalog/b/%d/wide.%s.html',
- '/api/include/book/%d.%s.json',
- '/api/include/book/%d.%s.xml',
- ]
- for lang in languages
- ])
-
- def cover_info(self, inherit=True):
- """Returns a dictionary to serve as fallback for BookInfo.
-
- For now, the only thing inherited is the cover image.
- """
- need = False
- info = {}
- for field in ('cover_url', 'cover_by', 'cover_source'):
- val = self.extra_info.get(field)
- if val:
- info[field] = val
- else:
- need = True
- if inherit and need and self.parent is not None:
- parent_info = self.parent.cover_info()
- parent_info.update(info)
- info = parent_info
- return info
-
- def related_themes(self):
- return Tag.objects.usage_for_queryset(
- Fragment.objects.filter(models.Q(book=self) | models.Q(book__ancestor=self)),
- counts=True).filter(category='theme')
-
- def parent_cover_changed(self):
- """Called when parent book's cover image is changed."""
- if not self.cover_info(inherit=False):
- if 'cover' not in app_settings.DONT_BUILD:
- self.cover.build_delay()
- self.cover_thumb.build_delay()
- for format_ in constants.EBOOK_FORMATS_WITH_COVERS:
- if format_ not in app_settings.DONT_BUILD:
- getattr(self, '%s_file' % format_).build_delay()
- for child in self.children.all():
- child.parent_cover_changed()
-
- def other_versions(self):
- """Find other versions (i.e. in other languages) of the book."""
- return type(self).objects.filter(common_slug=self.common_slug).exclude(pk=self.pk)
-
- def parents(self):
- books = []
- parent = self.parent
- while parent is not None:
- books.insert(0, parent)
- parent = parent.parent
- return books
-
- def pretty_title(self, html_links=False):
- names = [(tag.name, tag.get_absolute_url())
- for tag in self.tags.filter(category='author')]
- books = self.parents() + [self]
- names.extend([(b.title, b.get_absolute_url()) for b in books])
-
- if html_links:
- names = ['<a href="%s">%s</a>' % (tag[1], tag[0]) for tag in names]
- else:
- names = [tag[0] for tag in names]
- return ', '.join(names)
-
- @classmethod
- def tagged_top_level(cls, tags):
- """ Returns top-level books tagged with `tags`.
-
- It only returns those books which don't have ancestors which are
- also tagged with those tags.
-
- """
- objects = cls.tagged.with_all(tags)
- return objects.exclude(ancestor__in=objects)
-
- @classmethod
- def book_list(cls, filter=None):
- """Generates a hierarchical listing of all books.
-
- Books are optionally filtered with a test function.
-
- """
-
- books_by_parent = {}
- books = cls.objects.all().order_by('parent_number', 'sort_key').only(
- 'title', 'parent', 'slug')
- if filter:
- books = books.filter(filter).distinct()
-
- book_ids = set(b['pk'] for b in books.values("pk").iterator())
- for book in books.iterator():
- parent = book.parent_id
- if parent not in book_ids:
- parent = None
- books_by_parent.setdefault(parent, []).append(book)
- else:
- for book in books.iterator():
- books_by_parent.setdefault(book.parent_id, []).append(book)
-
- orphans = []
- books_by_author = OrderedDict()
- for tag in Tag.objects.filter(category='author').iterator():
- books_by_author[tag] = []
-
- for book in books_by_parent.get(None, ()):
- authors = list(book.tags.filter(category='author'))
- if authors:
- for author in authors:
- books_by_author[author].append(book)
- else:
- orphans.append(book)
-
- return books_by_author, orphans, books_by_parent
-
- _audiences_pl = {
- "SP": (1, u"szkoła podstawowa"),
- "SP1": (1, u"szkoła podstawowa"),
- "SP2": (1, u"szkoła podstawowa"),
- "P": (1, u"szkoła podstawowa"),
- "G": (2, u"gimnazjum"),
- "L": (3, u"liceum"),
- "LP": (3, u"liceum"),
- }
- def audiences_pl(self):
- audiences = self.extra_info.get('audiences', [])
- audiences = sorted(set([self._audiences_pl.get(a, (99, a)) for a in audiences]))
- return [a[1] for a in audiences]
-
- def stage_note(self):
- stage = self.extra_info.get('stage')
- if stage and stage < '0.4':
- return (_('This work needs modernisation'),
- reverse('infopage', args=['wymagajace-uwspolczesnienia']))
- else:
- return None, None
-
- def choose_fragment(self):
- fragments = self.fragments.order_by()
- fragments_count = fragments.count()
- if not fragments_count and self.children.exists():
- fragments = Fragment.objects.filter(book__ancestor=self).order_by()
- fragments_count = fragments.count()
- if fragments_count:
- return fragments[randint(0, fragments_count - 1)]
- elif self.parent:
- return self.parent.choose_fragment()
- else:
- return None
-
-
-# add the file fields
-for format_ in Book.formats:
- field_name = "%s_file" % format_
- # This weird globals() assignment makes Django migrations comfortable.
- _upload_to = _ebook_upload_to('book/%s/%%s.%s' % (format_, format_))
- _upload_to.__name__ = '_%s_upload_to' % format_
- globals()[_upload_to.__name__] = _upload_to
-
- EbookField(format_, _("%s file" % format_.upper()),
- upload_to=_upload_to,
- storage=bofh_storage,
- max_length=255,
- blank=True,
- default=''
- ).contribute_to_class(Book, field_name)