+from django.db.models import FilteredRelation, Q
+from django.views.decorators.cache import cache_control
+from django.views.decorators.vary import vary_on_headers
+from django.utils.decorators import method_decorator
+
+
+@method_decorator([
+ vary_on_headers('X-Requested-With'),
+ cache_control(max_age=1),
+], 'dispatch')
+class ObjectListView(TemplateView):
+ page_size = 100
+ themed_page_size = 10
+ item_template_name = ''
+ orderings = {}
+ default_ordering = None
+
+ def analyse(self):
+ self.is_themed = False
+ self.ctx = ctx = {}
+ ctx['tags'] = []
+
+ def dispatch(self, *args, **kwargs):
+ try:
+ self.analyse()
+ except ResponseInstead as e:
+ return e.response
+ return super().dispatch(*args, **kwargs)
+
+ def get_orderings(self):
+ order = self.get_order()
+ return [
+ {
+ "slug": k,
+ "name": v[1],
+ "active": k == order,
+ "default": v[0] is None,
+ }
+ for k, v in self.orderings.items()
+ ]
+
+ def get_order(self):
+ order = self.request.GET.get('order')
+ if order not in self.orderings:
+ order = self.default_ordering
+ return order
+
+ def order(self, qs):
+ order_tag = self.get_order()
+ if order_tag:
+ order = self.orderings[order_tag]
+ order_by = order[0]
+ if order_by:
+ qs = qs.order_by(order_by)
+ return qs
+
+ def search(self, qs):
+ return qs
+
+ def get_template_names(self):
+ if is_ajax(self.request) or self.request.GET.get('dyn'):
+ if self.is_themed:
+ return self.dynamic_themed_template_name
+ else:
+ return self.dynamic_template_name
+ else:
+ if self.is_themed:
+ return self.themed_template_name
+ else:
+ return self.template_name
+
+ def get_context_data(self, **kwargs):
+ ctx = super().get_context_data()
+ ctx.update(self.ctx)
+ qs = self.get_queryset()
+ qs = self.search(qs)
+ qs = self.order(qs)
+
+ ctx['object_list'] = qs
+ ctx['suggested_tags'] = self.get_suggested_tags(qs)
+ ctx['suggested_tags_by_category'] = split_tags(ctx['suggested_tags'])
+ return ctx
+
+
+class BookList(ObjectListView):
+ title = gettext_lazy('Literatura')
+ list_type = 'books'
+ template_name = 'catalogue/book_list.html'
+ dynamic_template_name = 'catalogue/dynamic_book_list.html'
+ themed_template_name = 'catalogue/themed_book_list.html'
+ dynamic_themed_template_name = 'catalogue/dynamic_themed_book_list.html'
+
+ orderings = {
+ 'pop': ('-popularity__count', gettext_lazy('najpopularniejsze')),
+ 'alpha': (None, gettext_lazy('alfabetycznie')),
+ }
+ default_ordering = 'alpha'
+
+ def get_queryset(self):
+ return Book.objects.filter(parent=None, findable=True)
+
+ def search(self, qs):
+ term = self.request.GET.get('search')
+ if term:
+ meta_rels = TagRelation.objects.filter(tag__category='author')
+ # TODO: search tags in currently displaying language
+ if self.is_themed:
+ rels = meta_rels.filter(tag__name_pl__icontains=term)
+ qs = qs.filter(
+ Q(book__title__icontains=term) |
+ Q(tag_relations__in=rels) |
+ Q(text__icontains=term)
+ ).distinct()
+ else:
+ qs = qs.annotate(
+ meta=FilteredRelation('tag_relations', condition=Q(tag_relations__in=meta_rels))
+ )
+ qs = qs.filter(Q(title__icontains=term) | Q(meta__tag__name_pl__icontains=term)).distinct()
+ return qs
+
+
+class LiteratureView(BookList):
+ def get_suggested_tags(self, queryset):
+ tags = list(get_top_level_related_tags([]))
+ tags.sort(key=lambda t: -t.count)
+ if self.request.user.is_authenticated:
+ tags.extend(list(Tag.objects.filter(user=self.request.user).exclude(name='')))
+ return tags
+
+
+class AudiobooksView(LiteratureView):
+ title = gettext_lazy('Audiobooki')
+ list_type = 'audiobooks'
+
+ def get_queryset(self):
+ return Book.objects.filter(findable=True, media__type='mp3').distinct()
+
+
+class TaggedObjectList(BookList):
+ def analyse(self):
+ super().analyse()
+
+ self.ctx['tags'] = analyse_tags(self.request, self.kwargs['tags'])
+ if len(self.ctx['tags']) > 4:
+ raise Http404
+ self.ctx.update({
+ 'fragment_tags': [],
+ 'work_tags': [],
+ 'user_lists': [],
+ })
+ for tag in self.ctx['tags']:
+ if isinstance(tag, UserList):
+ self.ctx['user_lists'].append(tag)
+ elif tag.category == 'theme':
+ self.ctx['fragment_tags'].append(tag)
+ else:
+ self.ctx['work_tags'].append(tag)
fnp / wolnelektury.git / blobdiff