polishing map fixes

[wolnelektury.git] / src / picture / views.py

diff --git a/src/picture/views.py b/src/picture/views.py
index 194bb98..6bf513d 100644 (file)
--- a/src/picture/views.py
+++ b/src/picture/views.py
@@ -1,26 +1,40 @@
 # 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 Counter
 from django.conf import settings
 from django.contrib.auth.decorators import permission_required
 from django.shortcuts import render, get_object_or_404, render
 from picture.models import Picture
+from catalogue.models import Tag
 from catalogue.utils import split_tags
 from sponsors.models import Sponsor
 from wolnelektury.utils import ajax
+from catalogue.helpers import get_top_level_related_tags
 
 
-def picture_list_thumb(request, filter=None, get_filter=None, template_name='picture/picture_list_thumb.html',
-                       cache_key=None, context=None):
+def picture_list_thumb(request):
     pictures = Picture.objects.all()
-    if filter:
-        pictures = pictures.filter(filter)
-    if get_filter:
-        pictures = pictures.filter(get_filter())
-
+        
     if request.EXPERIMENTS['layout'].value:
+        related_tags = Tag.objects.usage_for_model(Picture, counts=True)
+        related_tags = sorted(related_tags, key=lambda t: -t.count)
+        suggestion_categories = Counter()
+        suggestions = []
+        for t in related_tags:
+            if suggestion_categories[t.category] < 3:
+                suggestion_categories[t.category] += 1
+                suggestions.append(t)
+                if sum(suggestion_categories.values()) == 10:
+                    break
         template_name = 'catalogue/2022/author_detail.html'
-        return render(request, template_name, {'object_list': pictures})
+        return render(request, template_name, {
+            'object_list': pictures,
+            'title': 'Galeria',
+            'tags': [],
+            'suggest': suggestions,
+            'list_type': 'gallery',
+        })
     else:
         return render(request, template_name, {'picture_list': list(pictures)})