b71926ec366296de53a2bd03cf709b862701ca77
[wolnelektury.git] / src / catalogue / models / book.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 from collections import OrderedDict
6 from datetime import date, timedelta
7 from random import randint
8 import os.path
9 import re
10 import urllib
11 from django.conf import settings
12 from django.db import connection, models, transaction
13 from django.db.models import permalink
14 import django.dispatch
15 from django.contrib.contenttypes.fields import GenericRelation
16 from django.core.urlresolvers import reverse
17 from django.utils.translation import ugettext_lazy as _, get_language
18 from django.utils.deconstruct import deconstructible
19 import jsonfield
20 from fnpdjango.storage import BofhFileSystemStorage
21 from ssify import flush_ssi_includes
22
23 from librarian.html import transform_abstrakt
24 from newtagging import managers
25 from catalogue import constants
26 from catalogue.fields import EbookField
27 from catalogue.models import Tag, Fragment, BookMedia
28 from catalogue.utils import create_zip, gallery_url, gallery_path, split_tags
29 from catalogue.models.tag import prefetched_relations
30 from catalogue import app_settings
31 from catalogue import tasks
32 from wolnelektury.utils import makedirs
33
34 bofh_storage = BofhFileSystemStorage()
35
36
37 @deconstructible
38 class UploadToPath(object):
39     def __init__(self, path):
40         self.path = path
41
42     def __call__(self, instance, filename):
43         return self.path % instance.slug
44
45
46 _cover_upload_to = UploadToPath('book/cover/%s.jpg')
47 _cover_thumb_upload_to = UploadToPath('book/cover_thumb/%s.jpg')
48 _cover_api_thumb_upload_to = UploadToPath('book/cover_api_thumb/%s.jpg')
49 _simple_cover_upload_to = UploadToPath('book/cover_simple/%s.jpg')
50
51
52 def _ebook_upload_to(upload_path):
53     return UploadToPath(upload_path)
54
55
56 class Book(models.Model):
57     """Represents a book imported from WL-XML."""
58     title = models.CharField(_('title'), max_length=32767)
59     sort_key = models.CharField(_('sort key'), max_length=120, db_index=True, editable=False)
60     sort_key_author = models.CharField(
61         _('sort key by author'), max_length=120, db_index=True, editable=False, default=u'')
62     slug = models.SlugField(_('slug'), max_length=120, db_index=True, unique=True)
63     common_slug = models.SlugField(_('slug'), max_length=120, db_index=True)
64     language = models.CharField(_('language code'), max_length=3, db_index=True, default=app_settings.DEFAULT_LANGUAGE)
65     description = models.TextField(_('description'), blank=True)
66     abstract = models.TextField(_('abstract'), blank=True)
67     created_at = models.DateTimeField(_('creation date'), auto_now_add=True, db_index=True)
68     changed_at = models.DateTimeField(_('change date'), auto_now=True, db_index=True)
69     parent_number = models.IntegerField(_('parent number'), default=0)
70     extra_info = jsonfield.JSONField(_('extra information'), default={})
71     gazeta_link = models.CharField(blank=True, max_length=240)
72     wiki_link = models.CharField(blank=True, max_length=240)
73     print_on_demand = models.BooleanField(_('print on demand'), default=False)
74     recommended = models.BooleanField(_('recommended'), default=False)
75     audio_length = models.CharField(_('audio length'), blank=True, max_length=8)
76     preview = models.BooleanField(_('preview'), default=False)
77     preview_until = models.DateField(_('preview until'), blank=True, null=True)
78
79     # files generated during publication
80     cover = EbookField(
81         'cover', _('cover'),
82         null=True, blank=True,
83         upload_to=_cover_upload_to,
84         storage=bofh_storage, max_length=255)
85     # Cleaner version of cover for thumbs
86     cover_thumb = EbookField(
87         'cover_thumb', _('cover thumbnail'),
88         null=True, blank=True,
89         upload_to=_cover_thumb_upload_to,
90         max_length=255)
91     cover_api_thumb = EbookField(
92         'cover_api_thumb', _('cover thumbnail for mobile app'),
93         null=True, blank=True,
94         upload_to=_cover_api_thumb_upload_to,
95         max_length=255)
96     simple_cover = EbookField(
97         'simple_cover', _('cover for mobile app'),
98         null=True, blank=True,
99         upload_to=_simple_cover_upload_to,
100         max_length=255)
101     ebook_formats = constants.EBOOK_FORMATS
102     formats = ebook_formats + ['html', 'xml']
103
104     parent = models.ForeignKey('self', blank=True, null=True, related_name='children')
105     ancestor = models.ManyToManyField('self', blank=True, editable=False, related_name='descendant', symmetrical=False)
106
107     cached_author = models.CharField(blank=True, max_length=240, db_index=True)
108     has_audience = models.BooleanField(default=False)
109
110     objects = models.Manager()
111     tagged = managers.ModelTaggedItemManager(Tag)
112     tags = managers.TagDescriptor(Tag)
113     tag_relations = GenericRelation(Tag.intermediary_table_model)
114
115     html_built = django.dispatch.Signal()
116     published = django.dispatch.Signal()
117
118     short_html_url_name = 'catalogue_book_short'
119
120     class AlreadyExists(Exception):
121         pass
122
123     class Meta:
124         ordering = ('sort_key_author', 'sort_key')
125         verbose_name = _('book')
126         verbose_name_plural = _('books')
127         app_label = 'catalogue'
128
129     def __unicode__(self):
130         return self.title
131
132     def get_initial(self):
133         try:
134             return re.search(r'\w', self.title, re.U).group(0)
135         except AttributeError:
136             return ''
137
138     def authors(self):
139         return self.tags.filter(category='author')
140
141     def tag_unicode(self, category):
142         relations = prefetched_relations(self, category)
143         if relations:
144             return ', '.join(rel.tag.name for rel in relations)
145         else:
146             return ', '.join(self.tags.filter(category=category).values_list('name', flat=True))
147
148     def tags_by_category(self):
149         return split_tags(self.tags.exclude(category__in=('set', 'theme')))
150
151     def author_unicode(self):
152         return self.cached_author
153
154     def translator(self):
155         translators = self.extra_info.get('translators')
156         if not translators:
157             return None
158         if len(translators) > 3:
159             translators = translators[:2]
160             others = ' i inni'
161         else:
162             others = ''
163         return ', '.join(u'\xa0'.join(reversed(translator.split(', ', 1))) for translator in translators) + others
164
165     def cover_source(self):
166         return self.extra_info.get('cover_source', self.parent.cover_source() if self.parent else '')
167
168     def save(self, force_insert=False, force_update=False, **kwargs):
169         from sortify import sortify
170
171         self.sort_key = sortify(self.title)[:120]
172         self.title = unicode(self.title)  # ???
173
174         try:
175             author = self.authors().first().sort_key
176         except AttributeError:
177             author = u''
178         self.sort_key_author = author
179
180         self.cached_author = self.tag_unicode('author')
181         self.has_audience = 'audience' in self.extra_info
182
183         ret = super(Book, self).save(force_insert, force_update, **kwargs)
184
185         return ret
186
187     @permalink
188     def get_absolute_url(self):
189         return 'catalogue.views.book_detail', [self.slug]
190
191     @staticmethod
192     @permalink
193     def create_url(slug):
194         return 'catalogue.views.book_detail', [slug]
195
196     def gallery_path(self):
197         return gallery_path(self.slug)
198
199     def gallery_url(self):
200         return gallery_url(self.slug)
201
202     @property
203     def name(self):
204         return self.title
205
206     def language_code(self):
207         return constants.LANGUAGES_3TO2.get(self.language, self.language)
208
209     def language_name(self):
210         return dict(settings.LANGUAGES).get(self.language_code(), "")
211
212     def is_foreign(self):
213         return self.language_code() != settings.LANGUAGE_CODE
214
215     def set_audio_length(self):
216         length = self.get_audio_length()
217         if length > 0:
218             self.audio_length = self.format_audio_length(length)
219             self.save()
220
221     @staticmethod
222     def format_audio_length(seconds):
223         if seconds < 60*60:
224             minutes = seconds // 60
225             seconds = seconds % 60
226             return '%d:%02d' % (minutes, seconds)
227         else:
228             hours = seconds // 3600
229             minutes = seconds % 3600 // 60
230             seconds = seconds % 60
231             return '%d:%02d:%02d' % (hours, minutes, seconds)
232
233     def get_audio_length(self):
234         from mutagen.mp3 import MP3
235         total = 0
236         for media in self.get_mp3() or ():
237             audio = MP3(media.file.path)
238             total += audio.info.length
239         return int(total)
240
241     def has_media(self, type_):
242         if type_ in Book.formats:
243             return bool(getattr(self, "%s_file" % type_))
244         else:
245             return self.media.filter(type=type_).exists()
246
247     def has_audio(self):
248         return self.has_media('mp3')
249
250     def get_media(self, type_):
251         if self.has_media(type_):
252             if type_ in Book.formats:
253                 return getattr(self, "%s_file" % type_)
254             else:
255                 return self.media.filter(type=type_)
256         else:
257             return None
258
259     def get_mp3(self):
260         return self.get_media("mp3")
261
262     def get_odt(self):
263         return self.get_media("odt")
264
265     def get_ogg(self):
266         return self.get_media("ogg")
267
268     def get_daisy(self):
269         return self.get_media("daisy")
270
271     def media_url(self, format_):
272         media = self.get_media(format_)
273         if media:
274             if self.preview:
275                 return reverse('embargo_link', kwargs={'slug': self.slug, 'format_': format_})
276             else:
277                 return media.url
278         else:
279             return None
280
281     def html_url(self):
282         return self.media_url('html')
283
284     def pdf_url(self):
285         return self.media_url('pdf')
286
287     def epub_url(self):
288         return self.media_url('epub')
289
290     def mobi_url(self):
291         return self.media_url('mobi')
292
293     def txt_url(self):
294         return self.media_url('txt')
295
296     def fb2_url(self):
297         return self.media_url('fb2')
298
299     def xml_url(self):
300         return self.media_url('xml')
301
302     def has_description(self):
303         return len(self.description) > 0
304     has_description.short_description = _('description')
305     has_description.boolean = True
306
307     def has_mp3_file(self):
308         return self.has_media("mp3")
309     has_mp3_file.short_description = 'MP3'
310     has_mp3_file.boolean = True
311
312     def has_ogg_file(self):
313         return self.has_media("ogg")
314     has_ogg_file.short_description = 'OGG'
315     has_ogg_file.boolean = True
316
317     def has_daisy_file(self):
318         return self.has_media("daisy")
319     has_daisy_file.short_description = 'DAISY'
320     has_daisy_file.boolean = True
321
322     def get_audiobooks(self):
323         ogg_files = {}
324         for m in self.media.filter(type='ogg').order_by().iterator():
325             ogg_files[m.name] = m
326
327         audiobooks = []
328         projects = set()
329         for mp3 in self.media.filter(type='mp3').iterator():
330             # ogg files are always from the same project
331             meta = mp3.extra_info
332             project = meta.get('project')
333             if not project:
334                 # temporary fallback
335                 project = u'CzytamySłuchając'
336
337             projects.add((project, meta.get('funded_by', '')))
338
339             media = {'mp3': mp3}
340
341             ogg = ogg_files.get(mp3.name)
342             if ogg:
343                 media['ogg'] = ogg
344             audiobooks.append(media)
345
346         projects = sorted(projects)
347         return audiobooks, projects
348
349     def wldocument(self, parse_dublincore=True, inherit=True):
350         from catalogue.import_utils import ORMDocProvider
351         from librarian.parser import WLDocument
352
353         if inherit and self.parent:
354             meta_fallbacks = self.parent.cover_info()
355         else:
356             meta_fallbacks = None
357
358         return WLDocument.from_file(
359             self.xml_file.path,
360             provider=ORMDocProvider(self),
361             parse_dublincore=parse_dublincore,
362             meta_fallbacks=meta_fallbacks)
363
364     @staticmethod
365     def zip_format(format_):
366         def pretty_file_name(book):
367             return "%s/%s.%s" % (
368                 book.extra_info['author'],
369                 book.slug,
370                 format_)
371
372         field_name = "%s_file" % format_
373         books = Book.objects.filter(parent=None).exclude(**{field_name: ""}).exclude(preview=True)
374         paths = [(pretty_file_name(b), getattr(b, field_name).path) for b in books.iterator()]
375         return create_zip(paths, app_settings.FORMAT_ZIPS[format_])
376
377     def zip_audiobooks(self, format_):
378         bm = BookMedia.objects.filter(book=self, type=format_)
379         paths = map(lambda bm: (None, bm.file.path), bm)
380         return create_zip(paths, "%s_%s" % (self.slug, format_))
381
382     def search_index(self, book_info=None, index=None, index_tags=True, commit=True):
383         if index is None:
384             from search.index import Index
385             index = Index()
386         try:
387             index.index_book(self, book_info)
388             if index_tags:
389                 index.index_tags()
390             if commit:
391                 index.index.commit()
392         except Exception, e:
393             index.index.rollback()
394             raise e
395
396     # will make problems in conjunction with paid previews
397     def download_pictures(self, remote_gallery_url):
398         gallery_path = self.gallery_path()
399         # delete previous files, so we don't include old files in ebooks
400         if os.path.isdir(gallery_path):
401             for filename in os.listdir(gallery_path):
402                 file_path = os.path.join(gallery_path, filename)
403                 os.unlink(file_path)
404         ilustr_elements = list(self.wldocument().edoc.findall('//ilustr'))
405         if ilustr_elements:
406             makedirs(gallery_path)
407             for ilustr in ilustr_elements:
408                 ilustr_src = ilustr.get('src')
409                 ilustr_path = os.path.join(gallery_path, ilustr_src)
410                 urllib.urlretrieve('%s/%s' % (remote_gallery_url, ilustr_src), ilustr_path)
411
412     def load_abstract(self):
413         abstract = self.wldocument(parse_dublincore=False).edoc.getroot().find('.//abstrakt')
414         if abstract is not None:
415             self.abstract = transform_abstrakt(abstract)
416         else:
417             self.abstract = ''
418
419     @classmethod
420     def from_xml_file(cls, xml_file, **kwargs):
421         from django.core.files import File
422         from librarian import dcparser
423
424         # use librarian to parse meta-data
425         book_info = dcparser.parse(xml_file)
426
427         if not isinstance(xml_file, File):
428             xml_file = File(open(xml_file))
429
430         try:
431             return cls.from_text_and_meta(xml_file, book_info, **kwargs)
432         finally:
433             xml_file.close()
434
435     @classmethod
436     def from_text_and_meta(cls, raw_file, book_info, overwrite=False, dont_build=None, search_index=True,
437                            search_index_tags=True, remote_gallery_url=None, days=0):
438         if dont_build is None:
439             dont_build = set()
440         dont_build = set.union(set(dont_build), set(app_settings.DONT_BUILD))
441
442         # check for parts before we do anything
443         children = []
444         if hasattr(book_info, 'parts'):
445             for part_url in book_info.parts:
446                 try:
447                     children.append(Book.objects.get(slug=part_url.slug))
448                 except Book.DoesNotExist:
449                     raise Book.DoesNotExist(_('Book "%s" does not exist.') % part_url.slug)
450
451         # Read book metadata
452         book_slug = book_info.url.slug
453         if re.search(r'[^a-z0-9-]', book_slug):
454             raise ValueError('Invalid characters in slug')
455         book, created = Book.objects.get_or_create(slug=book_slug)
456
457         if created:
458             book_shelves = []
459             old_cover = None
460             book.preview = bool(days)
461             if book.preview:
462                 book.preview_until = date.today() + timedelta(days)
463         else:
464             if not overwrite:
465                 raise Book.AlreadyExists(_('Book %s already exists') % book_slug)
466             # Save shelves for this book
467             book_shelves = list(book.tags.filter(category='set'))
468             old_cover = book.cover_info()
469
470         # Save XML file
471         book.xml_file.save('%s.xml' % book.slug, raw_file, save=False)
472         if book.preview:
473             book.xml_file.set_readable(False)
474
475         book.language = book_info.language
476         book.title = book_info.title
477         if book_info.variant_of:
478             book.common_slug = book_info.variant_of.slug
479         else:
480             book.common_slug = book.slug
481         book.extra_info = book_info.to_dict()
482         book.load_abstract()
483         book.save()
484
485         meta_tags = Tag.tags_from_info(book_info)
486
487         for tag in meta_tags:
488             if not tag.for_books:
489                 tag.for_books = True
490                 tag.save()
491
492         book.tags = set(meta_tags + book_shelves)
493
494         cover_changed = old_cover != book.cover_info()
495         obsolete_children = set(b for b in book.children.all()
496                                 if b not in children)
497         notify_cover_changed = []
498         for n, child_book in enumerate(children):
499             new_child = child_book.parent != book
500             child_book.parent = book
501             child_book.parent_number = n
502             child_book.save()
503             if new_child or cover_changed:
504                 notify_cover_changed.append(child_book)
505         # Disown unfaithful children and let them cope on their own.
506         for child in obsolete_children:
507             child.parent = None
508             child.parent_number = 0
509             child.save()
510             if old_cover:
511                 notify_cover_changed.append(child)
512
513         cls.repopulate_ancestors()
514         tasks.update_counters.delay()
515
516         if remote_gallery_url:
517             book.download_pictures(remote_gallery_url)
518
519         # No saves beyond this point.
520
521         # Build cover.
522         if 'cover' not in dont_build:
523             book.cover.build_delay()
524             book.cover_thumb.build_delay()
525             book.cover_api_thumb.build_delay()
526             book.simple_cover.build_delay()
527
528         # Build HTML and ebooks.
529         book.html_file.build_delay()
530         if not children:
531             for format_ in constants.EBOOK_FORMATS_WITHOUT_CHILDREN:
532                 if format_ not in dont_build:
533                     getattr(book, '%s_file' % format_).build_delay()
534         for format_ in constants.EBOOK_FORMATS_WITH_CHILDREN:
535             if format_ not in dont_build:
536                 getattr(book, '%s_file' % format_).build_delay()
537
538         if not settings.NO_SEARCH_INDEX and search_index:
539             tasks.index_book.delay(book.id, book_info=book_info, index_tags=search_index_tags)
540
541         for child in notify_cover_changed:
542             child.parent_cover_changed()
543
544         book.save()  # update sort_key_author
545         book.update_popularity()
546         cls.published.send(sender=cls, instance=book)
547         return book
548
549     @classmethod
550     @transaction.atomic
551     def repopulate_ancestors(cls):
552         """Fixes the ancestry cache."""
553         # TODO: table names
554         cursor = connection.cursor()
555         if connection.vendor == 'postgres':
556             cursor.execute("TRUNCATE catalogue_book_ancestor")
557             cursor.execute("""
558                 WITH RECURSIVE ancestry AS (
559                     SELECT book.id, book.parent_id
560                     FROM catalogue_book AS book
561                     WHERE book.parent_id IS NOT NULL
562                     UNION
563                     SELECT ancestor.id, book.parent_id
564                     FROM ancestry AS ancestor, catalogue_book AS book
565                     WHERE ancestor.parent_id = book.id
566                         AND book.parent_id IS NOT NULL
567                     )
568                 INSERT INTO catalogue_book_ancestor
569                     (from_book_id, to_book_id)
570                     SELECT id, parent_id
571                     FROM ancestry
572                     ORDER BY id;
573                 """)
574         else:
575             cursor.execute("DELETE FROM catalogue_book_ancestor")
576             for b in cls.objects.exclude(parent=None):
577                 parent = b.parent
578                 while parent is not None:
579                     b.ancestor.add(parent)
580                     parent = parent.parent
581
582     def flush_includes(self, languages=True):
583         if not languages:
584             return
585         if languages is True:
586             languages = [lc for (lc, _ln) in settings.LANGUAGES]
587         flush_ssi_includes([
588             template % (self.pk, lang)
589             for template in [
590                 '/katalog/b/%d/mini.%s.html',
591                 '/katalog/b/%d/mini_nolink.%s.html',
592                 '/katalog/b/%d/short.%s.html',
593                 '/katalog/b/%d/wide.%s.html',
594                 '/api/include/book/%d.%s.json',
595                 '/api/include/book/%d.%s.xml',
596                 ]
597             for lang in languages
598             ])
599
600     def cover_info(self, inherit=True):
601         """Returns a dictionary to serve as fallback for BookInfo.
602
603         For now, the only thing inherited is the cover image.
604         """
605         need = False
606         info = {}
607         for field in ('cover_url', 'cover_by', 'cover_source'):
608             val = self.extra_info.get(field)
609             if val:
610                 info[field] = val
611             else:
612                 need = True
613         if inherit and need and self.parent is not None:
614             parent_info = self.parent.cover_info()
615             parent_info.update(info)
616             info = parent_info
617         return info
618
619     def related_themes(self):
620         return Tag.objects.usage_for_queryset(
621             Fragment.objects.filter(models.Q(book=self) | models.Q(book__ancestor=self)),
622             counts=True).filter(category='theme')
623
624     def parent_cover_changed(self):
625         """Called when parent book's cover image is changed."""
626         if not self.cover_info(inherit=False):
627             if 'cover' not in app_settings.DONT_BUILD:
628                 self.cover.build_delay()
629                 self.cover_thumb.build_delay()
630                 self.cover_api_thumb.build_delay()
631                 self.simple_cover.build_delay()
632             for format_ in constants.EBOOK_FORMATS_WITH_COVERS:
633                 if format_ not in app_settings.DONT_BUILD:
634                     getattr(self, '%s_file' % format_).build_delay()
635             for child in self.children.all():
636                 child.parent_cover_changed()
637
638     def other_versions(self):
639         """Find other versions (i.e. in other languages) of the book."""
640         return type(self).objects.filter(common_slug=self.common_slug).exclude(pk=self.pk)
641
642     def parents(self):
643         books = []
644         parent = self.parent
645         while parent is not None:
646             books.insert(0, parent)
647             parent = parent.parent
648         return books
649
650     def pretty_title(self, html_links=False):
651         names = [(tag.name, tag.get_absolute_url()) for tag in self.authors().only('name', 'category', 'slug')]
652         books = self.parents() + [self]
653         names.extend([(b.title, b.get_absolute_url()) for b in books])
654
655         if html_links:
656             names = ['<a href="%s">%s</a>' % (tag[1], tag[0]) for tag in names]
657         else:
658             names = [tag[0] for tag in names]
659         return ', '.join(names)
660
661     def publisher(self):
662         publisher = self.extra_info['publisher']
663         if isinstance(publisher, basestring):
664             return publisher
665         elif isinstance(publisher, list):
666             return ', '.join(publisher)
667
668     @classmethod
669     def tagged_top_level(cls, tags):
670         """ Returns top-level books tagged with `tags`.
671
672         It only returns those books which don't have ancestors which are
673         also tagged with those tags.
674
675         """
676         objects = cls.tagged.with_all(tags)
677         return objects.exclude(ancestor__in=objects)
678
679     @classmethod
680     def book_list(cls, book_filter=None):
681         """Generates a hierarchical listing of all books.
682
683         Books are optionally filtered with a test function.
684
685         """
686
687         books_by_parent = {}
688         books = cls.objects.order_by('parent_number', 'sort_key').only('title', 'parent', 'slug')
689         if book_filter:
690             books = books.filter(book_filter).distinct()
691
692             book_ids = set(b['pk'] for b in books.values("pk").iterator())
693             for book in books.iterator():
694                 parent = book.parent_id
695                 if parent not in book_ids:
696                     parent = None
697                 books_by_parent.setdefault(parent, []).append(book)
698         else:
699             for book in books.iterator():
700                 books_by_parent.setdefault(book.parent_id, []).append(book)
701
702         orphans = []
703         books_by_author = OrderedDict()
704         for tag in Tag.objects.filter(category='author').iterator():
705             books_by_author[tag] = []
706
707         for book in books_by_parent.get(None, ()):
708             authors = list(book.authors().only('pk'))
709             if authors:
710                 for author in authors:
711                     books_by_author[author].append(book)
712             else:
713                 orphans.append(book)
714
715         return books_by_author, orphans, books_by_parent
716
717     _audiences_pl = {
718         "SP": (1, u"szkoła podstawowa"),
719         "SP1": (1, u"szkoła podstawowa"),
720         "SP2": (1, u"szkoła podstawowa"),
721         "SP3": (1, u"szkoła podstawowa"),
722         "P": (1, u"szkoła podstawowa"),
723         "G": (2, u"gimnazjum"),
724         "L": (3, u"liceum"),
725         "LP": (3, u"liceum"),
726     }
727
728     def audiences_pl(self):
729         audiences = self.extra_info.get('audiences', [])
730         audiences = sorted(set([self._audiences_pl.get(a, (99, a)) for a in audiences]))
731         return [a[1] for a in audiences]
732
733     def stage_note(self):
734         stage = self.extra_info.get('stage')
735         if stage and stage < '0.4':
736             return (_('This work needs modernisation'),
737                     reverse('infopage', args=['wymagajace-uwspolczesnienia']))
738         else:
739             return None, None
740
741     def choose_fragment(self):
742         fragments = self.fragments.order_by()
743         fragments_count = fragments.count()
744         if not fragments_count and self.children.exists():
745             fragments = Fragment.objects.filter(book__ancestor=self).order_by()
746             fragments_count = fragments.count()
747         if fragments_count:
748             return fragments[randint(0, fragments_count - 1)]
749         elif self.parent:
750             return self.parent.choose_fragment()
751         else:
752             return None
753
754     def fragment_data(self):
755         fragment = self.choose_fragment()
756         if fragment:
757             return {'title': fragment.book.pretty_title(), 'html': fragment.get_short_text()}
758         else:
759             return None
760
761     def update_popularity(self):
762         count = self.tags.filter(category='set').values('user').order_by('user').distinct().count()
763         try:
764             pop = self.popularity
765             pop.count = count
766             pop.save()
767         except BookPopularity.DoesNotExist:
768             BookPopularity.objects.create(book=self, count=count)
769
770     def ridero_link(self):
771         return 'https://ridero.eu/%s/books/wl_%s/' % (get_language(), self.slug.replace('-', '_'))
772
773     def like(self, user):
774         from social.utils import likes, get_set, set_sets
775         if not likes(user, self):
776             tag = get_set(user, '')
777             set_sets(user, self, [tag])
778
779     def unlike(self, user):
780         from social.utils import likes, set_sets
781         if likes(user, self):
782             set_sets(user, self, [])
783
784
785 def add_file_fields():
786     for format_ in Book.formats:
787         field_name = "%s_file" % format_
788         # This weird globals() assignment makes Django migrations comfortable.
789         _upload_to = _ebook_upload_to('book/%s/%%s.%s' % (format_, format_))
790         _upload_to.__name__ = '_%s_upload_to' % format_
791         globals()[_upload_to.__name__] = _upload_to
792
793         EbookField(
794             format_, _("%s file" % format_.upper()),
795             upload_to=_upload_to,
796             storage=bofh_storage,
797             max_length=255,
798             blank=True,
799             default=''
800         ).contribute_to_class(Book, field_name)
801
802
803 add_file_fields()
804
805
806 class BookPopularity(models.Model):
807     book = models.OneToOneField(Book, related_name='popularity')
808     count = models.IntegerField(default=0, db_index=True)