1 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
4 from random import randint, random
5 from urllib.parse import urlparse
6 from django.contrib.contenttypes.models import ContentType
8 from django.conf import settings
9 from django import template
10 from django.template import Node, Variable, Template, Context
11 from django.urls import reverse
12 from django.utils.cache import add_never_cache_headers
13 from django.utils.safestring import mark_safe
14 from django.utils.translation import gettext as _
16 from catalogue.helpers import get_audiobook_tags
17 from catalogue.models import Book, BookMedia, Fragment, Tag, Source
18 from catalogue.constants import LICENSES
19 from club.models import Membership
20 from picture.models import Picture
22 register = template.Library()
35 return '%s%s' % (text[0].upper(), text[1:])
41 def html_title_from_tags(tags):
43 return title_from_tags(tags)
44 template = Template("{{ category }}: <a href='{{ tag.get_absolute_url }}'>{{ tag.name }}</a>")
45 return mark_safe(capfirst(",<br/>".join(
46 template.render(Context({'tag': tag, 'category': _(tag.category)})) for tag in tags)))
49 def simple_title(tags):
52 title.append("%s: %s" % (_(tag.category), tag.name))
53 return capfirst(', '.join(title))
57 def book_title(book, html_links=False):
58 return mark_safe(book.pretty_title(html_links))
62 def book_title_html(book):
63 return book_title(book, html_links=True)
67 def title_from_tags(tags):
68 # TODO: Remove this after adding flection mechanism
69 return simple_title(tags)
73 def nice_title_from_tags(tags, related_tags):
77 result.setdefault(tag.category, []).append(tag)
80 self = split_tags(tags)
84 epoch_reduntant = False
88 t.plural or t.name for t in self['genre']
90 epoch_reduntant = self['genre'][-1].genre_epoch_specific
92 # If we don't have genre,
93 # look if maybe we only have one genre in this context?
94 if 'genre' in related_tags and len(related_tags['genre']) == 1:
96 t.plural or t.name for t in related_tags['genre']
98 epoch_reduntant = related_tags['genre'][-1].genre_epoch_specific
100 # Only use kind if not talking about genre.
102 t.collective_noun or t.name for t in self['kind']
105 elif 'kind' in related_tags and len(related_tags['kind']) == 1:
106 # No info on genre, but there's only one kind related.
109 t.collective_noun or t.name for t in related_tags['kind']
113 # We can't say anything about genre or kind.
114 pieces.append(['Twórczość'])
117 if not epoch_reduntant and 'epoch' in self:
119 form = lambda t: t.adjective_nonmasculine_plural or t.name
121 form = lambda t: t.adjective_feminine_singular or t.name
123 form(t) for t in self['epoch']
128 t.genitive or t.name for t in self['author']
132 for sublist in pieces:
133 for item in sublist[:-2]:
135 for item in sublist[-2:-1]:
136 p.append(item + ' i')
137 p.append(sublist[-1])
143 def book_tree(book_list, books_by_parent):
144 text = "".join("<li><a href='%s'>%s</a>%s</li>" % (
145 book.get_absolute_url(), book.title, book_tree(books_by_parent.get(book, ()), books_by_parent)
146 ) for book in book_list)
149 return mark_safe("<ol>%s</ol>" % text)
155 def audiobook_tree(book_list, books_by_parent):
156 text = "".join("<li><a class='open-player' href='%s'>%s</a>%s</li>" % (
157 reverse("book_player", args=[book.slug]), book.title,
158 audiobook_tree(books_by_parent.get(book, ()), books_by_parent)
159 ) for book in book_list)
162 return mark_safe("<ol>%s</ol>" % text)
168 def book_tree_texml(book_list, books_by_parent, depth=1):
169 return mark_safe("".join("""
170 <cmd name='hspace'><parm>%(depth)dem</parm></cmd>%(title)s
171 <spec cat='align' /><cmd name="note"><parm>%(audiences)s</parm></cmd>
172 <spec cat='align' /><cmd name="note"><parm>%(audiobook)s</parm></cmd>
178 "audiences": ", ".join(book.audiences_pl()),
179 "audiobook": "audiobook" if book.has_media('mp3') else "",
180 "children": book_tree_texml(books_by_parent.get(book.id, ()), books_by_parent, depth + 1)
181 } for book in book_list))
185 def book_tree_csv(author, book_list, books_by_parent, depth=1, max_depth=3, delimeter="\t"):
186 def quote_if_necessary(s):
189 s.replace('"', '\\"')
194 return mark_safe("".join("""%(author)s%(d)s%(preindent)s%(title)s%(d)s%(postindent)s%(audiences)s%(d)s%(audiobook)s
197 "preindent": delimeter * (depth - 1),
198 "postindent": delimeter * (max_depth - depth),
200 "author": quote_if_necessary(author.name),
201 "title": quote_if_necessary(book.title),
202 "audiences": ", ".join(book.audiences_pl()),
203 "audiobook": "audiobook" if book.has_media('mp3') else "",
204 "children": book_tree_csv(author, books_by_parent.get(book.id, ()), books_by_parent, depth + 1)
205 } for book in book_list))
209 def all_editors(extra_info):
211 if 'editors' in extra_info:
212 editors += extra_info['editors']
213 if 'technical_editors' in extra_info:
214 editors += extra_info['technical_editors']
215 # support for extra_info-s from librarian<1.2
216 if 'editor' in extra_info:
217 editors.append(extra_info['editor'])
218 if 'technical_editor' in extra_info:
219 editors.append(extra_info['technical_editor'])
221 ' '.join(p.strip() for p in person.rsplit(',', 1)[::-1])
222 for person in sorted(set(editors)))
226 def catalogue_url(parser, token):
227 bits = token.split_contents()
233 tags_to_remove.append(bit[1:])
235 tags_to_add.append(bit)
237 return CatalogueURLNode(bits[1], tags_to_add, tags_to_remove)
240 class CatalogueURLNode(Node):
241 def __init__(self, list_type, tags_to_add, tags_to_remove):
242 self.tags_to_add = [Variable(tag) for tag in tags_to_add]
243 self.tags_to_remove = [Variable(tag) for tag in tags_to_remove]
244 self.list_type_var = Variable(list_type)
246 def render(self, context):
247 list_type = self.list_type_var.resolve(context)
251 for tag_variable in self.tags_to_add:
252 tag = tag_variable.resolve(context)
253 if isinstance(tag, (list, dict)):
254 tags_to_add += [t for t in tag]
256 tags_to_add.append(tag)
258 for tag_variable in self.tags_to_remove:
259 tag = tag_variable.resolve(context)
261 tags_to_remove += [t for t in tag]
263 tags_to_remove.append(tag)
265 tag_slugs = [tag.url_chunk for tag in tags_to_add]
266 for tag in tags_to_remove:
268 tag_slugs.remove(tag.url_chunk)
272 if len(tag_slugs) > 0:
273 if list_type == 'gallery':
274 return reverse('tagged_object_list_gallery', kwargs={'tags': '/'.join(tag_slugs)})
275 elif list_type == 'audiobooks':
276 return reverse('tagged_object_list_audiobooks', kwargs={'tags': '/'.join(tag_slugs)})
278 return reverse('tagged_object_list', kwargs={'tags': '/'.join(tag_slugs)})
280 if list_type == 'gallery':
281 return reverse('gallery')
282 elif list_type == 'audiobooks':
283 return reverse('audiobook_list')
285 return reverse('book_list')
288 # @register.inclusion_tag('catalogue/tag_list.html')
289 def tag_list(tags, choices=None, category=None, list_type='books'):
293 if category is None and tags:
294 category = tags[0].category
296 category_choices = [tag for tag in choices if tag.category == category]
298 if len(tags) == 1 and category not in [t.category for t in choices]:
303 if category is not None:
304 other = Tag.objects.filter(category=category).exclude(pk__in=[t.pk for t in tags])\
305 .exclude(pk__in=[t.pk for t in category_choices])
306 # Filter out empty tags.
307 ct = ContentType.objects.get_for_model(Picture if list_type == 'gallery' else Book)
308 other = other.filter(items__content_type=ct).distinct()
309 if list_type == 'audiobooks':
310 other = other.filter(id__in=get_audiobook_tags())
311 other = other.only('name', 'slug', 'category')
318 'category_choices': category_choices,
321 'list_type': list_type,
325 @register.inclusion_tag('catalogue/inline_tag_list.html')
326 def inline_tag_list(tags, choices=None, category=None, list_type='books'):
327 return tag_list(tags, choices, category, list_type)
330 @register.inclusion_tag('catalogue/collection_list.html')
331 def collection_list(collections):
332 return {'collections': collections}
335 @register.inclusion_tag('catalogue/book_info.html')
338 'is_picture': isinstance(book, Picture),
343 @register.inclusion_tag('catalogue/work-list.html', takes_context=True)
344 def work_list(context, object_list):
345 request = context.get('request')
346 return {'object_list': object_list, 'request': request}
349 @register.inclusion_tag('catalogue/plain_list.html', takes_context=True)
350 def plain_list(context, object_list, with_initials=True, by_author=False, choice=None, book=None, list_type='books',
351 paged=True, initial_blocks=False):
354 if len(object_list) < settings.CATALOGUE_MIN_INITIALS and not by_author:
355 with_initials = False
356 initial_blocks = False
357 for obj in object_list:
360 initial = obj.sort_key_author
362 initial = obj.get_initial().upper()
363 if initial != last_initial:
364 last_initial = initial
365 names.append((obj.author_unicode() if by_author else initial, []))
366 names[-1][1].append(obj)
367 if names[0] == ('', []):
372 'initial_blocks': initial_blocks,
374 'list_type': list_type,
379 # TODO: These are no longer just books.
380 @register.inclusion_tag('catalogue/related_books.html', takes_context=True)
381 def related_books(context, instance, limit=6, random=1, taken=0):
383 max_books = limit - random
384 is_picture = isinstance(instance, Picture)
386 pics_qs = Picture.objects.all()
388 pics_qs = pics_qs.exclude(pk=instance.pk)
389 pics = Picture.tagged.related_to(instance, pics_qs)
391 # Reserve one spot for an image.
394 books_qs = Book.objects.filter(findable=True)
396 books_qs = books_qs.exclude(common_slug=instance.common_slug).exclude(ancestor=instance)
397 books = Book.tagged.related_to(instance, books_qs)[:max_books]
399 pics = pics[:1 + max_books - books.count()]
401 random_excluded_books = [b.pk for b in books]
402 random_excluded_pics = [p.pk for p in pics]
403 (random_excluded_pics if is_picture else random_excluded_books).append(instance.pk)
406 'request': context['request'],
410 'random_excluded_books': random_excluded_books,
411 'random_excluded_pics': random_excluded_pics,
416 def related_books_2022(book=None, picture=None, limit=4, taken=0):
420 books_qs = Book.objects.filter(findable=True)
422 books_qs = books_qs.exclude(common_slug=book.common_slug).exclude(ancestor=book)
423 instance = book or picture
424 books = Book.tagged.related_to(instance, books_qs)[:max_books]
429 def related_pictures_2022(book=None, picture=None, limit=4, taken=0):
433 books_qs = Picture.objects.all()
434 instance = book or picture
435 books = Picture.tagged.related_to(instance, books_qs)[:max_books]
441 def download_audio(book, daisy=True, mp3=True):
443 if mp3 and book.has_media('mp3'):
444 links.append("<a href='%s'>%s</a>" % (
445 reverse('download_zip_mp3', args=[book.slug]), BookMedia.formats['mp3'].name))
446 if book.has_media('ogg'):
447 links.append("<a href='%s'>%s</a>" % (
448 reverse('download_zip_ogg', args=[book.slug]), BookMedia.formats['ogg'].name))
449 if daisy and book.has_media('daisy'):
450 for dsy in book.get_media('daisy'):
451 links.append("<a href='%s'>%s</a>" % (dsy.file.url, BookMedia.formats['daisy'].name))
452 if daisy and book.has_media('audio.epub'):
453 for dsy in book.get_media('audio.epub'):
454 links.append("<a href='%s'>%s</a>" % (dsy.file.url, BookMedia.formats['audio.epub'].name))
455 return mark_safe("".join(links))
458 @register.inclusion_tag("catalogue/snippets/custom_pdf_link_li.html")
459 def custom_pdf_link_li(book):
462 'NO_CUSTOM_PDF': settings.NO_CUSTOM_PDF,
466 @register.inclusion_tag("catalogue/snippets/license_icon.html")
467 def license_icon(license_url):
468 """Creates a license icon, if the license_url is known."""
469 known = LICENSES.get(license_url)
473 "license_url": license_url,
474 "icon": "img/licenses/%s.png" % known['icon'],
475 "license_description": known['description'],
480 def license_locative(license_url, default):
481 return LICENSES.get(license_url, {}).get('locative', default)
486 return obj.__class__.__name__
490 def source_name(url):
492 netloc = urlparse(url).netloc
494 netloc = urlparse('http://' + url).netloc
497 source, created = Source.objects.get_or_create(netloc=netloc)
498 return source.name or netloc
502 def catalogue_random_book(exclude_ids):
503 from .. import app_settings
504 if random() < app_settings.RELATED_RANDOM_PICTURE_CHANCE:
506 queryset = Book.objects.filter(findable=True).exclude(pk__in=exclude_ids)
507 count = queryset.count()
509 return queryset[randint(0, count - 1)]
515 def choose_fragment(book=None, tag_ids=None):
517 fragment = book.choose_fragment()
519 if tag_ids is not None:
520 tags = Tag.objects.filter(pk__in=tag_ids)
521 fragments = Fragment.tagged.with_all(tags).filter(book__findable=True).order_by().only('id')
523 fragments = Fragment.objects.filter(book__findable=True).order_by().only('id')
524 fragment_count = fragments.count()
525 fragment = fragments[randint(0, fragment_count - 1)] if fragment_count else None
530 def strip_tag(html, tag_name):
531 # docelowo może być warto zainstalować BeautifulSoup do takich rzeczy
533 return re.sub(r"<.?%s\b[^>]*>" % tag_name, "", html)
537 def status(book, user):
540 elif book.is_accessible_to(user):
546 @register.inclusion_tag('catalogue/snippets/content_warning.html')
547 def content_warning(book):
549 'wulgaryzmy': _('vulgar language'),
551 warnings = book.get_extra_info_json().get('content_warnings', [])
553 warnings_def.get(w, w)
561 @register.inclusion_tag('catalogue/preview_ad.html', takes_context=True)
562 def preview_ad(context):
563 book = Book.objects.filter(parent=None, preview=True).first()
567 'accessible': book.is_accessible_to(context['request'].user),
571 @register.inclusion_tag('catalogue/preview_ad_homepage.html', takes_context=True)
572 def preview_ad_homepage(context):
573 book = Book.objects.filter(parent=None, preview=True).first()
577 'accessible': book.is_accessible_to(context['request'].user),