X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/d2b0cc75d0835ad5384bec638fea6ea244eadb40..14eaee7aef2e0f89f0210f087d10eeaed0bedf43:/src/wolnelektury/views.py?ds=inline diff --git a/src/wolnelektury/views.py b/src/wolnelektury/views.py index 415a4fd1b..5e56f5906 100644 --- a/src/wolnelektury/views.py +++ b/src/wolnelektury/views.py @@ -46,13 +46,28 @@ def main_page(request): break break - # Choose a collection for main. - try: - ctx['collection'] = Collection.objects.order_by('?')[:1][0] - except IndexError: - pass - - ctx['best'] = Book.objects.filter(findable=True).order_by('?')[:5] + # Choose collections for main. + ctx['collections'] = Collection.objects.filter(listed=True).order_by('?')[:4] + + best = [] + best_places = 5 + recommended_collection = None + for recommended in Collection.objects.filter(listed=True, role='recommend').order_by('?'): + if recommended_collection is None: + recommended_collection = recommended + books = list(recommended.get_books().exclude(id__in=[b.id for b in best]).order_by('?')[:best_places]) + best.extend(books) + best_places -= len(books) + if not best_places: + break + ctx['recommended_collection'] = recommended_collection + if best_places: + best.extend( + list( + Book.objects.filter(findable=True).exclude(id__in=[b.id for b in best]).order_by('?')[:best_places] + ) + ) + ctx['best'] = best return render(request, "main_page.html", ctx)