1
0
Fork 0
mirror of synced 2024-06-22 16:10:54 +12:00

ensure results have icons

This commit is contained in:
apkallum 2020-08-25 13:50:31 -04:00 committed by Cristian
parent 3048c0f6dc
commit 6f7cc2b3ef

View file

@ -114,8 +114,8 @@ class PublicArchiveView(ListView):
model = Snapshot
paginate_by = 100
def get_queryset(self, *args, **kwargs):
qs = super(PublicArchiveView, self).get_queryset(*args, **kwargs)
def get_queryset(self, **kwargs):
qs = super().get_queryset(**kwargs)
for snapshot in qs:
snapshot.icons = get_icons(snapshot)
return qs
@ -128,8 +128,9 @@ class PublicArchiveView(ListView):
return redirect(f'/admin/login/?next={self.request.path}')
class SearchResultsView(PublicArchiveView):
def get_queryset(self, *args, **kwargs):
qs = super(PublicArchiveView, self).get_queryset(*args, **kwargs)
def get_queryset(self):
query = self.request.GET.get('q')
results = qs.filter(title__icontains=query)
results = Snapshot.objects.filter(title__icontains=query)
for snapshot in results:
snapshot.icons = get_icons(snapshot)
return results