From 6f7cc2b3ef98ae190d11514917fc6c09edb88474 Mon Sep 17 00:00:00 2001 From: apkallum Date: Tue, 25 Aug 2020 13:50:31 -0400 Subject: [PATCH] ensure results have icons --- archivebox/core/views.py | 11 ++++++----- 1 file changed, 6 insertions(+), 5 deletions(-) diff --git a/archivebox/core/views.py b/archivebox/core/views.py index e6d19de7..3d349a0f 100644 --- a/archivebox/core/views.py +++ b/archivebox/core/views.py @@ -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 \ No newline at end of file