Feature implementation

archivebox/core/admin.py

@@ -96,6 +96,13 @@ class SnapshotAdmin(SearchResultsAdminMixin, admin.ModelAdmin):
     actions_template = 'admin/actions_as_select.html'
     form = SnapshotAdminForm
 
+    def get_urls(self):
+        urls = super().get_urls()
+        custom_urls = [
+            path('grid/', self.admin_site.admin_view(self.grid_view),name='grid')
+        ]
+        return custom_urls + urls
+
     def get_queryset(self, request):
         return super().get_queryset(request).prefetch_related('tags')
 
@@ -153,6 +160,31 @@ class SnapshotAdmin(SearchResultsAdminMixin, admin.ModelAdmin):
             obj.url.split('://www.', 1)[-1].split('://', 1)[-1][:64],
         )
 
+    def grid_view(self, request):
+
+        # cl = self.get_changelist_instance(request)
+
+        # Save before monkey patching to restore for changelist list view
+        saved_change_list_template = self.change_list_template
+        saved_list_per_page = self.list_per_page
+        saved_list_max_show_all = self.list_max_show_all
+
+        # Monkey patch here plus core_tags.py
+        self.change_list_template = 'admin/grid_change_list.html'
+        self.list_per_page = 20
+        self.list_max_show_all = self.list_per_page
+
+        # Call monkey patched view
+        rendered_response = self.changelist_view(request)
+
+        # Restore values
+        self.change_list_template =  saved_change_list_template
+        self.list_per_page = saved_list_per_page
+        self.list_max_show_all = saved_list_max_show_all
+
+        return rendered_response
+        
+
     id_str.short_description = 'ID'
     title_str.short_description = 'Title'
     url_str.short_description = 'Original URL'
@@ -218,7 +250,6 @@ class ArchiveBoxAdmin(admin.AdminSite):
 
         return render(template_name='add_links.html', request=request, context=context)
 
-
 admin.site = ArchiveBoxAdmin()
 admin.site.register(get_user_model())
 admin.site.register(Snapshot, SnapshotAdmin)

archivebox/core/templatetags/core_tags.py

@@ -0,0 +1,47 @@
+from django import template
+from django.urls import reverse
+from django.contrib.admin.templatetags.base import InclusionAdminNode
+from django.templatetags.static import static
+
+
+from typing import Union
+
+from core.models import ArchiveResult
+
+register = template.Library()
+
+@register.simple_tag
+def snapshot_image(snapshot):
+    result = ArchiveResult.objects.filter(snapshot=snapshot, extractor='screenshot', status='succeeded').first()
+    if result:
+        return reverse('LinkAssets', args=[f'{str(snapshot.timestamp)}/{result.output}'])
+    
+    return static('archive.png')
+
+@register.filter
+def file_size(num_bytes: Union[int, float]) -> str:
+    for count in ['Bytes','KB','MB','GB']:
+        if num_bytes > -1024.0 and num_bytes < 1024.0:
+            return '%3.1f %s' % (num_bytes, count)
+        num_bytes /= 1024.0
+    return '%3.1f %s' % (num_bytes, 'TB')
+
+def result_list(cl):
+    """
+    Monkey patched result
+    """
+    num_sorted_fields = 0
+    return {
+        'cl': cl,
+        'num_sorted_fields': num_sorted_fields,
+        'results': cl.result_list,
+    }
+
+@register.tag(name='snapshots_grid')
+def result_list_tag(parser, token):
+    return InclusionAdminNode(
+        parser, token,
+        func=result_list,
+        template_name='snapshots_grid.html',
+        takes_context=False,
+    )